Deep Attention Q-Network for Personalized Treatment Recommendation
Not on the map. Why?
Abstract (arXiv)
Tailoring treatment for individual patients is crucial yet challenging in order to achieve optimal healthcare outcomes. Recent advances in reinforcement learning offer promising personalized treatment recommendations; however, they rely solely on current patient observations (vital signs, demographics) as the patient's state, which may not accurately represent the true health status of the patient. This limitation hampers policy learning and evaluation, ultimately limiting treatment effectiveness. In this study, we propose the Deep Attention Q-Network for personalized treatment recommendations, utilizing the Transformer architecture within a deep reinforcement learning framework to efficiently incorporate all past patient observations. We evaluated the model on real-world sepsis and acute hypotension cohorts, demonstrating its superiority to state-of-the-art models. The source code for our model is available at https://github.com/stevenmsm/RL-ICU-DAQN.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 29 of its 64 references that are in the map. First posted on arXiv 4 Jul 2023.
Builds on
29 of 64 references.
References · 29
- 2017Attention Is All You Needcited by 60,361
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2016Layer Normalizationcited by 5,669
- 2014Neural Machine Translation by Jointly Learning to Align and Translatecited by 4,548
- 2015Continuous control with deep reinforcement learningcited by 3,407
- 2019XLNet: Generalized Autoregressive Pretraining for Language Understandingcited by 2,811
- 2015Deep Reinforcement Learning with Double Q-learningcited by 1,745
- 2021Autoformer: Decomposition Transformers with Auto-Correlation for Long-Term Series Forecastingcited by 1,676
- 2021Decision Transformer: Reinforcement Learning via Sequence Modelingcited by 1,546
- 2015Prioritized Experience Replaycited by 1,099
- 2015Dueling Network Architectures for Deep Reinforcement Learningcited by 785
- 2015Deep Recurrent Q-Learning for Partially Observable MDPscited by 465
17 more
- 2015Doubly Robust Off-policy Value Evaluation for Reinforcement Learningcited by 339
- 2016Data-Efficient Off-Policy Policy Evaluation for Reinforcement Learningcited by 286
- 2015Doctor AI: Predicting Clinical Events via Recurrent Neural Networkscited by 208
- 2018Deep Variational Reinforcement Learning for POMDPscited by 125
- 2018Supervised Reinforcement Learning with Recurrent Neural Network for Dynamic Treatment Recommendationcited by 101
- 2017Deep Reinforcement Learning for Sepsis Treatmentcited by 71
- 2017Continuous State-Space Models for Optimal Sepsis Treatment - a Deep Reinforcement Learning Approachcited by 68
- 2018Evaluating Reinforcement Learning Algorithms in Observational Health Settingscited by 60
- 2017A Reinforcement Learning Approach to Weaning of Mechanical Ventilation in Intensive Care Unitscited by 50
- 2020Interpretable Off-Policy Evaluation in Reinforcement Learning by Highlighting Influential Transitionscited by 41
- 2022Deep Transformer Q-Networks for Partially Observable Reinforcement Learningcited by 36
- 2019Improving Sepsis Treatment Strategies by Combining Deep and Kernel-Based Reinforcement Learningcited by 34
- 2018Model-Based Reinforcement Learning for Sepsis Treatmentcited by 21
- 2017Representation and Reinforcement Learning for Personalized Glycemic Control in Septic Patientscited by 13
- 2019Optimizing Sequential Medical Treatments with Auto-Encoding Heuristic Search in POMDPscited by 12
- 2018TextRay: Mining Clinical Reports to Gain a Broad Understanding of Chest X-rayscited by 8
- 2019Deep Reinforcement Learning for Optimal Critical Care Pain Management with Morphine using Dueling Double-Deep Q Networkscited by 8
Built on by
Not cited yet.