Radiomics meets transformers: A novel approach to tumor segmentation and classification in mammography for breast cancer.
Authors
Affiliations (13)
Affiliations (13)
- Faculty of Pharmacy, Middle East University, Amman, Jordan.
- College of pharmacy, Alnoor University, Mosul, Iraq.
- Ahl al Bayt University, Kerbala, Iraq.
- Marwadi University Research Center, Department of Computer Engineering, Faculty of Engineering & Technology, Marwadi University, Rajkot, India.
- Department of Chemistry and Biochemistry, School of Sciences, JAIN (Deemed to be University), Bangalore, India.
- Centre for Research Impact & Outcome, Chitkara University Institute of Engineering and Technology, Chitkara University, Rajpura, India.
- Department of Chemistry, Sathyabama Institute of Science and Technology, Chennai, India.
- Department of Public Health and Healthcare management, Rector, Samarkand State Medical University, Samarkand, Uzbekistan.
- College of Nursing, National University of Science and Technology, Dhi Qar, Iraq.
- Pharmacy College, Al-Farahidi University, Baghdad, Iraq.
- Department of Pharmacy, Al-Zahrawi University College, Karbala, Iraq.
- Gilgamesh Ahliya University, Baghdad, Iraq.
- Department of Medical Physics and Radiology, Faculty of Paramedical Sciences, Kashan University of Medical Sciences, Kashan, Iran.
Abstract
ObjectiveThis study aimed to develop a robust framework for breast cancer diagnosis by integrating advanced segmentation and classification approaches. Transformer-based and U-Net segmentation models were combined with radiomic feature extraction and machine learning classifiers to improve segmentation precision and classification accuracy in mammographic images.Materials and MethodsA multi-center dataset of 8000 mammograms (4200 normal, 3800 abnormal) was used. Segmentation was performed using Transformer-based and U-Net models, evaluated through Dice Coefficient (DSC), Intersection over Union (IoU), Hausdorff Distance (HD95), and Pixel-Wise Accuracy. Radiomic features were extracted from segmented masks, with Recursive Feature Elimination (RFE) and Analysis of Variance (ANOVA) employed to select significant features. Classifiers including Logistic Regression, XGBoost, CatBoost, and a Stacking Ensemble model were applied to classify tumors into benign or malignant. Classification performance was assessed using accuracy, sensitivity, F1 score, and AUC-ROC. SHAP analysis validated feature importance, and Q-value heatmaps evaluated statistical significance.ResultsThe Transformer-based model achieved superior segmentation results with DSC (0.94 ± 0.01 training, 0.92 ± 0.02 test), IoU (0.91 ± 0.01 training, 0.89 ± 0.02 test), HD95 (3.0 ± 0.3 mm training, 3.3 ± 0.4 mm test), and Pixel-Wise Accuracy (0.96 ± 0.01 training, 0.94 ± 0.02 test), consistently outperforming U-Net across all metrics. For classification, Transformer-segmented features with the Stacking Ensemble achieved the highest test results: 93% accuracy, 92% sensitivity, 93% F1 score, and 95% AUC. U-Net-segmented features achieved lower metrics, with the best test accuracy at 84%. SHAP analysis confirmed the importance of features like Gray-Level Non-Uniformity and Zone Entropy.ConclusionThis study demonstrates the superiority of Transformer-based segmentation integrated with radiomic feature selection and robust classification models. The framework provides a precise and interpretable solution for breast cancer diagnosis, with potential for scalability to 3D imaging and multimodal datasets.