Self-Adaptive Transfer Learning for Multicenter Glaucoma Classification in Fundus Retina Images

Yiming Bao,Jun Wang,Tong Li,Linyan Wang,Jianwei Xu,Juan Ye,Dahong Qian
DOI: https://doi.org/10.1007/978-3-030-87000-3_14
2021-01-01
Abstract:The early screening of glaucoma is important for patients to receive treatment in time and maintain eyesight. Deep learning (DL) based models have been successfully used for computer-aided diagnosis (CAD) of glaucoma. However, a DL model pre-trained on certain dataset from one hospital may have poor performance on other hospital data, therefore its applications in the real scene are limited. In this paper, we propose a self-adaptive transfer learning (SATL) strategy to fill the domain gap between multi-center datasets. Specifically, the encoder of a DL model that is pre-trained on the source domain is used to initialize the encoder of a reconstruction model. Then, this reconstruction model is trained using only unlabeled image data from the target domain, which makes the encoder in the model adapt itself to extract useful features both for target domain images encoding and glaucoma classification, simultaneously. Experimental results on a private and two public glaucoma diagnosis datasets demonstrate that the proposed SATL strategy is effective. Also, it meets the real scene application and the privacy protection policy due to its independence from the source domain data.
What problem does this paper attempt to address?