TumorDet: A Breast Tumor Detection Model Based on Transfer Learning and ShuffleNet

Abstract

Breast tumor is among the most malignant tumors and early detection can improve patient’s survival rate. Currently, mammography is the most reliable method for diagnosing breast tumor because of high image resolution. Because of the rapid development of medical and artificial intelligence techniques, computer-aided diagnosis technology can greatly improve the detection accuracy of breast tumors and medical imaging has begun to use deep-learning-based approaches. In this study, the TumorDet model is proposed to detect the benign and malignant lesions of breast tumor, which has positive significance for assisting doctors in diagnosis. We use the proposed TumorDet to analyze and predict breast tumors on the real MRI dataset. (1) We introduce an adaptive gamma correction (AGC) method to balance brightness equalization and increase the contrast of mammography images; (2) we use the ShuffleNet model to exchange information between different feature layers and extract the hidden high-level features of medical images; and (3) we use the transfer learning method to fine-tune the ShuffleNet model and obtain the optimal parameters. The proposed TumorDet model has shown that accuracy, sensitivity, and specificity reach 90.43%, 89.37%, and 87.81%, respectively. TumorDet performs well in the breast tumor detection task. In addition, we use the proposed TumorDet to conduct experiments on other tasks, such as forest fires, and the robustness of TumorDet is proved by experimental results. TumorDet employs the ShuffleNet model to exchange information between different feature layers without increasing the number of network parameters and applies transfer learning methods to further extract the basic features of medical images by fine-tuning. The model is beneficial for the localization and classification of breast tumors and also performs well in forest fire detection.