An Automated In-Depth Feature Learning Algorithm for Breast Abnormality Prognosis and Robust Characterization from Mammography Images Using Deep Transfer Learning

被引:32
|
作者
Mahmood, Tariq [1 ,2 ]
Li, Jianqiang [1 ,3 ]
Pei, Yan [4 ]
Akhtar, Faheem [5 ]
机构
[1] Beijing Univ Technol, Sch Software Engn, Beijing 100024, Peoples R China
[2] Univ Educ, Div Sci & Technol, Lahore 54000, Pakistan
[3] Beijing Engn Res Ctr IoT Software & Syst, Beijing 100124, Peoples R China
[4] Univ Aizu, Comp Sci Div, Aizu Wakamatsu, Fukushima 9658580, Japan
[5] Sukkur IBA Univ, Dept Comp Sci, Sukkur 65200, Pakistan
来源
BIOLOGY-BASEL | 2021年 / 10卷 / 09期
基金
国家重点研发计划;
关键词
breast cancer mass; deep learning; mammography classification; deep transfer learning; augmentation; computer-aided diagnosis; COMPUTER-AIDED DIAGNOSIS; CLASSIFICATION; MASSES;
D O I
10.3390/biology10090859
中图分类号
Q [生物科学];
学科分类号
07 ; 0710 ; 09 ;
摘要
Simple Summary Diagnosing breast cancer masses and calcification clusters is crucial in mammography, which reduces disease consequences and initiates treatment at an early stage. A misinterpretation of mammography may lead to an unneeded biopsy of the false-positive results, decreasing the patient's chances of survival. This study aims to increase the probability of early breast mass identification to ensure better treatment and minimize mortality risk. However, this study proposes a deep learning method based on convolutional neural networks to extract features of varying densities and classify normal and suspicious mammography regions. Two different experiments were carried out to validate the consistency of diagnoses and classification. The first experiment consisted of five end-to-end pre-trained and fine-tuned deep convolution neural networks. Additionally, the deep features extracted are used to train the support vector machine algorithm, resulting in an outstanding performance in the second experiment. Furthermore, this study confirms an improvement in mass recognition accuracy through data cleaning, preprocessing, and augmentation. Our deep learning hybrid model obtained a classification accuracy of 97.8%, outperforming the current state-of-the-art approaches. The proposed model's improvements are appropriated in conventional pathological practices that conceivably reduce the pathologist's strain in predicting clinical outcomes by analyzing patients' mammography images. Background: Diagnosing breast cancer masses and calcification clusters have paramount significance in mammography, which aids in mitigating the disease's complexities and curing it at early stages. However, a wrong mammogram interpretation may lead to an unnecessary biopsy of the false-positive findings, which reduces the patient's survival chances. Consequently, approaches that learn to discern breast masses can reduce the number of misconceptions and incorrect diagnoses. Conventionally used classification models focus on feature extraction techniques specific to a particular problem based on domain information. Deep learning strategies are becoming promising alternatives to solve the many challenges of feature-based approaches. Methods: This study introduces a convolutional neural network (ConvNet)-based deep learning method to extract features at varying densities and discern mammography's normal and suspected regions. Two different experiments were carried out to make an accurate diagnosis and classification. The first experiment consisted of five end-to-end pre-trained and fine-tuned deep convolution neural networks (DCNN). The in-depth features extracted from the ConvNet are also used to train the support vector machine algorithm to achieve excellent performance in the second experiment. Additionally, DCNN is the most frequently used image interpretation and classification method, including VGGNet, GoogLeNet, MobileNet, ResNet, and DenseNet. Moreover, this study pertains to data cleaning, preprocessing, and data augmentation, and improving mass recognition accuracy. The efficacy of all models is evaluated by training and testing three mammography datasets and has exhibited remarkable results. Results: Our deep learning ConvNet+SVM model obtained a discriminative training accuracy of 97.7% and validating accuracy of 97.8%, contrary to this, VGGNet16 method yielded 90.2%, 93.5% for VGGNet19, 63.4% for GoogLeNet, 82.9% for MobileNetV2, 75.1% for ResNet50, and 72.9% for DenseNet121. Conclusions: The proposed model's improvement and validation are appropriated in conventional pathological practices that conceivably reduce the pathologist's strain in predicting clinical outcomes by analyzing patients' mammography images.
引用
收藏
页数:25
相关论文
共 50 条
  • [1] Application of deep transfer learning for automated brain abnormality classification using MR images
    Talo, Muhammed
    Baloglu, Ulas Baran
    Yildirim, Ozal
    Acharya, U. Rajendra
    [J]. COGNITIVE SYSTEMS RESEARCH, 2019, 54 : 176 - 188
  • [2] Advanced feature learning and classification of microscopic breast abnormalities using a robust deep transfer learning technique
    Rehman, Amjad
    Mahmood, Tariq
    Alamri, Faten S.
    Saba, Tanzila
    Naseem, Shahid
    [J]. MICROSCOPY RESEARCH AND TECHNIQUE, 2024, 87 (08) : 1862 - 1888
  • [3] Reducing overfitting of a deep learning breast mass detection algorithm in mammography using synthetic images
    Cha, Kenny H.
    Petrick, Nicholas
    Pezeshk, Aria
    Graff, Christian G.
    Sharma, Diksha
    Badal, Andreu
    Badano, Aldo
    Sahiner, Berkman
    [J]. MEDICAL IMAGING 2019: COMPUTER-AIDED DIAGNOSIS, 2019, 10950
  • [4] Breast Cancer Detection from Histopathological Images using Deep Learning and Transfer Learning
    Muntean, Cristina H.
    Chowkkar, Mansi
    [J]. PROCEEDINGS OF 2022 7TH INTERNATIONAL CONFERENCE ON MACHINE LEARNING TECHNOLOGIES, ICMLT 2022, 2022, : 164 - 169
  • [5] Deep learning to calculate breast density from processed mammography images
    Warren, Lucy M.
    Harris, Peter
    Gomes, Sandra
    Trumble, Matthew
    Halling-Brown, Mark D.
    Dance, David R.
    Wilkinson, Louise
    Given-Wilson, Ro
    Young, Kenneth C.
    [J]. 15TH INTERNATIONAL WORKSHOP ON BREAST IMAGING (IWBI2020), 2020, 11513
  • [6] MAMMOGRAPHY IMAGE BREAST CANCER DETECTION USING DEEP TRANSFER LEARNING
    Krishna, Timmana Hari
    Rajabhushanam, C.
    [J]. ADVANCES AND APPLICATIONS IN MATHEMATICAL SCIENCES, 2021, 20 (07): : 1187 - 1196
  • [7] Deep Reinforcement Learning With Adversarial Training for Automated Excavation Using Depth Images
    Osa, Takayuki
    Aizawa, Masanori
    [J]. IEEE ACCESS, 2022, 10 : 4523 - 4535
  • [8] Deep Learning Algorithm for Atomization Characterization using Shadowgraph Images
    Narayanan, Barath Narayanan
    Gunasekaran, Sidaard
    Ivarson, Joseph
    Maneck, Lars
    [J]. PROCEEDINGS OF THE 2021 IEEE NATIONAL AEROSPACE AND ELECTRONICS CONFERENCE (NAECON), 2021, : 74 - 79
  • [9] Benchmarking of deep learning irradiance forecasting models from sky images - An in-depth analysis
    Paletta, Quentin
    Arbod, Guillaume
    Lasenby, Joan
    [J]. SOLAR ENERGY, 2021, 224 : 855 - 867
  • [10] Classification of Breast Cancer from Digital Mammography Using Deep Learning
    Daniel Lopez-Cabrera, Jose
    Lopez Rodriguez, Luis Alberto
    Perez-Diaz, Marlen
    [J]. INTELIGENCIA ARTIFICIAL-IBEROAMERICAN JOURNAL OF ARTIFICIAL INTELLIGENCE, 2020, 23 (65): : 56 - 66