arXiv:2608.10969cs.LG2026-08被引 1

用图注意力模型处理重症监护数据,既准又可解释。

GARLIC: Graph Attention-based Relational Learning of Multivariate Time Series in Intensive Care

论文配图:GARLIC: Graph Attention-based Relational Learning of Multivariate Time Series in Intensive Care
图 1 · 摘自论文原文
  • 用可学习的指数衰减编码器补全缺失数据
  • 通过时滞图捕捉传感器间依赖关系,提升预测性能
  • 端到端学习注意力与图结构,提供逐层解释

重症监护室(ICU)数据是异构多变量时间序列,采样不规则且普遍存在缺失。临床应用需要高精度且可解释的预测模型。我们提出GARLIC模型,一种新型神经网络架构:通过可学习的指数衰减编码器填补缺失值,利用时滞汇总图捕捉传感器间依赖关系,并融合全局模式与跨维度序列注意力。所有注意力权重与图边均端到端学习,提供观测、信号与边级别的内置解释。为平衡辅助重建与主分类目标,设计交替解耦优化方案以稳定训练。在三个ICU基准数据集(PhysioNet 2012 & 2019,MIMIC-III)上,GARLIC在结果预测上达到新最优,显著提升AUROC和AUPRC,计算成本相当。消融实验验证各模块贡献,特征移除试验显示重要性归因的可靠性(全集 > 前50% > 随机50% > 后50%)。实时案例研究展示可操作的风险预警与透明解释,标志着对不规则采样ICU时间序列实现精准可解释深度学习的重大进展。此外,该方法在多个非ICU时间序列数据集上也表现出色,证明其泛化能力与广泛适用性。

原文摘要 · Abstract (English)

Healthcare data, such as Intensive Care Unit (ICU) records, comprise heterogeneous multivariate time series sampled at irregular intervals with pervasive missingness. However, clinical applications demand predictive models that are both accurate and interpretable. We present our Graph Attention-based Relational Learning for Intensive Care (GARLIC) model, a novel neural network architecture that imputes missing data through a learnable exponential-decay encoder, captures inter-sensor dependencies via time-lagged summary graphs, and fuses global patterns with cross-dimensional sequential attention. All attention weights and graph edges are learned end-to-end to serve as built-in observation-, signal-, and edge-level explanations. To reconcile auxiliary reconstruction and primary classification objectives, we developed an alternating decoupled optimization scheme that stabilizes training. On three ICU benchmarks (PhysioNet 2012 & 2019, MIMIC-III), GARLIC sets the new state of the art in outcome prediction, significantly improving AUROC and AUPRC over best-performing baselines at comparable computational cost. Ablation studies confirm the contribution of each module, and feature-removal trials validate the fidelity of importance attribution through a monotonic performance drop (full > top 50% > random 50% > bottom 50%). Real-time case studies demonstrate actionable risk warnings with transparent explanations, marking a significant advance toward accurate, explainable deep learning for irregularly sampled ICU time series data. Moreover, we demonstrated \proposed{}'s superiority in data imputation and classification on various time-series datasets beyond the ICU domain, showing its generalizability and applicability to broader tasks.

时间序列图神经网络医疗AI可解释性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。