Login Register Cart 0
Generic placeholder image

Current Bioinformatics

Editor-in-Chief

ISSN (Print): 1574-8936
ISSN (Online): 2212-392X

Back
Journal Home About Journal Editorial Board Journal Insight Current Issue Volumes/Issues
Subscribe
Research Article

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

Author(s): Tao Zhang, Leying Pan*, Qiang Yang, Guoping Yang, Nan Han and Shaojie Qiao

Volume 19, Issue 2, 2024

Published on: 20 November, 2023

Page: [119 - 128] Pages: 10

DOI: 10.2174/1574893618666230815121150

Price: $65

conference banner
Abstract

Background: 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.

Objective: We use the proposed TumorDet to analyze and predict breast tumors on the real MRI dataset.

Methods: (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.

Results: 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.

Conclusion: TumorDet employs the ShuffleNet model to exchange information between different feature layers without increasing the number of network parameters and applies transfer learning method 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.

« Previous Next »
Graphical Abstract

[1]
Li Y, Liu Y, Huang L, Wang Z, Luo J. Deep weakly-supervised breast tumor segmentation in ultrasound images with explicit anatomical constraints. Med Image Anal 2022; 76(76): 102315.
[http://dx.doi.org/10.1016/j.media.2021.102315] [PMID: 34902792]
[2]
Mohammed FE, Zghal NS. Multiclassification Model of Histopathological Breast Cancer Based on Deep Neural Network 2022 19th International Multi-Conference on Systems, Signals & Devices (SSD). Sétif, Algeria 2022.
[http://dx.doi.org/10.1109/SSD54932.2022.9955814]
[3]
Veena M, Padma MC. Fusion of Features from Mammogram and DBT Views for Detection of Breast TumourCognition and Recognition. Heidelberg: Springer 2023; pp. 229-42.
[http://dx.doi.org/10.1007/978-3-031-22405-8_18]
[4]
Deshmukh PB, Kashyap KL. Research Challenges in Breast Cancer Classification through Medical Imaging Modalities using Machine Learning. 2021 International Conference on Industrial Electronics Research and Applications. New Delhi, India. 2021; pp. 167-83.
[http://dx.doi.org/10.1109/ICIERA53202.2021.9726746]
[5]
Taori K, Dhakate S, Rathod J. Evaluation of Breast Masses Using Mammography and Sonography as First Line Investigations J Med Imaging (Bellingham) 2013; 3(1): 40-9.
[http://dx.doi.org/10.4236/ojmi.2013.31006]
[6]
Wang T, Wang H, Deng J, Zhang D, Feng J, Chen B. Feature generation and multi-sequence fusion based deep convolutional network for breast tumor diagnosis with missing MR sequences. Biomed Signal Process Control 2023; 82(82): 104536.
[http://dx.doi.org/10.1016/j.bspc.2022.104536]
[7]
Jiang Y, Xie J, Zhang D. An Adaptive Offset Activation Function for CNN Image Classification Tasks. Electronics (Basel) 2022; 11(22): 3799.
[http://dx.doi.org/10.3390/electronics11223799]
[8]
Chhapariya K. CNN-based Salient Object Detection on Hyperspectral Images using Extended Morphology. TechRxiv 2022.
[http://dx.doi.org/10.36227/techrxiv.19422101]
[9]
Tripathi S, Singh SK. An Object Aware Hybrid U-Net for Breast Tumour AnnotationBiomedical Signal and Image Processing with Artificial Intelligence. Heidelberg: Springer 2023; pp. 87-105.
[http://dx.doi.org/10.1007/978-3-031-15816-2_5]
[10]
Chen G, Li L, Dai Y. AAU-net: An Adaptive Attention U-net for Breast Lesions Segmentation in Ultrasound Images. IEEE Trans Med Imaging 2022; 385-97.
[http://dx.doi.org/10.1109/TMI.2022.3226268] [PMID: 36455083]
[11]
Punn NS, Agarwal S. RCA-IUnet: A residual cross-spatial attention-guided inception U-Net model for tumor segmentation in breast ultrasound imaging. Mach Vis Appl 2022; 33(2): 27.
[http://dx.doi.org/10.1007/s00138-022-01280-3]
[12]
Girshick R, Donahue J, Darrell T. Rich Feature Hierarchies for Accurate Object Detection and Semantic Segmentation. IEEE Conference on Computer Vision and Pattern Recognition. Columbus. 2014; pp. 580-7.
[http://dx.doi.org/10.1109/CVPR.2014.81]
[13]
Girshick R. Fast R-CNN. IEEE International Conference on Computer Vision. Santiago, Chile. 2015; pp. 1440-8.
[http://dx.doi.org/10.1109/ICCV.2015.169]
[14]
He K, Gkioxari G, Dollar P, Girshick R. Mask R-CNN. IEEE Trans Pattern Anal Mach Intell. IEEE. 2020;. pp. 42(2): 386-97.
[http://dx.doi.org/10.1109/TPAMI.2018.2844175] [PMID: 29994331]
[15]
Padma T. Image Segmentation using Mask R-CNN for Tumor Detection from Medical Images. 2022 International Conference on Electronics and Renewable Systems. Tuticorin, India. 2022; pp. 1015-21.
[http://dx.doi.org/10.1109/ICEARS53579.2022.9751891]
[16]
Qiu X, Lei H, Xie H. Segmentation of Multiple Myeloma Cells Using Feature Selection Pyramid Network and Semantic Cascade Mask RCNN 2022 IEEE 19th International Symposium on Biomedical Imaging (ISBI). Kolkata, India. 2022; pp. 1015-21.
[http://dx.doi.org/10.1109/ISBI52829.2022.9761460]
[17]
Ranjbarzadeh R, Jafarzadeh Ghoushchi S, Tataei Sarshar N, et al. ME-CCNN: Multi-encoded images and a cascade convolutional neural network for breast tumor segmentation and recognition. Artif Intell Rev 2023; 56(9): 10099-136.
[http://dx.doi.org/10.1007/s10462-023-10426-2]
[18]
Chiang TC, Huang YS, Chen RT, Huang CS, Chang RF. Tumor Detection in Automated Breast Ultrasound Using 3-D CNN and Prioritized Candidate Aggregation. IEEE Trans Med Imaging 2019; 38(1): 240-9.
[http://dx.doi.org/10.1109/TMI.2018.2860257] [PMID: 30059297]
[19]
Tao C, Chen K, Han L, et al. New one-step model of breast tumor locating based on deep learning. J XRay Sci Technol 2019; 27(5): 839-56.
[http://dx.doi.org/10.3233/XST-190548] [PMID: 31306148]
[20]
Daoud MI, Al-Ali A, Alazrai R, et al. An Edge-Based Selection Method for Improving Regions-of-Interest Localizations Obtained Using Multiple Deep Learning Object-Detection Models in Breast Ultrasound Images. Sensors (Basel) 2022; 22(18): 6721.
[http://dx.doi.org/10.3390/s22186721] [PMID: 36146070]
[21]
Lu W, Wang Z, He Y. Breast Cancer Detection Based on Merging Four Modes MRI Using Convolutional Neural Networks. ICASSP 2019 - 2019 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP). Brighton, UK . 2019; p. 1015-21.
[http://dx.doi.org/10.1109/ICASSP.2019.8683149]
[22]
Liu Y, Li J, Xu D, et al. Auxiliary diagnosis of small tumor in mammography based on deep learning. J Ambient Intell Humaniz Comput 2023; 14(2): 1061-9.
[http://dx.doi.org/10.1007/s12652-021-03358-8]
[23]
Rao S. MITOS-RCNN: A Novel Approach to Mitotic Figure Detection in Breast Cancer Histopathology Images using Region Based Convolutional. Neural Networks arXiv:180701788 2018.
[24]
Jayandhi G, Leena Jasmine JS, Seetharaman R. Breast Cancer Prediction Based on Mammographic data by Hybrid Resnet and Decision Tree. International Conference on Sustainable Computing and Data Communication Systems. Erode, India . 2022; pp. 234-8.
[http://dx.doi.org/10.1109/ICSCDS53736.2022.9760754]
[25]
Ghosal P, Nandanwar L, Kanchan S. Brain Tumor Classification Using ResNet-101 Based Squeeze and Excitation Deep Neural Network. International Conference on Advanced Computational and Communication Paradigms. Gangtok, India . 2019; pp. 1-6.
[http://dx.doi.org/10.1109/ICACCP.2019.8882973]
[26]
Özyurt F. Automatic Detection of COVID-19 Disease by Using Transfer Learning of Light Weight Deep Learning Model. TS Traitement Signal 2021; 38(1): 147-53.
[http://dx.doi.org/10.18280/ts.380115]
[27]
Wiggin J. Gamma Correction in Live Color TV Cameras. IEEE Trans Broadcast 1968; BC-14(1): 8-13.
[http://dx.doi.org/10.1109/TBC.1968.265945]
[28]
Huang SC, Cheng FC, Chiu YS. Efficient contrast enhancement using adaptive gamma correction with weighting distribution. IEEE Trans Image Process 2013; 22(3): 1032-41.
[http://dx.doi.org/10.1109/TIP.2012.2226047] [PMID: 23144035]
[29]
Matic Z, Kadry S. Tumor Segmentation in Breast MRI Using Deep Learning. 2022 Fifth International Conference of Women in Data Science at Prince Sultan University (WiDS PSU). Riyadh, Saudi Arabia . 2022.
[http://dx.doi.org/10.1109/WiDS-PSU54548.2022.00021]
[30]
Boudouh SS, Bouakkaz M. Breast Cancer: Breast Tumor Detection Using Deep Transfer Learning Techniques in Mammogram Images. 2022 International Conference on Computer Science and Software Engineering 2022. Duhok, Iraq. . 2022; pp. 289-94.
[http://dx.doi.org/10.1109/CSASE51777.2022.9759702]
[31]
Zhang X, Zhou X, Lin M. ShuffleNet: An Extremely Efficient Convolutional Neural Network for Mobile Devices. 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition. Salt Lake City, UT, USA. 2018; pp. 6848-56.
[http://dx.doi.org/10.1109/CVPR.2018.00716]
[32]
Deng J, Dong W, Socher R. ImageNet: A large-scale hierarchical image database. IEEE Computer Society Conference on Computer Vision and Pattern Recognition. Florida. 2009; pp. 248-55.
[http://dx.doi.org/10.1109/CVPR.2009.5206848]
[33]
Kingma DP, Ba J. Adam: A method for stochastic optimization. arXiv:14126980
[http://dx.doi.org/10.48550/arXiv.1412.6980]
[34]
Ding Z, Zhao Y, Li A, Zheng Z. Spatial–Temporal Attention Two-Stream Convolution Neural Network for Smoke Region Detection. Fire (Basel) 2021; 4(4): 66.
[http://dx.doi.org/10.3390/fire4040066]

Rights & Permissions Print Cite
© 2024 Bentham Science Publishers | Privacy Policy