arXiv:2508.17554cs.LG2025-08中稿 · NeurIPS被引 1

融合图模型与状态空间,提升重症患者住院时长预测精度

Bridging Graph and State-Space Modeling for Intensive Care Unit Length of Stay Prediction

  • 用状态空间模型捕捉时间序列轨迹,图神经网络整合多源患者相似性
  • 在MIMIC-IV数据集上优于主流序列和图模型,各指标均领先
  • 适合临床决策支持系统,尤其关注多模态医疗数据建模的研究者

预测重症监护室(ICU)住院时长对医院资源管理至关重要,但因电子健康记录(EHR)的异质性和非规则采样特性而极具挑战。本文提出S²G-Net,一种将状态空间序列建模与多视角图神经网络(GNN)统一的新架构。时间路径采用Mamba状态空间模型(SSMs)捕捉患者轨迹,图路径则使用优化的GraphGPS主干网络,整合来自诊断、行政及语义特征的异质患者相似性图。在大规模MIMIC-IV队列数据集上的实验表明,S²G-Net在所有主要指标上持续优于序列模型(BiLSTM、Mamba、Transformer)、图模型(经典GNN、GraphGPS)及混合方法。大量消融实验与可解释性分析揭示了各组件的互补贡献,并强调了合理构建图的重要性。结果表明,S²G-Net为多模态临床数据下的ICU住院时长预测提供了高效且可扩展的解决方案。

原文摘要 · Abstract (English)

Predicting a patient's length of stay (LOS) in the intensive care unit (ICU) is a critical task for hospital resource management, yet remains challenging due to the heterogeneous and irregularly sampled nature of electronic health records (EHRs). In this work, we propose S$^2$G-Net, a novel neural architecture that unifies state-space sequence modeling with multi-view Graph Neural Networks (GNNs) for ICU LOS prediction. The temporal path employs Mamba state-space models (SSMs) to capture patient trajectories, while the graph path leverages an optimized GraphGPS backbone, designed to integrate heterogeneous patient similarity graphs derived from diagnostic, administrative, and semantic features. Experiments on the large-scale MIMIC-IV cohort dataset show that S$^2$G-Net consistently outperforms sequence models (BiLSTM, Mamba, Transformer), graph models (classic GNNs, GraphGPS), and hybrid approaches across all primary metrics. Extensive ablation studies and interpretability analyses highlight the complementary contributions of each component of our architecture and underscore the importance of principled graph construction. These results demonstrate that S$^2$G-Net provides an effective and scalable solution for ICU LOS prediction with multi-modal clinical data. The code can be found at https://github.com/ShuqiZi1/S2G-Net.

ICU预测图神经网络状态空间模型多模态建模

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