Causal hybrid modeling with double machine learning

Kai-Hendrik Cohrs, Gherardo Varando, Nuno Carvalhais, Markus Reichstein, Gustau Camps-Valls
2/20/2024
cs.LGstat.ME

No official code found. Best implementation: KaiHCohrs/dml-4-fluxes-chm (1 star, MIT, last commit 2025-04-15). 1 more below.

Abstract

Hybrid modeling integrates machine learning with scientific knowledge to enhance interpretability, generalization, and adherence to natural laws. Nevertheless, equifinality and regularization biases pose challenges in hybrid modeling to achieve these purposes. This paper introduces a novel approach to estimating hybrid models via a causal inference framework, specifically employing Double Machine Learning (DML) to estimate causal effects. We showcase its use for the Earth sciences on two problems related to carbon dioxide fluxes. In the $Q_{10}$ model, we demonstrate that DML-based hybrid modeling is superior in estimating causal parameters over end-to-end deep neural network (DNN) approaches, proving efficiency, robustness to bias from regularization methods, and circumventing equifinality. Our approach, applied to carbon flux partitioning, exhibits flexibility in accommodating heterogeneous causal effects. The study emphasizes the necessity of explicitly defining causal graphs and relationships, advocating for this as a general best practice. We encourage the continued exploration of causality in hybrid models for more interpretable and trustworthy results in knowledge-guided machine learning.

View on arXivView PDF

Code Implementations(2)

MIT
10Python, Jupyter NotebookDec 8, 20231 years agoMIT
MIT
00PythonDec 8, 20231 years agoMIT

Similar Papers

Cite this paper

@article{cohrs2024causal,
  title  = {Causal hybrid modeling with double machine learning},
  author = {Kai-Hendrik Cohrs and Gherardo Varando and Nuno Carvalhais and Markus Reichstein and Gustau Camps-Valls},
  year   = {2024},
  eprint = {2402.13332},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2402.13332v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2402.13332.svg)](https://codeofpaper.com/paper/2402.13332)

Discussion