Semi-supervised Medical Image Segmentation Via Cross Teaching Between MobileNet and MobileViT

Yuan Yang,Lin Zhang,Lei Ren
DOI: https://doi.org/10.1016/j.imavis.2024.105196
IF: 3.86
2024-01-01
Image and Vision Computing
Abstract:Recently, deep learning methods that use a combination of convolutional neural networks and Transformers have shown excellent results in both completely supervised and semi-supervised medical image segmentation tasks. This study provides a simple and effective framework for semi-supervised medical image segmentation by introducing cross teaching between MobileNet and MobileViT. Specifically, we built two different two-path parallel semantic segmentation networks with MobileNet and MobileViT as the main modules. Cross teaching between MobileNet and MobileViT was performed, in which the prediction of each network was used as a pseudo-label to supervise the training of other networks in a direct end-to-end manner. Finally, experiments on two common benchmark tests showed that the proposed framework was superior to six existing semi-supervised learning methods. This shows that our framework can effectively use unlabeled data to improve the performance and is superior to the latest semi-supervised segmentation method. Codes are available at: github.com/yywbkn /MV2-cross-teaching-MobileViT.
What problem does this paper attempt to address?