TY - GEN
T1 - Dual-Contrastive Dual-Consistency Dual-Transformer: A Semi-Supervised Approach to Medical Image Segmentation
AU - Wang, Ziyang
AU - Ma, Congying
PY - 2023/12/25
Y1 - 2023/12/25
N2 - Medical image segmentation serves as a crucial under-pinning for a myriad of clinical applications. The advent of deep learning techniques has significantly propelled advancements in this field. However, challenges persist due to the limited availability of labelled medical imaging data and the substantial cost of data annotation. This paper introduces a novel semi-supervised learning strategy, amalgamating pseudo-labelling and contrastive learning with a consistency regularization framework. This innovative approach incorporates a modified contrastive learning strategy and a confidence-aware pseudo-labeling strategy, both of which are integrated into a dual-segmentation network ensemble learning structure. Inspired by the recent success of self-attention mechanisms, we harness the power of the Vision Transofmer(ViT) within our proposed semi-supervised framework, and conduct a comprehensive comparison among various combinations of ViT and Convolutional Neural Network(CNN) with the proposed strategy. The efficacy of our proposed method is validated using a publicly available medical image segmentation dataset, where it demonstrates state-of-the-art performance against established methods. The proposed method, all baseline methods, and dataset are available at https://github.com/ziyangwang007/CV-SSL-MIS.
AB - Medical image segmentation serves as a crucial under-pinning for a myriad of clinical applications. The advent of deep learning techniques has significantly propelled advancements in this field. However, challenges persist due to the limited availability of labelled medical imaging data and the substantial cost of data annotation. This paper introduces a novel semi-supervised learning strategy, amalgamating pseudo-labelling and contrastive learning with a consistency regularization framework. This innovative approach incorporates a modified contrastive learning strategy and a confidence-aware pseudo-labeling strategy, both of which are integrated into a dual-segmentation network ensemble learning structure. Inspired by the recent success of self-attention mechanisms, we harness the power of the Vision Transofmer(ViT) within our proposed semi-supervised framework, and conduct a comprehensive comparison among various combinations of ViT and Convolutional Neural Network(CNN) with the proposed strategy. The efficacy of our proposed method is validated using a publicly available medical image segmentation dataset, where it demonstrates state-of-the-art performance against established methods. The proposed method, all baseline methods, and dataset are available at https://github.com/ziyangwang007/CV-SSL-MIS.
UR - http://www.scopus.com/inward/record.url?scp=85182944429&partnerID=8YFLogxK
UR - https://ieeexplore.ieee.org/document/10350732
U2 - 10.1109/ICCVW60793.2023.00094
DO - 10.1109/ICCVW60793.2023.00094
M3 - Conference publication
AN - SCOPUS:85182944429
T3 - Proceedings - IEEE/CVF International Conference on Computer Vision Workshops, ICCVW 2023
SP - 870
EP - 879
BT - Proceedings - 2023 IEEE/CVF International Conference on Computer Vision Workshops (ICCVW)
PB - IEEE
T2 - 2023 IEEE/CVF International Conference on Computer Vision Workshops, ICCVW 2023
Y2 - 2 October 2023 through 6 October 2023
ER -