RPF-ELD: Regional Prior Fusion Using Early and Late Distillation for Breast Cancer Recognition in Ultrasound Images

Haosen Wang,Gengyuan Zhang,Yingnan Zhao,Fang Lai,Wenwei Cui,Jiexiao Xue,Qihang Wang,Hao Zhang,Yi Lin
DOI: https://doi.org/10.20944/preprints202411.1419.v1
2024-01-01
Abstract:Breast cancer is one of the main factors responsible for the deaths of women worldwide. Ultrasound imaging is a key method for early detection of breast cancer, which can help pa- tients gain valuable treatment time and improve their chances of survival. The computer-aided system of breast cancer recognition has started to receive attention due to the lack of experienced sonographers. Presently, most breast cancer recognition methods typically suffer from uncertain locations and proportions of tumor regions in ultrasound images. In this paper, we propose a novel Regional Prior Fusion framework using Early and Late Distillation (RPF-ELD), inspired by the knowledge distillation of the teacher-student framework, for breast cancer recognition in ultrasound images. Firstly, to enhance the concentration of the tumor regions, a high-performing prior-fused model is trained as the teacher model using ultrasound images with the corresponding regional prior information. Next, a diagnostic model is trained as the student model under the prior-fused model distillation using early and late features to implicitly obtain the regional prior knowledge. Finally, the diagnostic model recognizes the categories of breast cancer from only ultrasound images using the experience from distilled prior knowledge. Two publicly released datasets are used to evaluate the proposed RPF-ELD framework. Experimental results demonstrate that the proposed RPF-ELD surpasses current state-of-the-art methods.
What problem does this paper attempt to address?