论文标题

通过教师学习利用未诊断的数据进行青光眼分类

Leveraging Undiagnosed Data for Glaucoma Classification with Teacher-Student Learning

论文作者

Wu, Junde, Yu, Shuang, Chen, Wenting, Ma, Kai, Fu, Rao, Liu, Hanruo, Di, Xiaoguang, Zheng, Yefeng

论文摘要

最近,深度学习已被用于青光眼分类任务,其性能与人类专家的表现相当。但是,训练有素的深度学习模型需要大量适当标记的数据,这相对昂贵,因为青光眼的准确标记需要多年的专业培训。为了减轻此问题,我们提出了一个青光眼分类框架,该框架不仅利用了正确标记的图像,而且还利用没有青光眼标签的未诊断图像。更具体地说,拟议的框架是根据教师学习范式改编的。教师模型将未诊断的图像的包裹信息编码为潜在的特征空间,同时,学生模型通过知识转移向教师学习,以改善青光眼分类。对于模型培训程序,我们提出了一种新颖的培训策略,该策略模拟了现实世界中的教学实践,称为“学习以知识转移(L2T-KT)的教学”,并建立一个“测验池”作为教师的优化目标。实验表明,所提出的框架能够有效地利用未诊断的数据来改善青光眼预测性能。

Recently, deep learning has been adopted to the glaucoma classification task with performance comparable to that of human experts. However, a well trained deep learning model demands a large quantity of properly labeled data, which is relatively expensive since the accurate labeling of glaucoma requires years of specialist training. In order to alleviate this problem, we propose a glaucoma classification framework which takes advantage of not only the properly labeled images, but also undiagnosed images without glaucoma labels. To be more specific, the proposed framework is adapted from the teacher-student-learning paradigm. The teacher model encodes the wrapped information of undiagnosed images to a latent feature space, meanwhile the student model learns from the teacher through knowledge transfer to improve the glaucoma classification. For the model training procedure, we propose a novel training strategy that simulates the real-world teaching practice named as 'Learning To Teach with Knowledge Transfer (L2T-KT)', and establish a 'Quiz Pool' as the teacher's optimization target. Experiments show that the proposed framework is able to utilize the undiagnosed data effectively to improve the glaucoma prediction performance.

扫码加入交流群

加入微信交流群

微信交流群二维码

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