Attention-Enhanced Deep Learning Ensemble for Breast Density Classification in Mammography

2025-07-08 · Peyman Sharifian, Xiaotong Hong, Alireza Karimian, Mehdi Amini, Hossein Arabi arxiv

Breast density assessment is a crucial component of mammographic interpretation, with high breast density (BI-RADS categories C and D) representing both a significant risk factor for developing breast cancer and a technical challenge for tumor detection. This study proposes an automated deep learning system for robust binary classification of breast density (low: A/B vs. high: C/D) using the VinDr-Mammo dataset. We implemented and compared four advanced convolutional neural networks: ResNet18, ResNet50, EfficientNet-B0, and DenseNet121, each enhanced with channel attention mechanisms. To address the inherent class imbalance, we developed a novel Combined Focal Label Smoothing Loss function that integrates focal loss, label smoothing, and class-balanced weighting. Our preprocessing pipeline incorporated advanced techniques, including contrast-limited adaptive histogram equalization (CLAHE) and comprehensive data augmentation. The individual models were combined through an optimized ensemble voting approach, achieving superior performance (AUC: 0.963, F1-score: 0.952) compared to any single model. This system demonstrates significant potential to standardize density assessments in clinical practice, potentially improving screening efficiency and early cancer detection rates while reducing inter-observer variability among radiologists.

📄 PDF Abstract BibTeX arXiv:2507.06410

Code (0)

등록된 구현이 없습니다.

Tasks

Binary Classification Data Augmentation

Stacking-Enhanced Bagging Ensemble Learning for Breast Cancer Classification with CNN

2024-07-15 · Peihceng Wu, Runze Ma, Teoh Teik Toe

This paper proposes a CNN classification network based on Bagging and stacking ensemble learning methods for breast cancer classification. The model was trained and tested on the public dataset of DDSM. The model is capa…

Binary Classification Cancer Classification Classification Ensemble Learning

Attention-Guided Erasing: A Novel Augmentation Method for Enhancing Downstream Breast Density Classification

2024-01-08 · Adarsh Bhandary Panambur, Hui Yu, Sheethal Bhat, Prathmesh Madhu 외

The assessment of breast density is crucial in the context of breast cancer screening, especially in populations with a higher percentage of dense breast tissues. This study introduces a novel data augmentation technique…

breast density classification Data Augmentation Transfer Learning

Filtering for More Accurate Dense Tissue Segmentation in Digitized Mammograms

2013-10-01 · Mario Muštra, Mislav Grgić

Breast tissue segmentation into dense and fat tissue is important for determining the breast density in mammograms. Knowing the breast density is important both in diagnostic and computer-aided detection applications. Th…

Diagnostic General Classification Segmentation

Classification of Breast Cancer Lesions in Ultrasound Images by using Attention Layer and loss Ensembles in Deep Convolutional Neural Networks

2021-02-23 · Elham Yousef Kalaf, Ata Jodeiri, Seyed Kamaledin Setarehdan, Ng Wei Lin 외

Reliable classification of benign and malignant lesions in breast ultrasound images can provide an effective and relatively low cost method for early diagnosis of breast cancer. The accuracy of the diagnosis is however h…

Classification General Classification Transfer Learning

Breast Cancer classification by adaptive weighted average ensemble of previously trained models

2023-11-22 · Mosab S. M. Farea, Zhe Chen

Breast cancer is a serious disease that inflicts millions of people each year, and the number of cases is increasing. Early detection is the best way to reduce the impact of the disease. Researchers have developed many t…

Cancer Classification