Semi-supervised Segmentation of Histopathology Images with Noise-Aware Topological Consistency

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Xu, Meilong, Hu, Xiaoling, Gupta, Saumya, Abousamra, Shahira, Chen, Chao
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911957614854144
author Xu, Meilong
Hu, Xiaoling
Gupta, Saumya
Abousamra, Shahira
Chen, Chao
author_facet Xu, Meilong
Hu, Xiaoling
Gupta, Saumya
Abousamra, Shahira
Chen, Chao
contents In digital pathology, segmenting densely distributed objects like glands and nuclei is crucial for downstream analysis. Since detailed pixel-wise annotations are very time-consuming, we need semi-supervised segmentation methods that can learn from unlabeled images. Existing semi-supervised methods are often prone to topological errors, e.g., missing or incorrectly merged/separated glands or nuclei. To address this issue, we propose TopoSemiSeg, the first semi-supervised method that learns the topological representation from unlabeled histopathology images. The major challenge is for unlabeled images; we only have predictions carrying noisy topology. To this end, we introduce a noise-aware topological consistency loss to align the representations of a teacher and a student model. By decomposing the topology of the prediction into signal topology and noisy topology, we ensure that the models learn the true topological signals and become robust to noise. Extensive experiments on public histopathology image datasets show the superiority of our method, especially on topology-aware evaluation metrics. Code is available at https://github.com/Melon-Xu/TopoSemiSeg.
format Preprint
id arxiv_https___arxiv_org_abs_2311_16447
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Semi-supervised Segmentation of Histopathology Images with Noise-Aware Topological Consistency
Xu, Meilong
Hu, Xiaoling
Gupta, Saumya
Abousamra, Shahira
Chen, Chao
Image and Video Processing
Computer Vision and Pattern Recognition
In digital pathology, segmenting densely distributed objects like glands and nuclei is crucial for downstream analysis. Since detailed pixel-wise annotations are very time-consuming, we need semi-supervised segmentation methods that can learn from unlabeled images. Existing semi-supervised methods are often prone to topological errors, e.g., missing or incorrectly merged/separated glands or nuclei. To address this issue, we propose TopoSemiSeg, the first semi-supervised method that learns the topological representation from unlabeled histopathology images. The major challenge is for unlabeled images; we only have predictions carrying noisy topology. To this end, we introduce a noise-aware topological consistency loss to align the representations of a teacher and a student model. By decomposing the topology of the prediction into signal topology and noisy topology, we ensure that the models learn the true topological signals and become robust to noise. Extensive experiments on public histopathology image datasets show the superiority of our method, especially on topology-aware evaluation metrics. Code is available at https://github.com/Melon-Xu/TopoSemiSeg.
title Semi-supervised Segmentation of Histopathology Images with Noise-Aware Topological Consistency
topic Image and Video Processing
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2311.16447