LBNet: an optimized lightweight CNN for mammographic breast cancer classification with XAI-based interpretability

LBNet:一种基于可解释人工智能(XAI)的优化轻量级CNN,用于乳腺X线摄影乳腺癌分类。

阅读:2

Abstract

Breast cancer represents a major worldwide health burden, marked by high incidence and mortality rates across diverse socioeconomic populations. While deep learning has enabled advances in automated mammographic analysis, existing models often suffer from high computational complexity. They also face limited generalizability and a lack of interpretability. To overcome these challenges, this research introduces LBNet, a lightweight and interpretable convolutional neural network (CNN) built for accurate and efficient breast cancer detection, particularly in resource-constrained settings. With only 2.4 million trainable parameters, LBNet consists of five convolutional layers, leveraging ReLU activation, batch normalization, and max-pooling to optimize feature extraction while maintaining computational efficiency. LBNet was trained on the RSNA dataset using the Adam optimizer and five-fold cross-validation. It achieved 97.28% accuracy. For cancer cases, precision was 99% and recall was 96%. For non-cancer cases, precision was 96% and recall was 99%. In comparison, baseline models such as VGG19, SE-ResNet152, and ResNet152V2 yielded lower accuracies of 87.54%, 87.50%, and 85.24%, respectively, while transfer learning approaches peaked at 87.37% accuracy. LBNet's generalizability was validated in external datasets, achieving 99.54% accuracy on CBIS-DDSM and 98.50% on MIAS. To enhance clinical trust, this work integrated SHAP (SHapley Additive exPlanations) and Grad-CAM (Gradient-weighted Class Activation Mapping). These methods effectively highlighted diagnostically relevant regions in mammograms. This improved prediction transparency. LBNet demonstrates strong potential as an accurate, efficient, and interpretable solution for breast cancer screening, and future studies could explore its extension to multi-view mammography and real-time clinical deployment.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。