在重症监护中用多目标强化学习实现可调的个性化决策
Benchmarking Offline Multi-Objective Reinforcement Learning in Critical Care
- 基于历史数据离线训练多目标强化学习模型
- PEDA DT 在多目标权衡上表现最优,优于单目标基线
- 适合需要动态调整临床优先级的医疗决策场景
在重症监护等关键医疗场景中,临床医生需在提高患者存活率与减少资源消耗(如住院时长)之间权衡。传统单目标强化学习通过固定加权奖励优化策略,导致政策僵化,无法适应不同临床需求。多目标强化学习(MORL)通过学习帕累托前沿上的多个最优策略,可在测试时动态选择偏好。本文在MIMIC-IV数据集上对比了三种离线MORL算法(CPQL、Adaptive CPQL、PEDA DT)与三种单目标基线(BC、CQL、DDQN)。通过离线策略评估(OPE)发现,PEDA DT展现出更强灵活性。结果表明,序列建模架构在多目标条件生成中依然有效,支持无需重训即可实现个性化、可调的医疗决策。该研究验证了离线MORL在重症护理中的潜力。
原文摘要 · Abstract (English)
In critical care settings such as the Intensive Care Unit, clinicians face the complex challenge of balancing conflicting objectives, primarily maximizing patient survival while minimizing resource utilization (e.g., length of stay). Single-objective Reinforcement Learning approaches typically address this by optimizing a fixed scalarized reward function, resulting in rigid policies that fail to adapt to varying clinical priorities. Multi-objective Reinforcement Learning (MORL) offers a solution by learning a set of optimal policies along the Pareto Frontier, allowing for dynamic preference selection at test time. However, applying MORL in healthcare necessitates strict offline learning from historical data. In this paper, we benchmark three offline MORL algorithms, Conditioned Conservative Pareto Q-Learning (CPQL), Adaptive CPQL, and a modified Pareto Efficient Decision Agent (PEDA) Decision Transformer (PEDA DT), against three scalarized single-objective baselines (BC, CQL, and DDQN) on the MIMIC-IV dataset. Using Off-Policy Evaluation (OPE) metrics, we demonstrate that PEDA DT algorithm offers superior flexibility compared to static scalarized baselines. Notably, our results extend previous findings on single-objective Decision Transformers in healthcare, confirming that sequence modeling architectures remain robust and effective when scaled to multi-objective conditioned generation. These findings suggest that offline MORL is a promising framework for enabling personalized, adjustable decision-making in critical care without the need for retraining.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。