Semi-supervised Medical Image Segmentation Method Based on Cross-pseudo Labeling Leveraging Strong and Weak Data Augmentation Strategies

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Chen, Yifei, Zhang, Chenyan, Ke, Yifan, Huang, Yiyu, Dai, Xuezhou, Qin, Feiwei, Zhang, Yongquan, Zhang, Xiaodong, Wang, Changmiao
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910335119654912
author Chen, Yifei
Zhang, Chenyan
Ke, Yifan
Huang, Yiyu
Dai, Xuezhou
Qin, Feiwei
Zhang, Yongquan
Zhang, Xiaodong
Wang, Changmiao
author_facet Chen, Yifei
Zhang, Chenyan
Ke, Yifan
Huang, Yiyu
Dai, Xuezhou
Qin, Feiwei
Zhang, Yongquan
Zhang, Xiaodong
Wang, Changmiao
contents Traditional supervised learning methods have historically encountered certain constraints in medical image segmentation due to the challenging collection process, high labeling cost, low signal-to-noise ratio, and complex features characterizing biomedical images. This paper proposes a semi-supervised model, DFCPS, which innovatively incorporates the Fixmatch concept. This significantly enhances the model's performance and generalizability through data augmentation processing, employing varied strategies for unlabeled data. Concurrently, the model design gives appropriate emphasis to the generation, filtration, and refinement processes of pseudo-labels. The novel concept of cross-pseudo-supervision is introduced, integrating consistency learning with self-training. This enables the model to fully leverage pseudo-labels from multiple perspectives, thereby enhancing training diversity. The DFCPS model is compared with both baseline and advanced models using the publicly accessible Kvasir-SEG dataset. Across all four subdivisions containing different proportions of unlabeled data, our model consistently exhibits superior performance. Our source code is available at https://github.com/JustlfC03/DFCPS.
format Preprint
id arxiv_https___arxiv_org_abs_2402_11273
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Semi-supervised Medical Image Segmentation Method Based on Cross-pseudo Labeling Leveraging Strong and Weak Data Augmentation Strategies
Chen, Yifei
Zhang, Chenyan
Ke, Yifan
Huang, Yiyu
Dai, Xuezhou
Qin, Feiwei
Zhang, Yongquan
Zhang, Xiaodong
Wang, Changmiao
Computer Vision and Pattern Recognition
Artificial Intelligence
Traditional supervised learning methods have historically encountered certain constraints in medical image segmentation due to the challenging collection process, high labeling cost, low signal-to-noise ratio, and complex features characterizing biomedical images. This paper proposes a semi-supervised model, DFCPS, which innovatively incorporates the Fixmatch concept. This significantly enhances the model's performance and generalizability through data augmentation processing, employing varied strategies for unlabeled data. Concurrently, the model design gives appropriate emphasis to the generation, filtration, and refinement processes of pseudo-labels. The novel concept of cross-pseudo-supervision is introduced, integrating consistency learning with self-training. This enables the model to fully leverage pseudo-labels from multiple perspectives, thereby enhancing training diversity. The DFCPS model is compared with both baseline and advanced models using the publicly accessible Kvasir-SEG dataset. Across all four subdivisions containing different proportions of unlabeled data, our model consistently exhibits superior performance. Our source code is available at https://github.com/JustlfC03/DFCPS.
title Semi-supervised Medical Image Segmentation Method Based on Cross-pseudo Labeling Leveraging Strong and Weak Data Augmentation Strategies
topic Computer Vision and Pattern Recognition
Artificial Intelligence
url https://arxiv.org/abs/2402.11273