Loading [a11y]/accessibility-menu.js
Retina-TransNet: A Gradient-Guided Few-Shot Retinal Vessel Segmentation Net | IEEE Journals & Magazine | IEEE Xplore

Retina-TransNet: A Gradient-Guided Few-Shot Retinal Vessel Segmentation Net


Abstract:

Due to the high labor cost of physicians, it is difficult to collect a rich amount of manually-labeled medical images for developing learning-based computer-aided diagnos...Show More

Abstract:

Due to the high labor cost of physicians, it is difficult to collect a rich amount of manually-labeled medical images for developing learning-based computer-aided diagnosis (CADx) systems or segmentation algorithms. To tackle this issue, we reshape the image segmentation task as an image-to-image (I2I) translation problem and propose a retinal vascular segmentation network, which can achieve good cross-domain generalizability even with a small amount of training data. We devise primarily two components to facilitate this I2I-based segmentation method. The first is the constraints provided by the proposed gradient-vector-flow (GVF) loss, and, the second is a two-stage Unet (2Unet) generator with a skip connection. This configuration makes 2Unet's first-stage play a role similar to conventional Unet, but forces 2Unet's second stage to learn to be a refinement module. Extensive experiments show that by re-casting retinal vessel segmentation as an image-to-image translation problem, our I2I translator-based segmentation subnetwork achieves better cross-domain generalizability than existing segmentation methods. Our model, trained on one dataset, e.g., DRIVE, can produce segmentation results stably on datasets of other domains, e.g., CHASE-DB1, STARE, HRF, and DIARETDB1, even in low-shot circumstances.
Published in: IEEE Journal of Biomedical and Health Informatics ( Volume: 27, Issue: 10, October 2023)
Page(s): 4902 - 4913
Date of Publication: 25 July 2023

ISSN Information:

PubMed ID: 37490372

Funding Agency:


Contact IEEE to Subscribe

References

References is not available for this document.