Deep Inverse Reinforcement Learning for Sepsis Treatment

By Chao Yu, Guoqi Ren, Jiming Liu. 2019

49 citations recorded by the source corpus.

Sepsis is a leading cause of mortality in hospitals, but its optimal treatment strategy still remains unclear. Recent years have witnessed several successful applications of Reinforcement Learning (RL) approaches in sepsis treatment, achieving far more efficient strategies than those by clinicians. To ensure such applications, an explicit reward function encoding medical domain knowledge should be specified beforehand to indicate the goal of learning. However, due to the paucity of clear understanding of sepsis itself, there is still considerable inconsistency in the formulation of reward functions for sepsis treatment. In this poster, we address the reward learning problem in RL for treatment of sepsis, which has been largely neglected by previous studies. A deep inverse RL with Mini-Tree (DIRL-MT) model is proposed to infer the best reward functions from a set of presumably optimal treatment trajectories using retrospective real medical data. In the model, the MT component learns the factors that are most important in influencing the mortality during sepsis treatment, while the DIRL component infers the complete reward function in terms of weights of those factors. Our work shows that PaO2 and PT can play a vital role and should be paid more attention in the design of more efficient treatment strategies for sepsis in the future.

Subfields: Artificial Intelligence, Cardiology and Cardiovascular Medicine, Epidemiology

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Sergey Levine
  2. Sham M. Kakade
  3. Ruslan Salakhutdinov
  4. Aravind Rajeswaran
  5. Ruosong Wang
  6. Chelsea Finn
  7. Praneeth Netrapalli
  8. Rahul Kidambi

In Cardiology and Cardiovascular Medicine

  1. Sham M. Kakade
  2. Ruosong Wang
  3. Shamim Nemati
  4. Dean Phillips Foster
  5. Chao Yu
  6. Jiming Liu

In Epidemiology

  1. Chao Yu
  2. Guoqi Ren
  3. Jiming Liu
  4. Shamim Nemati
  5. Sham M. Kakade
  6. Ruosong Wang
  7. Yinzhao Dong
  8. Dean Phillips Foster

Citations in the corpus