Saved in:
Bibliographic Details
Main Authors: Adesunkanmi, Rahmat, Pokuri, Balaji Sesha Srikanth, Kumar, Ratnesh
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2404.16326
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911852877840384
author Adesunkanmi, Rahmat
Pokuri, Balaji Sesha Srikanth
Kumar, Ratnesh
author_facet Adesunkanmi, Rahmat
Pokuri, Balaji Sesha Srikanth
Kumar, Ratnesh
contents In many real-world applications where the system dynamics has an underlying interdependency among its variables (such as power grid, economics, neuroscience, omics networks, environmental ecosystems, and others), one is often interested in knowing whether the past values of one time series influences the future of another, known as Granger causality, and the associated underlying dynamics. This paper introduces a Koopman-inspired framework that leverages neural networks for data-driven learning of the Koopman bases, termed NeuroKoopman Dynamic Causal Discovery (NKDCD), for reliably inferring the Granger causality along with the underlying nonlinear dynamics. NKDCD employs an autoencoder architecture that lifts the nonlinear dynamics to a higher dimension using data-learned bases, where the lifted time series can be reliably modeled linearly. The lifting function, the linear Granger causality lag matrices, and the projection function (from lifted space to base space) are all represented as multilayer perceptrons and are all learned simultaneously in one go. NKDCD also utilizes sparsity-inducing penalties on the weights of the lag matrices, encouraging the model to select only the needed causal dependencies within the data. Through extensive testing on practically applicable datasets, it is shown that the NKDCD outperforms the existing nonlinear Granger causality discovery approaches.
format Preprint
id arxiv_https___arxiv_org_abs_2404_16326
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle NeuroKoopman Dynamic Causal Discovery
Adesunkanmi, Rahmat
Pokuri, Balaji Sesha Srikanth
Kumar, Ratnesh
Machine Learning
In many real-world applications where the system dynamics has an underlying interdependency among its variables (such as power grid, economics, neuroscience, omics networks, environmental ecosystems, and others), one is often interested in knowing whether the past values of one time series influences the future of another, known as Granger causality, and the associated underlying dynamics. This paper introduces a Koopman-inspired framework that leverages neural networks for data-driven learning of the Koopman bases, termed NeuroKoopman Dynamic Causal Discovery (NKDCD), for reliably inferring the Granger causality along with the underlying nonlinear dynamics. NKDCD employs an autoencoder architecture that lifts the nonlinear dynamics to a higher dimension using data-learned bases, where the lifted time series can be reliably modeled linearly. The lifting function, the linear Granger causality lag matrices, and the projection function (from lifted space to base space) are all represented as multilayer perceptrons and are all learned simultaneously in one go. NKDCD also utilizes sparsity-inducing penalties on the weights of the lag matrices, encouraging the model to select only the needed causal dependencies within the data. Through extensive testing on practically applicable datasets, it is shown that the NKDCD outperforms the existing nonlinear Granger causality discovery approaches.
title NeuroKoopman Dynamic Causal Discovery
topic Machine Learning
url https://arxiv.org/abs/2404.16326