TY - GEN
T1 - Semi-supervised Medical Image Segmentation based on Coarse-Fine Dual Training Streams
AU - Sun, Lizhi
AU - Huang, Yong
AU - Qiao, Zhengyu
AU - Yang, Boyu
AU - Li, Xiaochen
AU - Hao, Qun
N1 - Publisher Copyright:
© 2024 IEEE.
PY - 2024
Y1 - 2024
N2 - 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.
AB - 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.
KW - Dual training stream
KW - Large kernel attention
KW - Medical image segmentation
KW - Semi-supervised learning
UR - https://www.scopus.com/pages/publications/85217279004
U2 - 10.1109/BIBM62325.2024.10822484
DO - 10.1109/BIBM62325.2024.10822484
M3 - Conference contribution
AN - SCOPUS:85217279004
T3 - Proceedings - 2024 IEEE International Conference on Bioinformatics and Biomedicine, BIBM 2024
SP - 3722
EP - 3725
BT - Proceedings - 2024 IEEE International Conference on Bioinformatics and Biomedicine, BIBM 2024
A2 - Cannataro, Mario
A2 - Zheng, Huiru
A2 - Gao, Lin
A2 - Cheng, Jianlin
A2 - de Miranda, Joao Luis
A2 - Zumpano, Ester
A2 - Hu, Xiaohua
A2 - Cho, Young-Rae
A2 - Park, Taesung
PB - Institute of Electrical and Electronics Engineers Inc.
T2 - 2024 IEEE International Conference on Bioinformatics and Biomedicine, BIBM 2024
Y2 - 3 December 2024 through 6 December 2024
ER -