Segmentation for mammography classification utilizing deep convolutional neural network
Abstract Background Mammography for the diagnosis of early breast cancer (BC) relies heavily on the identification of breast masses. However, in the early stages, it might be challenging to ascertain whether a breast mass is benign or malignant. Consequently, many deep learning (DL)-based computer-a...
Saved in:
| Main Authors: | , , , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
BMC
2024-12-01
|
| Series: | BMC Medical Imaging |
| Subjects: | |
| Online Access: | https://doi.org/10.1186/s12880-024-01510-2 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1846111954146426880 |
|---|---|
| author | Dip Kumar Saha Tuhin Hossain Mejdl Safran Sultan Alfarhood M. F. Mridha Dunren Che |
| author_facet | Dip Kumar Saha Tuhin Hossain Mejdl Safran Sultan Alfarhood M. F. Mridha Dunren Che |
| author_sort | Dip Kumar Saha |
| collection | DOAJ |
| description | Abstract Background Mammography for the diagnosis of early breast cancer (BC) relies heavily on the identification of breast masses. However, in the early stages, it might be challenging to ascertain whether a breast mass is benign or malignant. Consequently, many deep learning (DL)-based computer-aided diagnosis (CAD) approaches for BC classification have been developed. Methods Recently, the transformer model has emerged as a method for overcoming the constraints of convolutional neural networks (CNN). Thus, our primary goal was to determine how well an improved transformer model could distinguish between benign and malignant breast tissues. In this instance, we drew on the Mendeley data repository’s INbreast dataset, which includes benign and malignant breast types. Additionally, the segmentation anything model (SAM) method was used to generate the optimized cutoff for region of interest (ROI) extraction from all mammograms. We implemented a successful architecture modification at the bottom layer of a pyramid transformer (PTr) to identify BC from mammography images. Results The proposed PTr model using a transfer learning (TL) approach with a segmentation technique achieved the best accuracy of 99.96% for binary classifications with an area under the curve (AUC) score of 99.98%, respectively. We also compared the performance of the proposed model with other transformer model vision transformers (ViT) and DL models, MobileNetV3 and EfficientNetB7, respectively. Conclusions In this study, a modified transformer model is proposed for BC prediction and mammography image classification using segmentation approaches. Data segmentation techniques accurately identify the regions affected by BC. Finally, the proposed transformer model accurately classified benign and malignant breast tissues, which is vital for radiologists to guide future treatment. |
| format | Article |
| id | doaj-art-11e6aafb6b1346a0b69fcf6911c4bbab |
| institution | Kabale University |
| issn | 1471-2342 |
| language | English |
| publishDate | 2024-12-01 |
| publisher | BMC |
| record_format | Article |
| series | BMC Medical Imaging |
| spelling | doaj-art-11e6aafb6b1346a0b69fcf6911c4bbab2024-12-22T12:55:38ZengBMCBMC Medical Imaging1471-23422024-12-0124111810.1186/s12880-024-01510-2Segmentation for mammography classification utilizing deep convolutional neural networkDip Kumar Saha0Tuhin Hossain1Mejdl Safran2Sultan Alfarhood3M. F. Mridha4Dunren Che5Department of Computer Science and Engineering, Stamford University BangladeshDepartment of Computer Science and Engineering, Jahangirnagar UniversityDepartment of Computer Science, College of Computer and Information Sciences, King Saud UniversityDepartment of Computer Science, College of Computer and Information Sciences, King Saud UniversityDepartment of Computer Science, American International University-BangladeshDepartment of Electrical Engineering and Computer Science, Texas A&M University-KingsvilleAbstract Background Mammography for the diagnosis of early breast cancer (BC) relies heavily on the identification of breast masses. However, in the early stages, it might be challenging to ascertain whether a breast mass is benign or malignant. Consequently, many deep learning (DL)-based computer-aided diagnosis (CAD) approaches for BC classification have been developed. Methods Recently, the transformer model has emerged as a method for overcoming the constraints of convolutional neural networks (CNN). Thus, our primary goal was to determine how well an improved transformer model could distinguish between benign and malignant breast tissues. In this instance, we drew on the Mendeley data repository’s INbreast dataset, which includes benign and malignant breast types. Additionally, the segmentation anything model (SAM) method was used to generate the optimized cutoff for region of interest (ROI) extraction from all mammograms. We implemented a successful architecture modification at the bottom layer of a pyramid transformer (PTr) to identify BC from mammography images. Results The proposed PTr model using a transfer learning (TL) approach with a segmentation technique achieved the best accuracy of 99.96% for binary classifications with an area under the curve (AUC) score of 99.98%, respectively. We also compared the performance of the proposed model with other transformer model vision transformers (ViT) and DL models, MobileNetV3 and EfficientNetB7, respectively. Conclusions In this study, a modified transformer model is proposed for BC prediction and mammography image classification using segmentation approaches. Data segmentation techniques accurately identify the regions affected by BC. Finally, the proposed transformer model accurately classified benign and malignant breast tissues, which is vital for radiologists to guide future treatment.https://doi.org/10.1186/s12880-024-01510-2MammographyBreast cancerSegmentationClassificationSAM |
| spellingShingle | Dip Kumar Saha Tuhin Hossain Mejdl Safran Sultan Alfarhood M. F. Mridha Dunren Che Segmentation for mammography classification utilizing deep convolutional neural network BMC Medical Imaging Mammography Breast cancer Segmentation Classification SAM |
| title | Segmentation for mammography classification utilizing deep convolutional neural network |
| title_full | Segmentation for mammography classification utilizing deep convolutional neural network |
| title_fullStr | Segmentation for mammography classification utilizing deep convolutional neural network |
| title_full_unstemmed | Segmentation for mammography classification utilizing deep convolutional neural network |
| title_short | Segmentation for mammography classification utilizing deep convolutional neural network |
| title_sort | segmentation for mammography classification utilizing deep convolutional neural network |
| topic | Mammography Breast cancer Segmentation Classification SAM |
| url | https://doi.org/10.1186/s12880-024-01510-2 |
| work_keys_str_mv | AT dipkumarsaha segmentationformammographyclassificationutilizingdeepconvolutionalneuralnetwork AT tuhinhossain segmentationformammographyclassificationutilizingdeepconvolutionalneuralnetwork AT mejdlsafran segmentationformammographyclassificationutilizingdeepconvolutionalneuralnetwork AT sultanalfarhood segmentationformammographyclassificationutilizingdeepconvolutionalneuralnetwork AT mfmridha segmentationformammographyclassificationutilizingdeepconvolutionalneuralnetwork AT dunrenche segmentationformammographyclassificationutilizingdeepconvolutionalneuralnetwork |