Semi-supervised Medical Image Segmentation based on Coarse-Fine Dual Training Streams

Lizhi Sun, Yong Huang*, Zhengyu Qiao, Boyu Yang, Xiaochen Li, Qun Hao

*Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

Commonly used semi-supervised medical segmentation networks usually use consistent learning under different data perturbations to regularise training, ignoring the multiscale information of the data itself. Therefore, this paper proposes a new network based on coarse and fine dual training streams(CF-UNet), which consists of a backbone network and auxiliary learning branches(ALB). Our approach has the following two novel designs: 1) We design a simple and effective coarse- fine dual training streams. Specifically, in the coarse training stream, we improve the robustness and generalisation of the model by establishing regularisation between different strong and weak perturbation views. In the fine training stream, we introduce an auxiliary learning branch to improve the prediction performance of the backbone network.2) In the ALB module, we design the channel spatial fusion attention module (CSMA) and multiscale large kernel convolutional attention (MS-LKA) to perform feature extraction and fusion from a variety of scales. We evaluate our proposed method on ACDC and DRIVE datasets and numerous experiments have shown that our CF-UNet outperforms state-of-the-art networks. Code is available at https://github.com/slz-bit/CF-UNet.

Original languageEnglish
Title of host publicationProceedings - 2024 IEEE International Conference on Bioinformatics and Biomedicine, BIBM 2024
EditorsMario Cannataro, Huiru Zheng, Lin Gao, Jianlin Cheng, Joao Luis de Miranda, Ester Zumpano, Xiaohua Hu, Young-Rae Cho, Taesung Park
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages3722-3725
Number of pages4
ISBN (Electronic)9798350386226
DOIs
Publication statusPublished - 2024
Event2024 IEEE International Conference on Bioinformatics and Biomedicine, BIBM 2024 - Lisbon, Portugal
Duration: 3 Dec 20246 Dec 2024

Publication series

NameProceedings - 2024 IEEE International Conference on Bioinformatics and Biomedicine, BIBM 2024

Conference

Conference2024 IEEE International Conference on Bioinformatics and Biomedicine, BIBM 2024
Country/TerritoryPortugal
CityLisbon
Period3/12/246/12/24

Keywords

  • Dual training stream
  • Large kernel attention
  • Medical image segmentation
  • Semi-supervised learning

Fingerprint

Dive into the research topics of 'Semi-supervised Medical Image Segmentation based on Coarse-Fine Dual Training Streams'. Together they form a unique fingerprint.

Cite this