论文标题

学生教师课程通过加强学习学习:预测住院住院地点

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

论文作者

el-Bouri, Rasheed, Eyre, David, Watkinson, Peter, Zhu, Tingting, Clifton, David

论文摘要

由于资源构成和临床环境中的空间可用性,尤其是在与来自急诊室的患者打交道时,准确可靠的医院入院地点预测很重要。在这项工作中,我们通过加强学习来解决这个特定问题,提出一个学生教师网络。学生网络权重的表示被视为国家,并作为教师网络的输入。教师网络的行动是选择最合适的数据,以根据熵对培训组进行训练。通过在三个数据集上验证,我们不仅表明我们的方法在表格数据上的表现优于最先进的方法,并在图像识别方面竞争性能,而且还通过教师网络学习了新颖的课程。我们通过实验证明,教师网络可以积极了解学生网络,并指导其取得比单独培训的更好的表现。

Accurate and reliable prediction of hospital admission location is important due 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 to deal with this specific problem. A representation of the weights of the student network 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 to train the student network on from a training set sorted according to entropy. By validating on three datasets, not only do we show that our approach outperforms state-of-the-art methods on tabular data and performs competitively on image recognition, but also that novel curricula are learned by the teacher network. We demonstrate experimentally that the teacher network can actively learn about the student network and guide it to achieve better performance than if trained alone.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源