Dual Structure-Aware Image Filterings for Semi-supervised Medical Image Segmentation

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Gu, Yuliang, Sun, Zhichao, Chen, Tian, Xiao, Xin, Liu, Yepeng, Xu, Yongchao, Najman, Laurent
Format: Preprint
Publié: 2023
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866917623293280256
author Gu, Yuliang
Sun, Zhichao
Chen, Tian
Xiao, Xin
Liu, Yepeng
Xu, Yongchao
Najman, Laurent
author_facet Gu, Yuliang
Sun, Zhichao
Chen, Tian
Xiao, Xin
Liu, Yepeng
Xu, Yongchao
Najman, Laurent
contents Semi-supervised image segmentation has attracted great attention recently. The key is how to leverage unlabeled images in the training process. Most methods maintain consistent predictions of the unlabeled images under variations (e.g., adding noise/perturbations, or creating alternative versions) in the image and/or model level. In most image-level variation, medical images often have prior structure information, which has not been well explored. In this paper, we propose novel dual structure-aware image filterings (DSAIF) as the image-level variations for semi-supervised medical image segmentation. Motivated by connected filtering that simplifies image via filtering in structure-aware tree-based image representation, we resort to the dual contrast invariant Max-tree and Min-tree representation. Specifically, we propose a novel connected filtering that removes topologically equivalent nodes (i.e. connected components) having no siblings in the Max/Min-tree. This results in two filtered images preserving topologically critical structure. Applying the proposed DSAIF to mutually supervised networks decreases the consensus of their erroneous predictions on unlabeled images. This helps to alleviate the confirmation bias issue of overfitting to noisy pseudo labels of unlabeled images, and thus effectively improves the segmentation performance. Extensive experimental results on three benchmark datasets demonstrate that the proposed method significantly/consistently outperforms some state-of-the-art methods. The source codes will be publicly available.
format Preprint
id arxiv_https___arxiv_org_abs_2312_07264
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Dual Structure-Aware Image Filterings for Semi-supervised Medical Image Segmentation
Gu, Yuliang
Sun, Zhichao
Chen, Tian
Xiao, Xin
Liu, Yepeng
Xu, Yongchao
Najman, Laurent
Computer Vision and Pattern Recognition
Semi-supervised image segmentation has attracted great attention recently. The key is how to leverage unlabeled images in the training process. Most methods maintain consistent predictions of the unlabeled images under variations (e.g., adding noise/perturbations, or creating alternative versions) in the image and/or model level. In most image-level variation, medical images often have prior structure information, which has not been well explored. In this paper, we propose novel dual structure-aware image filterings (DSAIF) as the image-level variations for semi-supervised medical image segmentation. Motivated by connected filtering that simplifies image via filtering in structure-aware tree-based image representation, we resort to the dual contrast invariant Max-tree and Min-tree representation. Specifically, we propose a novel connected filtering that removes topologically equivalent nodes (i.e. connected components) having no siblings in the Max/Min-tree. This results in two filtered images preserving topologically critical structure. Applying the proposed DSAIF to mutually supervised networks decreases the consensus of their erroneous predictions on unlabeled images. This helps to alleviate the confirmation bias issue of overfitting to noisy pseudo labels of unlabeled images, and thus effectively improves the segmentation performance. Extensive experimental results on three benchmark datasets demonstrate that the proposed method significantly/consistently outperforms some state-of-the-art methods. The source codes will be publicly available.
title Dual Structure-Aware Image Filterings for Semi-supervised Medical Image Segmentation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2312.07264