论文标题
与政策适应动态治疗方案的反兴演员批评网络
Deconfounding Actor-Critic Network with Policy Adaptation for Dynamic Treatment Regimes
论文作者
论文摘要
尽管在基础研究和临床研究方面做出了巨大的努力,但针对危重患者的个性化通风策略仍然是一个重大挑战。最近,通过电子健康记录(EHR)进行增强学习(RL)的动态治疗制度(DTR)引起了医疗保健行业和机器学习研究社区的兴趣。但是,由于存在混杂因素的存在,大多数博学的DTR政策可能会偏见。尽管某些治疗措施非幸存者可能会有所帮助,但如果混淆者导致死亡率,则以长期结局(例如90天死亡率)指导的RL模型训练将惩罚这些治疗措施,从而导致学习的DTR政策是次优的。在这项研究中,我们开发了一个新的变形参与者批评网络(DAC),以学习患者的最佳DTR政策。为了减轻混杂问题,我们将患者重新采样模块和混杂的平衡模块纳入了我们的参与者批评框架。为了避免惩罚非幸存者的有效治疗措施,我们设计了短期奖励,以捕捉患者的直接健康状况变化。将短期与长期奖励相结合可以进一步提高模型性能。此外,我们介绍了一种策略适应方法,以成功地将学习的模型转移到新来源的小规模数据集中。一个半合成和两个不同的现实世界数据集的实验结果表明,所提出的模型优于最先进的模型。提出的模型为机械通气提供了个性化的治疗决策,可以改善患者的预后。
Despite intense efforts in basic and clinical research, an individualized ventilation strategy for critically ill patients remains a major challenge. Recently, dynamic treatment regime (DTR) with reinforcement learning (RL) on electronic health records (EHR) has attracted interest from both the healthcare industry and machine learning research community. However, most learned DTR policies might be biased due to the existence of confounders. Although some treatment actions non-survivors received may be helpful, if confounders cause the mortality, the training of RL models guided by long-term outcomes (e.g., 90-day mortality) would punish those treatment actions causing the learned DTR policies to be suboptimal. In this study, we develop a new deconfounding actor-critic network (DAC) to learn optimal DTR policies for patients. To alleviate confounding issues, we incorporate a patient resampling module and a confounding balance module into our actor-critic framework. To avoid punishing the effective treatment actions non-survivors received, we design a short-term reward to capture patients' immediate health state changes. Combining short-term with long-term rewards could further improve the model performance. Moreover, we introduce a policy adaptation method to successfully transfer the learned model to new-source small-scale datasets. The experimental results on one semi-synthetic and two different real-world datasets show the proposed model outperforms the state-of-the-art models. The proposed model provides individualized treatment decisions for mechanical ventilation that could improve patient outcomes.