Inter- and intra-uncertainty based feature aggregation model for semi-supervised histopathology image segmentation

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Jin, Qiangguo, Cui, Hui, Sun, Changming, Song, Yang, Zheng, Jiangbin, Cao, Leilei, Wei, Leyi, Su, Ran
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866929282256732160
author Jin, Qiangguo
Cui, Hui
Sun, Changming
Song, Yang
Zheng, Jiangbin
Cao, Leilei
Wei, Leyi
Su, Ran
author_facet Jin, Qiangguo
Cui, Hui
Sun, Changming
Song, Yang
Zheng, Jiangbin
Cao, Leilei
Wei, Leyi
Su, Ran
contents Acquiring pixel-level annotations is often limited in applications such as histology studies that require domain expertise. Various semi-supervised learning approaches have been developed to work with limited ground truth annotations, such as the popular teacher-student models. However, hierarchical prediction uncertainty within the student model (intra-uncertainty) and image prediction uncertainty (inter-uncertainty) have not been fully utilized by existing methods. To address these issues, we first propose a novel inter- and intra-uncertainty regularization method to measure and constrain both inter- and intra-inconsistencies in the teacher-student architecture. We also propose a new two-stage network with pseudo-mask guided feature aggregation (PG-FANet) as the segmentation model. The two-stage structure complements with the uncertainty regularization strategy to avoid introducing extra modules in solving uncertainties and the aggregation mechanisms enable multi-scale and multi-stage feature integration. Comprehensive experimental results over the MoNuSeg and CRAG datasets show that our PG-FANet outperforms other state-of-the-art methods and our semi-supervised learning framework yields competitive performance with a limited amount of labeled data.
format Preprint
id arxiv_https___arxiv_org_abs_2403_12767
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Inter- and intra-uncertainty based feature aggregation model for semi-supervised histopathology image segmentation
Jin, Qiangguo
Cui, Hui
Sun, Changming
Song, Yang
Zheng, Jiangbin
Cao, Leilei
Wei, Leyi
Su, Ran
Computer Vision and Pattern Recognition
Acquiring pixel-level annotations is often limited in applications such as histology studies that require domain expertise. Various semi-supervised learning approaches have been developed to work with limited ground truth annotations, such as the popular teacher-student models. However, hierarchical prediction uncertainty within the student model (intra-uncertainty) and image prediction uncertainty (inter-uncertainty) have not been fully utilized by existing methods. To address these issues, we first propose a novel inter- and intra-uncertainty regularization method to measure and constrain both inter- and intra-inconsistencies in the teacher-student architecture. We also propose a new two-stage network with pseudo-mask guided feature aggregation (PG-FANet) as the segmentation model. The two-stage structure complements with the uncertainty regularization strategy to avoid introducing extra modules in solving uncertainties and the aggregation mechanisms enable multi-scale and multi-stage feature integration. Comprehensive experimental results over the MoNuSeg and CRAG datasets show that our PG-FANet outperforms other state-of-the-art methods and our semi-supervised learning framework yields competitive performance with a limited amount of labeled data.
title Inter- and intra-uncertainty based feature aggregation model for semi-supervised histopathology image segmentation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2403.12767