Leveraging Regular Fundus Images for Training UWF Fundus Diagnosis Models via Adversarial Learning and Pseudo-Labeling

被引:33
|
作者
Ju, Lie [1 ,2 ]
Wang, Xin [2 ]
Zhao, Xin
Bonnington, Paul [3 ]
Drummond, Tom [1 ]
Ge, Zongyuan [1 ,2 ]
机构
[1] Monash Univ, Dept Elect & Comp Syst Engn, Fac Engn, Melbourne, Vic 3800, Australia
[2] Airdoc, Beijing 100000, Peoples R China
[3] Monash Univ, eRes Ctr, Melbourne, Vic 3800, Australia
关键词
Retina; Diseases; Imaging; Task analysis; Training; Generative adversarial networks; Gallium nitride; Annotation-efficient deep learning; domain adaptation; adversarial learning; ultra-widefield fundus images; VISUAL IMPAIRMENT; BLINDNESS; NETWORK;
D O I
10.1109/TMI.2021.3056395
中图分类号
TP39 [计算机的应用];
学科分类号
081203 ; 0835 ;
摘要
Recently, ultra-widefield (UWF) 200 degrees fundus imaging by Optos cameras has gradually been introduced because of its broader insights for detecting more information on the fundus than regular 30 degrees - 60 degrees fundus cameras. Compared with UWF fundus images, regular fundus images contain a large amount of high-quality and well-annotated data. Due to the domain gap, models trained by regular fundus images to recognize UWF fundus images perform poorly. Hence, given that annotating medical data is labor intensive and time consuming, in this paper, we explore how to leverage regular fundus images to improve the limited UWF fundus data and annotations for more efficient training. We propose the use of a modified cycle generative adversarial network (CycleGAN) model to bridge the gap between regular and UWF fundus and generate additional UWF fundus images for training. A consistency regularization term is proposed in the loss of the GAN to improve and regulate the quality of the generated data. Our method does not require that images from the two domains be paired or even that the semantic labels be the same, which provides great convenience for data collection. Furthermore, we show that our method is robust to noise and errors introduced by the generated unlabeled data with the pseudo-labeling technique. We evaluated the effectiveness of our methods on several common fundus diseases and tasks, such as diabetic retinopathy (DR) classification, lesion detection and tessellated fundus segmentation. The experimental results demonstrate that our proposed method simultaneously achieves superior generalizability of the learned representations and performance improvements in multiple tasks.
引用
收藏
页码:2911 / 2925
页数:15
相关论文
共 6 条
  • [1] Learning From Synthetic Images via Active Pseudo-Labeling
    Song, Liangchen
    Xu, Yonghao
    Zhang, Lefei
    Du, Bo
    Zhang, Qian
    Wang, Xinggang
    IEEE TRANSACTIONS ON IMAGE PROCESSING, 2020, 29 : 6452 - 6465
  • [2] Comparative Study of Transfer Learning Models for Retinal Disease Diagnosis from Fundus Images
    Pin, Kuntha
    Chang, Jee Ho
    Nam, Yunyoung
    CMC-COMPUTERS MATERIALS & CONTINUA, 2022, 70 (03): : 5821 - 5834
  • [3] Evaluation of glaucoma diagnosis machine learning models based on color optical coherence tomography and color fundus images
    Akiba, Masahiro
    An, Guangzhou
    Yokota, Hideo
    Omodaka, Kazuko
    Hashimoto, Kazuki
    Tsuda, Satoru
    Shiga, Yukihiro
    Takada, Naoko
    Kikawa, Tsutomu
    Nakazawa, Toru
    INVESTIGATIVE OPHTHALMOLOGY & VISUAL SCIENCE, 2019, 60 (09)
  • [4] Deep learning can generate traditional retinal fundus photographs using ultra-widefield images via generative adversarial networks
    Yoo, Tae Keun
    Ryu, Ik Hee
    Kim, Jin Kuk
    Lee, In Sik
    Kim, Jung Sub
    Kim, Hong Kyu
    Choi, Joon Yul
    COMPUTER METHODS AND PROGRAMS IN BIOMEDICINE, 2020, 197 (197)
  • [5] A Trust-Based Methodology to Evaluate Deep Learning Models for Automatic Diagnosis of Ocular Toxoplasmosis from Fundus Images
    Parra, Rodrigo
    Ojeda, Verena
    Vazquez Noguera, Jose Luis
    Garcia-Torres, Miguel
    Mello-Roman, Julio Cesar
    Villalba, Cynthia
    Facon, Jacques
    Divina, Federico
    Cardozo, Olivia
    Castillo, Veronica Elisa
    Matto, Ingrid Castro
    DIAGNOSTICS, 2021, 11 (11)
  • [6] Time-frequency supervised contrastive learning via pseudo-labeling: An unsupervised domain adaptation network for rolling bearing fault diagnosis under time-varying speeds
    Pang, Bin
    Liu, Qiuhai
    Sun, Zhenduo
    Xu, Zhenli
    Hao, Ziyang
    ADVANCED ENGINEERING INFORMATICS, 2024, 59