Saved in:
Bibliographic Details
Main Authors: Demirci, Gozde Merve, Yao, Jiachen, Ho, Ming-Chih, Hu, Xiaoling, Wu, Wei-Chi, Chen, Chao, Tsai, Chia-Ling
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2411.09140
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913577894412288
author Demirci, Gozde Merve
Yao, Jiachen
Ho, Ming-Chih
Hu, Xiaoling
Wu, Wei-Chi
Chen, Chao
Tsai, Chia-Ling
author_facet Demirci, Gozde Merve
Yao, Jiachen
Ho, Ming-Chih
Hu, Xiaoling
Wu, Wei-Chi
Chen, Chao
Tsai, Chia-Ling
contents Accurate segmentation of retinal images plays a crucial role in aiding ophthalmologists in diagnosing retinopathy of prematurity (ROP) and assessing its severity. However, due to their underdeveloped, thinner vessels, manual annotation in infant fundus images is very complex, and this presents challenges for fully supervised learning. To address the scarcity of annotations, we propose a semi supervised segmentation framework designed to advance ROP studies without the need for extensive manual vessel annotation. Unlike previous methods that rely solely on limited labeled data, our approach leverages teacher student learning by integrating two powerful components: an uncertainty weighted vessel unveiling module and domain adversarial learning. The vessel unveiling module helps the model effectively reveal obscured and hard to detect vessel structures, while adversarial training aligns feature representations across different domains, ensuring robust and generalizable vessel segmentations. We validate our approach on public datasets (CHASEDB, STARE) and an in-house ROP dataset, demonstrating its superior performance across multiple evaluation metrics. Additionally, we extend the model's utility to a downstream task of ROP multi-stage classification, where vessel masks extracted by our segmentation model improve diagnostic accuracy. The promising results in classification underscore the model's potential for clinical application, particularly in early-stage ROP diagnosis and intervention. Overall, our work offers a scalable solution for leveraging unlabeled data in pediatric ophthalmology, opening new avenues for biomarker discovery and clinical research.
format Preprint
id arxiv_https___arxiv_org_abs_2411_09140
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Adversarial Vessel-Unveiling Semi-Supervised Segmentation for Retinopathy of Prematurity Diagnosis
Demirci, Gozde Merve
Yao, Jiachen
Ho, Ming-Chih
Hu, Xiaoling
Wu, Wei-Chi
Chen, Chao
Tsai, Chia-Ling
Computer Vision and Pattern Recognition
53-04, 68
I.4
Accurate segmentation of retinal images plays a crucial role in aiding ophthalmologists in diagnosing retinopathy of prematurity (ROP) and assessing its severity. However, due to their underdeveloped, thinner vessels, manual annotation in infant fundus images is very complex, and this presents challenges for fully supervised learning. To address the scarcity of annotations, we propose a semi supervised segmentation framework designed to advance ROP studies without the need for extensive manual vessel annotation. Unlike previous methods that rely solely on limited labeled data, our approach leverages teacher student learning by integrating two powerful components: an uncertainty weighted vessel unveiling module and domain adversarial learning. The vessel unveiling module helps the model effectively reveal obscured and hard to detect vessel structures, while adversarial training aligns feature representations across different domains, ensuring robust and generalizable vessel segmentations. We validate our approach on public datasets (CHASEDB, STARE) and an in-house ROP dataset, demonstrating its superior performance across multiple evaluation metrics. Additionally, we extend the model's utility to a downstream task of ROP multi-stage classification, where vessel masks extracted by our segmentation model improve diagnostic accuracy. The promising results in classification underscore the model's potential for clinical application, particularly in early-stage ROP diagnosis and intervention. Overall, our work offers a scalable solution for leveraging unlabeled data in pediatric ophthalmology, opening new avenues for biomarker discovery and clinical research.
title Adversarial Vessel-Unveiling Semi-Supervised Segmentation for Retinopathy of Prematurity Diagnosis
topic Computer Vision and Pattern Recognition
53-04, 68
I.4
url https://arxiv.org/abs/2411.09140