Text-Driven Weakly Supervised OCT Lesion Segmentation with Structural Guidance

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Yang, Jiaqi, Mehta, Nitish, Hu, Xiaoling, Chen, Chao, Tsai, Chia-Ling
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912790267035648
author Yang, Jiaqi
Mehta, Nitish
Hu, Xiaoling
Chen, Chao
Tsai, Chia-Ling
author_facet Yang, Jiaqi
Mehta, Nitish
Hu, Xiaoling
Chen, Chao
Tsai, Chia-Ling
contents Accurate segmentation of Optical Coherence Tomography (OCT) images is crucial for diagnosing and monitoring retinal diseases. However, the labor-intensive nature of pixel-level annotation limits the scalability of supervised learning for large datasets. Weakly Supervised Semantic Segmentation (WSSS) offers a promising alternative by using weaker forms of supervision, such as image-level labels, to reduce the annotation burden. Despite its advantages, weak supervision inherently carries limited information. We propose a novel WSSS framework with only image-level labels for OCT lesion segmentation that integrates structural and text-driven guidance to produce high-quality, pixel-level pseudo labels. The framework employs two visual processing modules: one that processes the original OCT images and another that operates on layer segmentations augmented with anomalous signals, enabling the model to associate lesions with their corresponding anatomical layers. Complementing these visual cues, we leverage large-scale pretrained models to provide two forms of textual guidance: label-derived descriptions that encode local semantics, and domain-agnostic synthetic descriptions that, although expressed in natural image terms, capture spatial and relational semantics useful for generating globally consistent representations. By fusing these visual and textual features in a multi-modal framework, our method aligns semantic meaning with structural relevance, thereby improving lesion localization and segmentation performance. Experiments on three OCT datasets demonstrate state-of-the-art results, highlighting its potential to advance diagnostic accuracy and efficiency in medical imaging.
format Preprint
id arxiv_https___arxiv_org_abs_2411_12615
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Text-Driven Weakly Supervised OCT Lesion Segmentation with Structural Guidance
Yang, Jiaqi
Mehta, Nitish
Hu, Xiaoling
Chen, Chao
Tsai, Chia-Ling
Computer Vision and Pattern Recognition
Machine Learning
Accurate segmentation of Optical Coherence Tomography (OCT) images is crucial for diagnosing and monitoring retinal diseases. However, the labor-intensive nature of pixel-level annotation limits the scalability of supervised learning for large datasets. Weakly Supervised Semantic Segmentation (WSSS) offers a promising alternative by using weaker forms of supervision, such as image-level labels, to reduce the annotation burden. Despite its advantages, weak supervision inherently carries limited information. We propose a novel WSSS framework with only image-level labels for OCT lesion segmentation that integrates structural and text-driven guidance to produce high-quality, pixel-level pseudo labels. The framework employs two visual processing modules: one that processes the original OCT images and another that operates on layer segmentations augmented with anomalous signals, enabling the model to associate lesions with their corresponding anatomical layers. Complementing these visual cues, we leverage large-scale pretrained models to provide two forms of textual guidance: label-derived descriptions that encode local semantics, and domain-agnostic synthetic descriptions that, although expressed in natural image terms, capture spatial and relational semantics useful for generating globally consistent representations. By fusing these visual and textual features in a multi-modal framework, our method aligns semantic meaning with structural relevance, thereby improving lesion localization and segmentation performance. Experiments on three OCT datasets demonstrate state-of-the-art results, highlighting its potential to advance diagnostic accuracy and efficiency in medical imaging.
title Text-Driven Weakly Supervised OCT Lesion Segmentation with Structural Guidance
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2411.12615