Granger Causality Detection with Kolmogorov-Arnold Networks

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Lin, Hongyu, Ren, Mohan, Barucca, Paolo, Aste, Tomaso
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910755532570624
author Lin, Hongyu
Ren, Mohan
Barucca, Paolo
Aste, Tomaso
author_facet Lin, Hongyu
Ren, Mohan
Barucca, Paolo
Aste, Tomaso
contents Discovering causal relationships in time series data is central in many scientific areas, ranging from economics to climate science. Granger causality is a powerful tool for causality detection. However, its original formulation is limited by its linear form and only recently nonlinear machine-learning generalizations have been introduced. This study contributes to the definition of neural Granger causality models by investigating the application of Kolmogorov-Arnold networks (KANs) in Granger causality detection and comparing their capabilities against multilayer perceptrons (MLP). In this work, we develop a framework called Granger Causality KAN (GC-KAN) along with a tailored training approach designed specifically for Granger causality detection. We test this framework on both Vector Autoregressive (VAR) models and chaotic Lorenz-96 systems, analysing the ability of KANs to sparsify input features by identifying Granger causal relationships, providing a concise yet accurate model for Granger causality detection. Our findings show the potential of KANs to outperform MLPs in discerning interpretable Granger causal relationships, particularly for the ability of identifying sparse Granger causality patterns in high-dimensional settings, and more generally, the potential of AI in causality discovery for the dynamical laws in physical systems.
format Preprint
id arxiv_https___arxiv_org_abs_2412_15373
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Granger Causality Detection with Kolmogorov-Arnold Networks
Lin, Hongyu
Ren, Mohan
Barucca, Paolo
Aste, Tomaso
Machine Learning
Artificial Intelligence
Discovering causal relationships in time series data is central in many scientific areas, ranging from economics to climate science. Granger causality is a powerful tool for causality detection. However, its original formulation is limited by its linear form and only recently nonlinear machine-learning generalizations have been introduced. This study contributes to the definition of neural Granger causality models by investigating the application of Kolmogorov-Arnold networks (KANs) in Granger causality detection and comparing their capabilities against multilayer perceptrons (MLP). In this work, we develop a framework called Granger Causality KAN (GC-KAN) along with a tailored training approach designed specifically for Granger causality detection. We test this framework on both Vector Autoregressive (VAR) models and chaotic Lorenz-96 systems, analysing the ability of KANs to sparsify input features by identifying Granger causal relationships, providing a concise yet accurate model for Granger causality detection. Our findings show the potential of KANs to outperform MLPs in discerning interpretable Granger causal relationships, particularly for the ability of identifying sparse Granger causality patterns in high-dimensional settings, and more generally, the potential of AI in causality discovery for the dynamical laws in physical systems.
title Granger Causality Detection with Kolmogorov-Arnold Networks
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2412.15373