用图神经网络建模心脏运动,实现精准重建与未来预测
Spatiotemporal graph neural process for reconstruction, extrapolation, and classification of cardiac trajectories
- 将心脏运动建模为时空多层图,结合神经微分方程与神经过程
- 仅凭单个周期观测即可准确重建并外推后续心脏周期
- 在真实心脏数据集上实现99%疾病分类准确率,适合医学时序分析
我们提出一种概率框架,用于从稀疏观测中建模结构化时空动态,聚焦心脏运动。方法融合神经微分方程(NODEs)、图神经网络(GNNs)和神经过程,统一建模不确定性、时间连续性与解剖结构。将动态系统表示为时空多层图,使用GNN参数化向量场建模潜在轨迹。基于节点与边级别的稀疏上下文观测,模型推断潜在初始状态与控制变量的分布,支持轨迹插值与外推。在三个合成系统(耦合摆、洛伦兹吸引子、库拉莫托振子)及两个真实心脏影像数据集(ACDC,N=150;UK Biobank,N=526)上验证。模型能准确重构轨迹,并从单个周期外推未来心脏周期。在ACDC分类任务上达到最高99%准确率,在UK Biobank中检测心房颤动达67%准确率,表现领先。本工作为心脏运动分析提供灵活范式,奠定生物医学时空序列图学习基础。
原文摘要 · Abstract (English)
We present a probabilistic framework for modeling structured spatiotemporal dynamics from sparse observations, focusing on cardiac motion. Our approach integrates neural ordinary differential equations (NODEs), graph neural networks (GNNs), and neural processes into a unified model that captures uncertainty, temporal continuity, and anatomical structure. We represent dynamic systems as spatiotemporal multiplex graphs and model their latent trajectories using a GNN-parameterized vector field. Given the sparse context observations at node and edge levels, the model infers a distribution over latent initial states and control variables, enabling both interpolation and extrapolation of trajectories. We validate the method on three synthetic dynamical systems (coupled pendulum, Lorenz attractor, and Kuramoto oscillators) and two real-world cardiac imaging datasets - ACDC (N=150) and UK Biobank (N=526) - demonstrating accurate reconstruction, extrapolation, and disease classification capabilities. The model accurately reconstructs trajectories and extrapolates future cardiac cycles from a single observed cycle. It achieves state-of-the-art results on the ACDC classification task (up to 99% accuracy), and detects atrial fibrillation in UK Biobank subjects with competitive performance (up to 67% accuracy). This work introduces a flexible approach for analyzing cardiac motion and offers a foundation for graph-based learning in structured biomedical spatiotemporal time-series data.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。