We introduce Granger Causality Kolmogorov-Arnold Networks (GCKAN), an innovative architecture that extends the recently proposed Kolmogorov-Arnold Networks (KAN) to the domain of causal inference. By extracting base weights from KAN layers and incorporating the sparsity-inducing penalty along with ridge regularization, GCKAN infers the Granger causality from time series while enabling automatic time lag selection. Additionally, we propose an algorithm leveraging time-reversed Granger causality to enhance inference accuracy. The algorithm compares prediction and sparse-inducing losses derived from the original and time-reversed series, automatically selecting the casual relationship with the higher score or integrating the results to mitigate spurious connectivities. Comprehensive experiments conducted on Lorenz-96, gene regulatory networks, fMRI BOLD signals, and VAR datasets demonstrate that the proposed model achieves competitive performance to state-of-the-art methods in inferring Granger causality from nonlinear, high-dimensional, and limited-sample time series.

本文提出了Granger因果关系Kolmogorov-Arnold网络（GCKAN），填补了因果推断领域的研究空白。GCKAN通过从KAN层提取基础权重并结合稀疏惩罚与岭回归，能够自动选择时间滞后并推断时间序列中的Granger因果关系。实验结果显示，该模型在推断非线性、高维和样本有限的时间序列中的Granger因果关系方面表现出与最先进方法相媲美的性能。

用于时间序列Granger因果关系推断的Kolmogorov-Arnold网络