Retinal Vessel Segmentation in Medical Diagnosis using Multi-scale Attention Generative Adversarial Networks

Minqiang Yang, Yinru Ye, Kai Ye, Wei Zhou, Xiping Hu*, Bin Hu*

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

1 Citation (Scopus)

Abstract

With the advancement of medical technology, the demand for efficient and precise medical diagnosis is growing. Retinal vessel segmentation using artificial intelligence techniques is vital for medical diagnosis of degenerative retinal diseases. In this paper, we introduce a multi-scale generative adversarial network with class activation mapping, which can effectively improve the efficiency and accuracy of vessel segmentation using artificial intelligence. The task of vessel segmentation is better achieved due to our proposed architecture, which incorporates an attention mechanism and a multi-scale discrimination. It not only strengthens the ability to locate and segment fine retinal vessels, but also enables the model to have the ability to discriminate different receptive fields. To tackle the instability problem caused by unsupervised learning of generative adversarial networks, we introduce a supervised segmentation loss to improve model stability and convergence speed. And we propose a data augmentation method by reconstructing and combining fundus images to make the model obtain better generalization ability. We compare our method with previous models by several metrics and perform ablation study on each component of the model, demonstrating the superiority and effectiveness of the model.

Original languageEnglish
JournalMobile Networks and Applications
DOIs
Publication statusAccepted/In press - 2023
Externally publishedYes

Keywords

  • Class activation mapping
  • Data augmentation
  • Multi-scale generative adversarial network
  • Retinal vessel segmentation

Fingerprint

Dive into the research topics of 'Retinal Vessel Segmentation in Medical Diagnosis using Multi-scale Attention Generative Adversarial Networks'. Together they form a unique fingerprint.

Cite this