Student-Teacher Curriculum Learning via Reinforcement Learning: Predicting Hospital Inpatient Admission Location

  • 2020-07-01 15:00:43
  • Rasheed el-Bouri, David Eyre, Peter Watkinson, Tingting Zhu, David Clifton
  • 3

Abstract

Accurate and reliable prediction of hospital admission location is importantdue to resource-constraints and space availability in a clinical setting,particularly when dealing with patients who come from the emergency department.In this work we propose a student-teacher network via reinforcement learning todeal with this specific problem. A representation of the weights of the studentnetwork is treated as the state and is fed as an input to the teacher network.The teacher network's action is to select the most appropriate batch of data totrain the student network on from a training set sorted according to entropy.By validating on three datasets, not only do we show that our approachoutperforms state-of-the-art methods on tabular data and performs competitivelyon image recognition, but also that novel curricula are learned by the teachernetwork. We demonstrate experimentally that the teacher network can activelylearn about the student network and guide it to achieve better performance thanif trained alone.

 

Quick Read (beta)

loading the full paper ...