arXiv:2601.14228cs.LG2026-01ICML

用可解释框架提升脓毒症治疗决策,融合聚类、生成数据与强化学习。

Attention-Based Offline Reinforcement Learning and Clustering for Interpretable Sepsis Treatment

  • 按风险分层患者,用生成模型补足稀疏治疗轨迹。
  • 离线强化学习实现安全精准的治疗推荐,准确率高。
  • 大模型生成临床语言解释,适合医生信任和使用。

脓毒症是重症监护病房的主要致死原因,及时准确的治疗决策对预后至关重要。本文提出一个可解释的决策支持框架,包含四个核心模块:(1) 基于聚类的分层模块,在入院时将患者分为低、中、高风险三组,采用统计验证;(2) 利用变分自编码器(VAE)和扩散模型的合成数据增强管道,丰富如液体或血管活性药物使用等稀疏轨迹;(3) 采用优势加权回归(AWR)训练的离线强化学习(RL)代理,配备轻量级注意力编码器,并通过集成模型提供保守且安全的治疗建议;(4) 多模态大语言模型驱动的推理生成模块,基于临床背景和专家知识生成自然语言解释。在MIMIC-III和eICU数据集上评估,本方法在保持高治疗准确率的同时,为临床医生提供可解释且稳健的策略推荐。

原文摘要 · Abstract (English)

Sepsis remains one of the leading causes of mortality in intensive care units, where timely and accurate treatment decisions can significantly impact patient outcomes. In this work, we propose an interpretable decision support framework. Our system integrates four core components: (1) a clustering-based stratification module that categorizes patients into low, intermediate, and high-risk groups upon ICU admission, using clustering with statistical validation; (2) a synthetic data augmentation pipeline leveraging variational autoencoders (VAE) and diffusion models to enrich underrepresented trajectories such as fluid or vasopressor administration; (3) an offline reinforcement learning (RL) agent trained using Advantage Weighted Regression (AWR) with a lightweight attention encoder and supported by an ensemble models for conservative, safety-aware treatment recommendations; and (4) a rationale generation module powered by a multi-modal large language model (LLM), which produces natural-language justifications grounded in clinical context and retrieved expert knowledge. Evaluated on the MIMIC-III and eICU datasets, our approach achieves high treatment accuracy while providing clinicians with interpretable and robust policy recommendations.

脓毒症治疗强化学习可解释性生成模型

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