Dual-view Label and Feature Supervision Network for Semi-supervised Medical Image Segmentation
Zhen Zhou, Zuoyong Li, Z. Jane Wang, Shenghua Teng, Tao Wang · 2024
Semi-supervised learning has garnered significant attention, particularly in medical image segmentation, owing to its capacity to leverage a large number of unlabeled data and a limited amount of labeled data to improve performance. However, most existing semi-supervised segmentation methods exhibit shortcomings in supervising unlabeled data, both in the label space due to potential noise in pseudo-labels, and the feature space due to indistinct class boundaries. To address the issues above, we propose Dual-view Label and Feature Supervision Network, termed DLFS-Net, to enhance label and feature supervision for unlabeled data. Our approach is based on a dual-view learning strategy that incorporates two modules: the Confidence Fusion Supervision Module (CFS) and the Dual-view Prototype Learning Module (DPL). Specifically, the CFS module generates weight matrices to integrate complementary information from two model outputs, producing more accurate pseudo-labels in the label space. The DPL module reduces intra-class variations in the feature space from two different views through prototype learning to generate clear class boundaries. Experiments on the LA and Pancreas-CT datasets demonstrate that our framework show solid gains (e.g.,1.76% Dice and 2.37% Jaccard improvement on Pancreas-CT dataset with 20% labeled data) compared with the state-of-the-art methods.