Abstract

Info

Title:: Semi-supervised lung tumor segmentation based on multi-scale consistency and regional reliability perception

Author(s):: LIU Weipeng1; 2; QI Yedong1; 2; LI Jian1; XU Haixing1; 2; 1. School of Artificial Intelligence, Hebei University of Technology, Tianjin 300130, China 2. Laboratory of Advanced Perception and Intelligent Control of High-End Equipment, Hebei University of Technology, Tianjin 300130, China

Keywords:: Keywords: semi-supervised learning medical image segmentation lung tumor reliability perception multi-scale consistency

Abstract:: Abstract: A semi-supervised learning method based on multi-scale consistency and regional reliability perception is proposed to combine unlabeled data with a small amount of labeled data to achieve high-performance lung tumor segmentation tasks. A multi-scale consistency mean teacher framework is used to construct a multi-scale consistency loss and constrain the outputs in the mean teacher network to be consistent across multiple scales, so that the model learns richer consistency knowledge. In addition, a regional reliability perception scheme is adopted to make the knowledge exchange between consistency learning more efficient, enabling the model to learn more valid and reliable knowledge from unlabeled data. The evaluation on the lung tumor dataset in the Medical Segmentation Decathlon shows superior performance of the proposed method over current state-of-the-art semi-supervised learning methods, validating its effectiveness.