用图神经网络处理纵向数据缺失值,支持任意缺失模式且高效可扩展。
Sampling-guided Heterogeneous Graph Neural Network with Temporal Smoothing for Scalable Longitudinal Data Imputation
- 构建异构图模型,将观测与协变量分作节点类型,时间序列连接成子网络。
- 在ADNI数据集上,即使缺失率达50%仍显著优于现有方法。
- 适合大规模纵向研究中的缺失数据填补,如医学追踪分析。
本文提出一种新型框架——采样引导的异构图神经网络(SHT-GNN),用于解决纵向研究中缺失数据插补问题。与传统方法需大量预处理不同,SHT-GNN能直接处理任意缺失模式,同时保持计算效率。该模型将观测值和协变量分别建模为不同类型的节点,通过受试者特有的纵向子网连接连续时间点的观测节点,协变量与观测间的交互则以二部图中的带属性边表示。借助受试者级小批量采样与多层时间平滑机制,SHT-GNN可高效扩展至大规模数据集,有效学习节点表示并完成缺失数据插补。在合成数据及真实世界数据集(包括阿尔茨海默病神经影像计划,ADNI)上的大量实验表明,即便在高达50%的缺失率下,SHT-GNN仍显著优于现有插补方法。实证结果凸显其在复杂、大规模纵向数据中的强大插补能力与卓越性能。
原文摘要 · Abstract (English)
In this paper, we propose a novel framework, the Sampling-guided Heterogeneous Graph Neural Network (SHT-GNN), to effectively tackle the challenge of missing data imputation in longitudinal studies. Unlike traditional methods, which often require extensive preprocessing to handle irregular or inconsistent missing data, our approach accommodates arbitrary missing data patterns while maintaining computational efficiency. SHT-GNN models both observations and covariates as distinct node types, connecting observation nodes at successive time points through subject-specific longitudinal subnetworks, while covariate-observation interactions are represented by attributed edges within bipartite graphs. By leveraging subject-wise mini-batch sampling and a multi-layer temporal smoothing mechanism, SHT-GNN efficiently scales to large datasets, while effectively learning node representations and imputing missing data. Extensive experiments on both synthetic and real-world datasets, including the Alzheimer's Disease Neuroimaging Initiative (ADNI) dataset, demonstrate that SHT-GNN significantly outperforms existing imputation methods, even with high missing data rates. The empirical results highlight SHT-GNN's robust imputation capabilities and superior performance, particularly in the context of complex, large-scale longitudinal data.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。