arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2408.08023cs.LGcs.AI

基于短期不变性的卷积神经网络进行时间序列数据因果发现

Causal Discovery from Time-Series Data with Short-Term Invariance-Based Convolutional Neural Networks

Faculty of Computing(计算机学院);Harbin Institute of Technology(哈尔滨工业大学);The Artificial Intelligence Institute(人工智能研究所);The Department of Computer Science(计算机科学系);The University of North Carolina at Chapel Hill(北卡罗来纳大学教堂山分校)

Rujia Shen, Boran Wang, Chao Zhao, Yi Guan, Jingchi Jiang

AI总结:

STIC通过卷积神经网络利用短期不变性从时间序列数据中发现因果关系,实验表明其在有限观测时间步长时性能优异。

摘要:

从时间序列数据中进行因果发现旨在捕捉变量在时间链中的内在(同时)和跨(时间滞后)因果性,这对于各种科学学科至关重要。与从非时间序列数据进行因果发现相比,从时间序列数据进行因果发现需要更多的序列样本和更多的观测时间步长。为了解决这些挑战,我们提出了一种新的基于梯度的因果发现方法STIC,该方法专注于使用卷积神经网络的短期和机制不变性来揭示时间序列数据中的因果关系。具体而言,STIC利用每个窗口观测中的因果短期时间和机制不变性,这种性质具有独立性,以提高样本效率。此外,我们构建了两个因果卷积核,分别对应于短期时间和机制不变性,以估计窗口因果图。为了证明卷积神经网络在时间序列数据因果发现中的必要性,我们从假设加性噪声模型可识别的角度出发,理论推导了卷积与时间序列数据生成原理之间的等价性。在合成和FMRI基准数据集上进行的实验评估表明,我们的STIC在显著优于基线方法,并在数据集包含有限观测时间步长时实现了最先进的性能。代码可在https://github.com/HITshenrj/STIC上获得。

英文摘要:

Causal discovery from time-series data aims to capture both intra-slice (contemporaneous) and inter-slice (time-lagged) causality between variables within the temporal chain, which is crucial for various scientific disciplines. Compared to causal discovery from non-time-series data, causal discovery from time-series data necessitates more serialized samples with a larger amount of observed time steps. To address the challenges, we propose a novel gradient-based causal discovery approach STIC, which focuses on \textbf{S}hort-\textbf{T}erm \textbf{I}nvariance using \textbf{C}onvolutional neural networks to uncover the causal relationships from time-series data. Specifically, STIC leverages both the short-term time and mechanism invariance of causality within each window observation, which possesses the property of independence, to enhance sample efficiency. Furthermore, we construct two causal convolution kernels, which correspond to the short-term time and mechanism invariance respectively, to estimate the window causal graph. To demonstrate the necessity of convolutional neural networks for causal discovery from time-series data, we theoretically derive the equivalence between convolution and the underlying generative principle of time-series data under the assumption that the additive noise model is identifiable. Experimental evaluations conducted on both synthetic and FMRI benchmark datasets demonstrate that our STIC outperforms baselines significantly and achieves the state-of-the-art performance, particularly when the datasets contain a limited number of observed time steps. Code is available at \url{https://github.com/HITshenrj/STIC}.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑