Diffusion model for relational inference

Shuhan Zheng,Ziqiang Li,Kantaro Fujiwara,Gouhei Tanaka
2024-06-20
Abstract:Dynamical behaviors of complex interacting systems, including brain activities, financial price movements, and physical collective phenomena, are associated with underlying interactions between the system's components. The issue of uncovering interaction relations in such systems using observable dynamics is called relational inference. In this study, we propose a Diffusion model for Relational Inference (DiffRI), inspired by a self-supervised method for probabilistic time series imputation. DiffRI learns to infer the probability of the presence of connections between components through conditional diffusion modeling.
Machine Learning,Artificial Intelligence
What problem does this paper attempt to address?
The paper is primarily dedicated to addressing the problem of inferring relationships between components from time series data, known as the relational inference problem. Specifically, the research focuses on uncovering the underlying interactions in complex systems (such as brain activity, financial market fluctuations, and physical collective phenomena) through observed time series behaviors. The paper proposes a novel method called the Diffusion model for Relational Inference (DiffRI), which is inspired by self-supervised time series imputation methods and is based on diffusion models. The main contributions of DiffRI include: 1. **Innovative use of diffusion models for relational inference**: To the best of the authors' knowledge, this is the first attempt to use diffusion-based generative modeling methods for neural network-driven relational inference tasks. 2. **High-precision relational inference capability**: DiffRI can accurately identify relationships in multiple time series datasets, is applicable to different interaction mechanisms, and achieves good results without the need for additional data augmentation techniques. 3. **Robustness to missing data**: The imputation training method adopted by DiffRI is more suitable for handling datasets with missing data compared to prediction training methods, thus exhibiting better robustness for time series data containing missing values. The experimental section presents comparative results of DiffRI with several well-known existing methods on different simulated datasets, including the Kuramoto model, spring system, and vector autoregression (VAR) model. Additionally, experiments were conducted on the Netsim dataset, which simulates human brain functional magnetic resonance imaging (fMRI) signals. These experiments demonstrate the effectiveness and superiority of DiffRI in relational inference tasks. In summary, this paper aims to solve the relational inference problem by proposing a new method based on diffusion models, particularly in the context of time series data, showcasing the advantages and effectiveness of this approach.