Triple-view feature learning for medical image segmentation

Conference item

Deep learning models, e.g. supervised Encoder-Decoder style networks, exhibit promising performance in medical image segmentation, but come with a high labelling cost. We propose TriSegNet, a semi-supervised semantic segmentation framework. It uses triple-view feature learning on a limited amount of labelled data and a large amount of unlabeled data. The triple-view architecture consists of three pixel-level classifiers and a low-level shared-weight learning module. The model is first initialized with labelled data. Label processing, including data perturbation, confidence label voting and unconfident label detection for annotation, enables the model to train on labelled and unlabeled data simultaneously. The confidence of each model gets improved through the other two views of the feature learning. This process is repeated until each model reaches the same confidence level as its counterparts. This strategy enables triple-view learning of generic medical image datasets. Bespoke overlap-based and boundary-based loss functions are tailored to the different stages of the training. The segmentation results are evaluated on four publicly available benchmark datasets including Ultrasound, CT, MRI, and Histology images. Repeated experiments demonstrate the effectiveness of the proposed network compared against other semi-supervised algorithms, across a large set of evaluation measures.

Authors: Wang, Z; Voiculescu, ID

• Publication status: Published
• Peer review status: Peer reviewed
• Publisher: Springer
• Host title: Resource-Efficient Medical Image Analysis: First MICCAI Workshop, REMIA 2022, Singapore, September 22, 2022, Proceedings
• Pages: 42–54
• Series: Lecture Notes in Computer Science
• Series number: 13543
• Place of publication: Cham, Switzerland
• Publication date: 2022-09-15
• Acceptance date: 2022-07-25
• Event title: Workshop on Resource-Efficient Medical Image Analysis, MICCAI 2022
• Event location: Singapore
• Event website: https://miccai-remia.github.io/
• Event start date: 2022-09-22
• Event end date: 2022-09-22
• DOI: 10.1007/978-3-031-16876-5_5
• ISSN: 0302-9743
• EISBN: 9783031168765
• ISBN: 9783031168758
• Language: English
• Keywords: FFR