Bhaskar Dhananjay, Magruder Daniel Sumner, Morales Matheo, De Brouwer Edward, Venkat Aarthi, Wenkel Frederik, Noonan James, Wolf Guy, Ivanova Natalia, Krishnaswamy Smita
Department of Genetics, Yale School of Medicine.
Department of Computer Science, Yale University.
Proc Mach Learn Res. 2024;231.
Complex systems are characterized by intricate interactions between entities that evolve dynamically over time. Accurate inference of these dynamic relationships is crucial for understanding and predicting system behavior. In this paper, we propose Regulatory Temporal Interaction Network Inference (RiTINI) for inferring time-varying interaction graphs in complex systems using a novel combination of space-and-time graph attentions and graph neural ordinary differential equations (ODEs). RiTINI leverages time-lapse signals on a graph prior, as well as perturbations of signals at various nodes in order to effectively capture the dynamics of the underlying system. This approach is distinct from traditional causal inference networks, which are limited to inferring acyclic and static graphs. In contrast, RiTINI can infer cyclic, directed, and time-varying graphs, providing a more comprehensive and accurate representation of complex systems. The graph attention mechanism in RiTINI allows the model to adaptively focus on the most relevant interactions in time and space, while the graph neural ODEs enable continuous-time modeling of the system's dynamics. We evaluate RiTINI's performance on simulations of dynamical systems, neuronal networks, and gene regulatory networks, demonstrating its state-of-the-art capability in inferring interaction graphs compared to previous methods.
复杂系统的特征是实体之间存在错综复杂的相互作用,且这些相互作用会随时间动态演变。准确推断这些动态关系对于理解和预测系统行为至关重要。在本文中,我们提出了调控时间交互网络推理(RiTINI)方法,用于使用时空图注意力和图神经常微分方程(ODE)的新颖组合来推断复杂系统中的时变交互图。RiTINI利用图先验上的延时信号以及各个节点处信号的扰动,以便有效捕捉底层系统的动态特性。这种方法不同于传统的因果推理网络,传统因果推理网络仅限于推断无环和静态图。相比之下,RiTINI可以推断循环、有向和时变图,从而为复杂系统提供更全面、准确的表示。RiTINI中的图注意力机制使模型能够在时间和空间上自适应地聚焦于最相关的交互,而图神经ODE则能够对系统动态进行连续时间建模。我们在动态系统、神经网络和基因调控网络的模拟中评估了RiTINI的性能,结果表明与先前方法相比,它在推断交互图方面具有领先的能力。