论文标题
通过知识蒸馏和多任务学习,用于评估视网膜疾病的协同对抗标签学习
Synergic Adversarial Label Learning for Grading Retinal Diseases via Knowledge Distillation and Multi-task Learning
论文作者
论文摘要
长期以来,人们已经认识到需要进行视网膜图像分类的全面和自动筛查方法的需求。合格的医生注释的图像非常昂贵,只有有限的数据可用于各种视网膜疾病,例如与年龄相关的黄斑变性(AMD)和糖尿病性视网膜病变(DR)。一些研究表明,AMD和DR具有一些共同的特征,例如出血点和渗出率,但大多数分类算法仅独立训练这些疾病模型。受知识蒸馏的启发,来自各种来源的其他监视信号对培训具有更少数据的强大模型是有益的。我们提出了一种称为Synergic对抗标签学习(SALL)的方法,该方法利用语义和特征空间中相关的视网膜疾病标签作为其他信号,并以协作方式训练该模型。我们对DR和AMD底面图像分类任务的实验表明,所提出的方法可以显着提高模型的分级疾病的准确性。此外,我们进行了其他实验,以在医学成像应用中的可靠性和可解释性方面显示SALL的有效性。
The need for comprehensive and automated screening methods for retinal image classification has long been recognized. Well-qualified doctors annotated images are very expensive and only a limited amount of data is available for various retinal diseases such as age-related macular degeneration (AMD) and diabetic retinopathy (DR). Some studies show that AMD and DR share some common features like hemorrhagic points and exudation but most classification algorithms only train those disease models independently. Inspired by knowledge distillation where additional monitoring signals from various sources is beneficial to train a robust model with much fewer data. We propose a method called synergic adversarial label learning (SALL) which leverages relevant retinal disease labels in both semantic and feature space as additional signals and train the model in a collaborative manner. Our experiments on DR and AMD fundus image classification task demonstrate that the proposed method can significantly improve the accuracy of the model for grading diseases. In addition, we conduct additional experiments to show the effectiveness of SALL from the aspects of reliability and interpretability in the context of medical imaging application.