GID: Global information distillation for medical semantic segmentation

Yong-Sen Ye,Min-Rong Chen,Hao-Li Zou,Bai-Bing Yang,Guo-Qiang Zeng
DOI: https://doi.org/10.1016/j.neucom.2022.06.065
IF: 6
2022-09-07
Neurocomputing
Abstract:In this work, we consider transferring global information from Transformer to Convolutional Neural Network (CNN) for medical semantic segmentation tasks. Previous network models for medical semantic segmentation tasks often suffer from difficulties in modeling global information or oversized model parameters. Here, to design a compact network with global and local information, we extract the global information modeling capability of Transformer into the CNN network and successfully apply it to the medical semantic segmentation tasks, called Global Information Distillation. In addition, the following two contributions are proposed to improve the effectiveness of distillation: i) We present an Information Transfer Module, which is based on a convolutional layer to prevent over-regularization and a Transformer layer to transfer global information; ii) For purpose of better transferring the teacher’s soft targets, a Shrinking Result-Pixel distillation method is proposed in this paper. The effectiveness of our knowledge distillation approach is demonstrated by the experiments on multi-organ and cardiac segmentation tasks.
computer science, artificial intelligence
What problem does this paper attempt to address?