Detection of Masses in Mammogram Images Based on the Enhanced RetinaNet Network With INbreast Dataset
Mingzhao Wang,1 Ran Liu,1 Joseph Luttrell IV,2 Chaoyang Zhang,2 Juanying Xie1 1School of Computer Science, Shaanxi Normal University, Xian, People’s Republic of China; 2School of Computing Sciences and Computer Engineering, University of Southern Mississippi, Hattiesburg, MS, USACorrespondence: Juan...
Saved in:
| Main Authors: | , , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
Dove Medical Press
2025-02-01
|
| Series: | Journal of Multidisciplinary Healthcare |
| Subjects: | |
| Online Access: | https://www.dovepress.com/detection-of-masses-in-mammogram-images-based-on-the-enhanced-retinane-peer-reviewed-fulltext-article-JMDH |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1823861392425353216 |
|---|---|
| author | Wang M Liu R Luttrell IV J Zhang C Xie J |
| author_facet | Wang M Liu R Luttrell IV J Zhang C Xie J |
| author_sort | Wang M |
| collection | DOAJ |
| description | Mingzhao Wang,1 Ran Liu,1 Joseph Luttrell IV,2 Chaoyang Zhang,2 Juanying Xie1 1School of Computer Science, Shaanxi Normal University, Xian, People’s Republic of China; 2School of Computing Sciences and Computer Engineering, University of Southern Mississippi, Hattiesburg, MS, USACorrespondence: Juanying Xie, School of Computer Science, Shaanxi Normal University, No. 620, West Chang’an Avenue, Chang’an District, Xi’an, 710119, Shaanxi, People’s Republic of China, Tel +86 13088965815, Email [email protected] Chaoyang Zhang, School of Computing Sciences and Computer Engineering, University of Southern Mississippi, 118 College Drive, Hattiesburg, MS, 39406-0001, USA, Email [email protected]: Breast cancer is the most common major public health problems of women in the world. Until now, analyzing mammogram images is still the main method used by doctors to diagnose and detect breast cancers. However, this process usually depends on the experience of radiologists and is always very time consuming.Patients and Methods: We propose to introduce deep learning technology into the process for the facilitation of computer-aided diagnosis (CAD), and address the challenges of class imbalance, enhance the detection of small masses and multiple targets, and reduce false positives and negatives in mammogram analysis. Therefore, we adopted and enhanced RetinaNet to detect masses in mammogram images. Specifically, we introduced a novel modification to the network structure, where the feature map M5 is processed by the ReLU function prior to the original convolution kernel. This strategic adjustment was designed to prevent the loss of resolution for small mass features. Additionally, we introduced transfer learning techniques into training process through leveraging pre-trained weights from other RetinaNet applications, and fine-tuned our improved model using the INbreast dataset.Results: The aforementioned innovations facilitate superior performance of the enhanced RetiaNet model on the public dataset INbreast, as evidenced by a mAP (mean average precision) of 1.0000 and TPR (true positive rate) of 1.00 at 0.00 FPPI (false positive per image) on the INbreast dataset.Conclusion: The experimental results demonstrate that our enhanced RetinaNet model defeats the existing models by having more generalization performance than other published studies, and it can also be applied to other types of patients to assist doctors in making a proper diagnosis. Keywords: computer-aided diagnosis, deep learning, object detection, RetinaNet, transfer learning |
| format | Article |
| id | doaj-art-081a18bd68384d92b0fd759dee161a2a |
| institution | Kabale University |
| issn | 1178-2390 |
| language | English |
| publishDate | 2025-02-01 |
| publisher | Dove Medical Press |
| record_format | Article |
| series | Journal of Multidisciplinary Healthcare |
| spelling | doaj-art-081a18bd68384d92b0fd759dee161a2a2025-02-09T16:10:20ZengDove Medical PressJournal of Multidisciplinary Healthcare1178-23902025-02-01Volume 1867569599986Detection of Masses in Mammogram Images Based on the Enhanced RetinaNet Network With INbreast DatasetWang MLiu RLuttrell IV JZhang CXie JMingzhao Wang,1 Ran Liu,1 Joseph Luttrell IV,2 Chaoyang Zhang,2 Juanying Xie1 1School of Computer Science, Shaanxi Normal University, Xian, People’s Republic of China; 2School of Computing Sciences and Computer Engineering, University of Southern Mississippi, Hattiesburg, MS, USACorrespondence: Juanying Xie, School of Computer Science, Shaanxi Normal University, No. 620, West Chang’an Avenue, Chang’an District, Xi’an, 710119, Shaanxi, People’s Republic of China, Tel +86 13088965815, Email [email protected] Chaoyang Zhang, School of Computing Sciences and Computer Engineering, University of Southern Mississippi, 118 College Drive, Hattiesburg, MS, 39406-0001, USA, Email [email protected]: Breast cancer is the most common major public health problems of women in the world. Until now, analyzing mammogram images is still the main method used by doctors to diagnose and detect breast cancers. However, this process usually depends on the experience of radiologists and is always very time consuming.Patients and Methods: We propose to introduce deep learning technology into the process for the facilitation of computer-aided diagnosis (CAD), and address the challenges of class imbalance, enhance the detection of small masses and multiple targets, and reduce false positives and negatives in mammogram analysis. Therefore, we adopted and enhanced RetinaNet to detect masses in mammogram images. Specifically, we introduced a novel modification to the network structure, where the feature map M5 is processed by the ReLU function prior to the original convolution kernel. This strategic adjustment was designed to prevent the loss of resolution for small mass features. Additionally, we introduced transfer learning techniques into training process through leveraging pre-trained weights from other RetinaNet applications, and fine-tuned our improved model using the INbreast dataset.Results: The aforementioned innovations facilitate superior performance of the enhanced RetiaNet model on the public dataset INbreast, as evidenced by a mAP (mean average precision) of 1.0000 and TPR (true positive rate) of 1.00 at 0.00 FPPI (false positive per image) on the INbreast dataset.Conclusion: The experimental results demonstrate that our enhanced RetinaNet model defeats the existing models by having more generalization performance than other published studies, and it can also be applied to other types of patients to assist doctors in making a proper diagnosis. Keywords: computer-aided diagnosis, deep learning, object detection, RetinaNet, transfer learninghttps://www.dovepress.com/detection-of-masses-in-mammogram-images-based-on-the-enhanced-retinane-peer-reviewed-fulltext-article-JMDHcomputer-aided diagnosisdeep learningobject detectionretinanettransfer learning |
| spellingShingle | Wang M Liu R Luttrell IV J Zhang C Xie J Detection of Masses in Mammogram Images Based on the Enhanced RetinaNet Network With INbreast Dataset Journal of Multidisciplinary Healthcare computer-aided diagnosis deep learning object detection retinanet transfer learning |
| title | Detection of Masses in Mammogram Images Based on the Enhanced RetinaNet Network With INbreast Dataset |
| title_full | Detection of Masses in Mammogram Images Based on the Enhanced RetinaNet Network With INbreast Dataset |
| title_fullStr | Detection of Masses in Mammogram Images Based on the Enhanced RetinaNet Network With INbreast Dataset |
| title_full_unstemmed | Detection of Masses in Mammogram Images Based on the Enhanced RetinaNet Network With INbreast Dataset |
| title_short | Detection of Masses in Mammogram Images Based on the Enhanced RetinaNet Network With INbreast Dataset |
| title_sort | detection of masses in mammogram images based on the enhanced retinanet network with inbreast dataset |
| topic | computer-aided diagnosis deep learning object detection retinanet transfer learning |
| url | https://www.dovepress.com/detection-of-masses-in-mammogram-images-based-on-the-enhanced-retinane-peer-reviewed-fulltext-article-JMDH |
| work_keys_str_mv | AT wangm detectionofmassesinmammogramimagesbasedontheenhancedretinanetnetworkwithinbreastdataset AT liur detectionofmassesinmammogramimagesbasedontheenhancedretinanetnetworkwithinbreastdataset AT luttrellivj detectionofmassesinmammogramimagesbasedontheenhancedretinanetnetworkwithinbreastdataset AT zhangc detectionofmassesinmammogramimagesbasedontheenhancedretinanetnetworkwithinbreastdataset AT xiej detectionofmassesinmammogramimagesbasedontheenhancedretinanetnetworkwithinbreastdataset |