VSG-GAN: A high-fidelity image synthesis method with semantic manipulation in retinal fundus image

被引:5
作者
Liu, Junjie [2 ,3 ,4 ,5 ]
Xu, Shixin [6 ]
He, Ping [2 ,3 ]
Wu, Sirong [2 ,3 ,4 ]
Luo, Xi [2 ,3 ,4 ]
Deng, Yuhui [2 ,3 ]
Huang, Huaxiong [1 ,2 ,7 ]
机构
[1] Beijing Normal Univ, Res Ctr Math, Zhuhai, Peoples R China
[2] Guangdong Prov Key Lab Interdisciplinary Res & App, Zhuhai, Peoples R China
[3] BNU HKBU United Int Coll, Zhuhai, Peoples R China
[4] Hong Kong Baptist Univ, Fac Sci, Hong Kong, Peoples R China
[5] Trinity Coll Dublin, Dublin, Ireland
[6] Duke Kunshan Univ, Data Sci Res Ctr, Kunshan, Jiangsu, Peoples R China
[7] York Univ, Dept Math & Stat, Toronto, ON, Canada
基金
欧洲研究理事会; 中国国家自然科学基金;
关键词
SEGMENTATION;
D O I
10.1016/j.bpj.2024.02.019
中图分类号
Q6 [生物物理学];
学科分类号
071011 ;
摘要
In recent years, advancements in retinal image analysis, driven by machine learning and deep learning techniques, have enhanced disease detection and diagnosis through automated feature extraction. However, challenges persist, including limited data set diversity due to privacy concerns and imbalanced sample pairs, hindering effective model training. To address these issues, we introduce the vessel and style guided generative adversarial network (VSG-GAN), an innovative algorithm building upon the foundational concept of GAN. In VSG-GAN, a generator and discriminator engage in an adversarial process to produce realistic retinal images. Our approach decouples retinal image generation into distinct modules: the vascular skeleton and background style. Leveraging style transformation and GAN inversion, our proposed hierarchical variational autoencoder module generates retinal images with diverse morphological traits. In addition, the spatially adaptive denormalization module ensures consistency between input and generated images. We evaluate our model on MESSIDOR and RITE data sets using various metrics, including structural similarity index measure, inception score, Frechet inception distance, and kernel inception distance. Our results demonstrate the superiority of VSG-GAN, outperforming existing methods across all evaluation assessments. This underscores its effectiveness in addressing data set limitations and imbalances. Our algorithm provides a novel solution to challenges in retinal image analysis by offering diverse and realistic retinal image generation. Implementing the VSG-GAN augmentation approach on downstream diabetic retinopathy classification tasks has shown enhanced disease diagnosis accuracy, further advancing the utility of machine learning in this domain.
引用
收藏
页码:2815 / 2829
页数:15
相关论文
共 55 条
[1]   The Medical Segmentation Decathlon [J].
Antonelli, Michela ;
Reinke, Annika ;
Bakas, Spyridon ;
Farahani, Keyvan ;
Kopp-Schneider, Annette ;
Landman, Bennett A. ;
Litjens, Geert ;
Menze, Bjoern ;
Ronneberger, Olaf ;
Summers, Ronald M. ;
van Ginneken, Bram ;
Bilello, Michel ;
Bilic, Patrick ;
Christ, Patrick F. ;
Do, Richard K. G. ;
Gollub, Marc J. ;
Heckers, Stephan H. ;
Huisman, Henkjan ;
Jarnagin, William R. ;
McHugo, Maureen K. ;
Napel, Sandy ;
Pernicka, Jennifer S. Golia ;
Rhode, Kawal ;
Tobon-Gomez, Catalina ;
Vorontsov, Eugene ;
Meakin, James A. ;
Ourselin, Sebastien ;
Wiesenfarth, Manuel ;
Arbelaez, Pablo ;
Bae, Byeonguk ;
Chen, Sihong ;
Daza, Laura ;
Feng, Jianjiang ;
He, Baochun ;
Isensee, Fabian ;
Ji, Yuanfeng ;
Jia, Fucang ;
Kim, Ildoo ;
Maier-Hein, Klaus ;
Merhof, Dorit ;
Pai, Akshay ;
Park, Beomhee ;
Perslev, Mathias ;
Rezaiifar, Ramin ;
Rippel, Oliver ;
Sarasua, Ignacio ;
Shen, Wei ;
Son, Jaemin ;
Wachinger, Christian ;
Wang, Liansheng .
NATURE COMMUNICATIONS, 2022, 13 (01)
[2]  
Antoniou A, 2018, Arxiv, DOI arXiv:1711.04340
[3]  
Ba J, 2014, ACS SYM SER
[4]  
Bau D, 2020, Arxiv, DOI [arXiv:2005.07727, 10.1145/3306346.3323023, DOI 10.1145/3306346.3323023, 10.48550/arXiv:2005.07727]
[5]  
Bińkowski M, 2021, Arxiv, DOI arXiv:1801.01401
[6]   Assessment of Deep Generative Models for High-Resolution Synthetic Retinal Image Generation of Age-Related Macular Degeneration [J].
Burlina, Philippe M. ;
Joshi, Neil ;
Pacheco, Katia D. ;
Liu, T. Y. Alvin ;
Bressler, Neil M. .
JAMA OPHTHALMOLOGY, 2019, 137 (03) :258-264
[7]   Evidence of outer retinal changes in glaucoma patients as revealed by ultrahigh-resolution in vivo retinal imaging [J].
Choi, Stacey S. ;
Zawadzki, Robert J. ;
Lim, Michele C. ;
Brandt, James D. ;
Keltner, John L. ;
Doble, Nathan ;
Werner, John S. .
BRITISH JOURNAL OF OPHTHALMOLOGY, 2011, 95 (01) :131-141
[8]  
Costa P, 2017, Arxiv, DOI [arXiv:1701.08974, 10.48550/arXiv:1701.08974]
[9]   End-to-End Adversarial Retinal Image Synthesis [J].
Costa, Pedro ;
Galdran, Adrian ;
Meyer, Maria Ines ;
Niemeijer, Meindert ;
Abramoff, Michael ;
Mendonca, Ana Maria ;
Campilho, Aurelio .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2018, 37 (03) :781-791
[10]   FEEDBACK ON A PUBLICLY DISTRIBUTED IMAGE DATABASE: THE MESSIDOR DATABASE [J].
Decenciere, Etienne ;
Zhang, Xiwei ;
Cazuguel, Guy ;
Lay, Bruno ;
Cochener, Beatrice ;
Trone, Caroline ;
Gain, Philippe ;
Ordonez-Varela, John-Richard ;
Massin, Pascale ;
Erginay, Ali ;
Charton, Beatrice ;
Klein, Jean-Claude .
IMAGE ANALYSIS & STEREOLOGY, 2014, 33 (03) :231-234