arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2411.05331cs.LGstat.ML

从时空数据中发现潜在因果图

Discovering Latent Causal Graphs from Spatiotemporal Data

Department of Computer Science and Engineering, University of California, San Diego, La Jolla, USA(计算机科学与工程系,加州大学圣地亚哥分校,拉乔拉,美国);Halıcıoglu Data Science Institute, University of California, San Diego, La Jolla, USA(Halıcıoglu数据科学研究所,加州大学圣地亚哥分校,拉乔拉,美国);Scripps Institution of Oceanography, University of California, San Diego, La Jolla, USA(斯克里普斯海洋研究所,加州大学圣地亚哥分校,拉乔拉,美国)

Kun Wang, Sumanth Varambally, Duncan Watson-Parris, Yi-An Ma, Rose Yu

AI总结:

针对时空网格数据高维、空间邻近点相关导致因果推断难的问题,提出基于变分推断的SPACY框架,在潜在空间发现因果结构,理论上证明可识别性,实验上优于基线且能识别真实气候数据中的关键现象。

摘要:

气候、神经科学和流行病学等科学领域中的许多重要现象,天然以具有复杂相互作用的时空网格数据形式呈现。从这类数据中推断因果关系是一个具有挑战性的问题,而数据的高维度以及空间邻近点之间的相关性进一步加剧了这一难题。我们提出了SPACY(时空因果发现),这是一种基于变分推断的新型框架,旨在从时空数据中建模潜在时间序列及其因果关系。SPACY通过在潜在空间中发现因果结构来缓解高维挑战。为了聚合空间邻近、具有相关性的网格点,我们使用由空间核函数参数化的空间因子,将观测时间序列映射到潜在表示。理论上,我们将该问题推广到连续空间域,并证明了当观测值由潜在序列与空间因子乘积的非线性可逆函数生成时的可识别性。通过这种方法,我们避免了通常无法验证的假设,包括关于瞬时效应或充分变异性的假设。实验上,SPACY在合成数据上优于现有最优基线,即使在现有方法难以应对的挑战性设置中也是如此,同时还能扩展到大型网格。SPACY还能从真实气候数据中识别出已知的关键现象。SPACY的实现可在https://github.com/Rose-STL-Lab/SPACY/获取。

英文摘要:

Many important phenomena in scientific fields like climate, neuroscience, and epidemiology are naturally represented as spatiotemporal gridded data with complex interactions. Inferring causal relationships from these data is a challenging problem compounded by the high dimensionality of such data and the correlations between spatially proximate points. We present SPACY (SPAtiotemporal Causal discoverY), a novel framework based on variational inference, designed to model latent time series and their causal relationships from spatiotemporal data. SPACY alleviates the high-dimensional challenge by discovering causal structures in the latent space. To aggregate spatially proximate, correlated grid points, we use spatial factors, parametrized by spatial kernel functions, to map observational time series to latent representations. Theoretically, we generalize the problem to a continuous spatial domain and establish identifiability when the observations arise from a nonlinear, invertible function of the product of latent series and spatial factors. Using this approach, we avoid assumptions that are often unverifiable, including those about instantaneous effects or sufficient variability. Empirically, SPACY outperforms state-of-the-art baselines on synthetic data, even in challenging settings where existing methods struggle, while remaining scalable for large grids. SPACY also identifies key known phenomena from real-world climate data. An implementation of SPACY is available at https://github.com/Rose-STL-Lab/SPACY/

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

↑