Deep Koopman operator framework for causal discovery in nonlinear dynamical systems

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Nathaniel, Juan, Roesch, Carla, Buch, Jatan, DeSantis, Derek, Rupe, Adam, Lamb, Kara, Gentine, Pierre
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866908372831305728
author Nathaniel, Juan
Roesch, Carla
Buch, Jatan
DeSantis, Derek
Rupe, Adam
Lamb, Kara
Gentine, Pierre
author_facet Nathaniel, Juan
Roesch, Carla
Buch, Jatan
DeSantis, Derek
Rupe, Adam
Lamb, Kara
Gentine, Pierre
contents We use a deep Koopman operator-theoretic formalism to develop a novel causal discovery algorithm, Kausal. Causal discovery aims to identify cause-effect mechanisms for better scientific understanding, explainable decision-making, and more accurate modeling. Standard statistical frameworks, such as Granger causality, lack the ability to quantify causal relationships in nonlinear dynamics due to the presence of complex feedback mechanisms, timescale mixing, and nonstationarity. This presents a challenge in studying many real-world systems, such as the Earth's climate. Meanwhile, Koopman operator methods have emerged as a promising tool for approximating nonlinear dynamics in a linear space of observables. In Kausal, we propose to leverage this powerful idea for causal analysis where optimal observables are inferred using deep learning. Causal estimates are then evaluated in a reproducing kernel Hilbert space, and defined as the distance between the marginal dynamics of the effect and the joint dynamics of the cause-effect observables. Our numerical experiments demonstrate Kausal's superior ability in discovering and characterizing causal signals compared to existing approaches of prescribed observables. Lastly, we extend our analysis to observations of El Niño-Southern Oscillation highlighting our algorithm's applicability to real-world phenomena. Our code is available at https://github.com/juannat7/kausal.
format Preprint
id arxiv_https___arxiv_org_abs_2505_14828
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Deep Koopman operator framework for causal discovery in nonlinear dynamical systems
Nathaniel, Juan
Roesch, Carla
Buch, Jatan
DeSantis, Derek
Rupe, Adam
Lamb, Kara
Gentine, Pierre
Machine Learning
We use a deep Koopman operator-theoretic formalism to develop a novel causal discovery algorithm, Kausal. Causal discovery aims to identify cause-effect mechanisms for better scientific understanding, explainable decision-making, and more accurate modeling. Standard statistical frameworks, such as Granger causality, lack the ability to quantify causal relationships in nonlinear dynamics due to the presence of complex feedback mechanisms, timescale mixing, and nonstationarity. This presents a challenge in studying many real-world systems, such as the Earth's climate. Meanwhile, Koopman operator methods have emerged as a promising tool for approximating nonlinear dynamics in a linear space of observables. In Kausal, we propose to leverage this powerful idea for causal analysis where optimal observables are inferred using deep learning. Causal estimates are then evaluated in a reproducing kernel Hilbert space, and defined as the distance between the marginal dynamics of the effect and the joint dynamics of the cause-effect observables. Our numerical experiments demonstrate Kausal's superior ability in discovering and characterizing causal signals compared to existing approaches of prescribed observables. Lastly, we extend our analysis to observations of El Niño-Southern Oscillation highlighting our algorithm's applicability to real-world phenomena. Our code is available at https://github.com/juannat7/kausal.
title Deep Koopman operator framework for causal discovery in nonlinear dynamical systems
topic Machine Learning
url https://arxiv.org/abs/2505.14828