Academic Journal
Improving Performance of Breast Lesion Classification Using a ResNet50 Model Optimized with a Novel Attention Mechanism
| العنوان: | Improving Performance of Breast Lesion Classification Using a ResNet50 Model Optimized with a Novel Attention Mechanism |
|---|---|
| المؤلفون: | Warid Islam, Meredith Jones, Rowzat Faiz, Negar Sadeghipour, Yuchen Qiu, Bin Zheng |
| المصدر: | Tomography, Vol 8, Iss 5, Pp 2411-2425 (2022) |
| بيانات النشر: | MDPI AG, 2022. |
| سنة النشر: | 2022 |
| المجموعة: | LCC:Computer applications to medicine. Medical informatics |
| مصطلحات موضوعية: | computer-aided diagnosis (CAD) scheme of mammograms, convolutional block attention module (CBAM), residual network (ResNet), breast lesion classification, Computer applications to medicine. Medical informatics, R858-859.7 |
| الوصف: | Background: The accurate classification between malignant and benign breast lesions detected on mammograms is a crucial but difficult challenge for reducing false-positive recall rates and improving the efficacy of breast cancer screening. Objective: This study aims to optimize a new deep transfer learning model by implementing a novel attention mechanism in order to improve the accuracy of breast lesion classification. Methods: ResNet50 is selected as the base model to develop a new deep transfer learning model. To enhance the accuracy of breast lesion classification, we propose adding a convolutional block attention module (CBAM) to the standard ResNet50 model and optimizing a new model for this task. We assembled a large dataset with 4280 mammograms depicting suspicious soft-tissue mass-type lesions. A region of interest (ROI) is extracted from each image based on lesion center. Among them, 2480 and 1800 ROIs depict verified benign and malignant lesions, respectively. The image dataset is randomly split into two subsets with a ratio of 9:1 five times to train and test two ResNet50 models with and without using CBAM. Results: Using the area under ROC curve (AUC) as an evaluation index, the new CBAM-based ResNet50 model yields AUC = 0.866 ± 0.015, which is significantly higher than that obtained by the standard ResNet50 model (AUC = 0.772 ± 0.008) (p < 0.01). Conclusion: This study demonstrates that although deep transfer learning technology attracted broad research interest in medical-imaging informatic fields, adding a new attention mechanism to optimize deep transfer learning models for specific application tasks can play an important role in further improving model performances. |
| نوع الوثيقة: | article |
| وصف الملف: | electronic resource |
| اللغة: | English |
| تدمد: | 2379-139X 2379-1381 |
| Relation: | https://www.mdpi.com/2379-139X/8/5/200; https://doaj.org/toc/2379-1381; https://doaj.org/toc/2379-139X |
| DOI: | 10.3390/tomography8050200 |
| URL الوصول: | https://doaj.org/article/c6d81fbe5cc44e08940f512b20b51c44 |
| رقم الانضمام: | edsdoj.6d81fbe5cc44e08940f512b20b51c44 |
| قاعدة البيانات: | Directory of Open Access Journals |
Full Text Finder | تدمد: | 2379139X 23791381 |
|---|---|
| DOI: | 10.3390/tomography8050200 |