Rotation-Oriented Collaborative Self-Supervised Learning for Retinal Disease Diagnosis

Xiaomeng Li,Xiaowei Hu,Xiaojuan Qi,Lequan Yu,Wei Zhao,Pheng-Ann Heng,Lei Xing
DOI: https://doi.org/10.1109/tmi.2021.3075244
IF: 10.6
2021-09-01
IEEE Transactions on Medical Imaging
Abstract:The automatic diagnosis of various conventional ophthalmic diseases from fundus images is important in clinical practice. However, developing such automatic solutions is challenging due to the requirement of a large amount of training data and the expensive annotations for medical images. This paper presents a novel self-supervised learning framework for retinal disease diagnosis to reduce the annotation efforts by learning the visual features from the unlabeled images. To achieve this, we present a rotation-oriented collaborative method that explores rotation-related and rotation-invariant features, which capture discriminative structures from fundus images and also explore the invariant property used for retinal disease classification. We evaluate the proposed method on two public benchmark datasets for retinal disease classification. The experimental results demonstrate that our method outperforms other self-supervised feature learning methods (around 4.2% area under the curve (AUC)). With a large amount of unlabeled data available, our method can surpass the supervised baseline for pathologic myopia (PM) and is very close to the supervised baseline for age-related macular degeneration (AMD), showing the potential benefit of our method in clinical practice.
engineering, biomedical,imaging science & photographic technology, electrical & electronic,computer science, interdisciplinary applications,radiology, nuclear medicine & medical imaging
What problem does this paper attempt to address?
The main problem that this paper attempts to solve is the lack of labeled data when automatically diagnosing retinal diseases. Specifically, automatically diagnosing various common ophthalmic diseases from fundus images is very important in clinical practice. However, developing such an automatic solution is challenging because a large amount of training data is required and the labeling cost of medical images is high. This paper proposes a new self - supervised learning framework, aiming to reduce the labeling workload by learning visual features from unlabeled images. ### Core problems of the paper 1. **Insufficient labeled data**: Traditional methods rely on a large amount of labeled data for training, and the labeling of medical images is not only time - consuming but also expensive. 2. **Improving diagnostic accuracy**: Although existing supervised learning methods are effective, they rely on a large amount of labeled data, while self - supervised learning can use a large amount of unlabeled data to improve model performance. ### Solutions The paper proposes a rotation - oriented collaborative self - supervised learning method, which solves the problem in the following ways: - **Rotation prediction task**: By predicting the rotation angle of the image, learn rotation - related features and capture the significant structural information in the fundus image. - **Multi - view instance discrimination task**: Learn rotation - invariant features to ensure that the images between different patients are different in the embedding space, while the images of the same patient remain similar under different transformations. - **Collaborative learning**: Combine the above two tasks, encourage the network to discover the discriminative structures in the fundus image, and explore the robust representation for retinal disease classification. ### Experimental results The authors evaluated the proposed method on two publicly available benchmark datasets. The experimental results show that this method is superior to other self - supervised feature learning methods and has achieved significant performance improvement in the classification tasks of age - related macular degeneration (AMD) and pathological myopia (PM). ### Main contributions 1. Proposed a new rotation - oriented collaborative self - supervised learning method, which can learn representative features without manual labeling. 2. Through collaborative learning tasks, simultaneously learn rotation - related and rotation - invariant features, improving the accuracy of retinal disease diagnosis. 3. The experimental results on multiple retinal disease classification tasks show that this method is superior to other state - of - the - art self - supervised methods. Through these innovations, this research provides an effective solution for the automatic diagnosis of retinal diseases, reduces the dependence on a large amount of labeled data, and shows its potential application value in clinical practice.