Acta Informatica Pragensia X:X | DOI: 10.18267/j.aip.32515
Mask-Prior Knowledge Distillation for Efficient Binary Breast Ultrasound Classification
- Faculty of Technology and Design, Universitas Ma Chung, Malang, Indonesia
Background: Binary classification of breast ultrasound (BUS) images into benign and malignant categories is a clinically critical task. Standard knowledge distillation (KD) compresses large teacher models into compact student models but transfers knowledge in a spatially blind manner, leaving ground-truth segmentation masks, routinely available in BUS datasets, entirely unexploited.
Objective: This study proposes mask-prior knowledge distillation (MP-KD), a novel distillation framework that integrates ground-truth lesion segmentation masks as spatial priors to improve both classification performance and attention interpretability in a lightweight student model.
Methods: A multi-teacher ensemble of EfficientNet-B4 and ResNet50, pre-trained and fine-tuned on the Breast Ultrasound Image (BUSI) dataset (697 images; 487 benign, 210 malignant), generates soft targets and attention guidance distilled into an EfficientNet-B0 student (5.3M parameters). The proposed mask-weighted attention transfer (MWAT) loss combines two components: MWAT-AT, which applies mask-derived spatial weights to the attention transfer loss to amplify distillation signals within lesion regions; and MWAT-MA, an auxiliary mask alignment objective that directly supervises student attention towards the ground-truth mask. Loss weights were selected via grid search (18 combinations) on the validation set. All the experiments were repeated with three random seeds (42, 123, 456) on a stratified 70/10/20 split; performance is reported as mean ± standard deviation on the held-out test set.
Results: MP-KD achieves the highest AUC of 0.954 ± 0.004 among all the student configurations, outperforming standard KD (0.944 ± 0.002) and the no-KD baseline (0.949 ± 0.004). MP-KD also yields the highest mask-attention alignment score among all the distillation-based configurations (MAAS = 0.608 ± 0.006), substantially recovering the attention alignment lost during standard KD (0.577) while maintaining superior classification performance. Ablation studies show that MWAT-AT primarily drives accuracy (0.908) and specificity (0.921) improvements, while MWAT-MA yields the highest single-run sensitivity (0.952). The combined MP-KD model provides the most robust performance across all the seeds (MCC = 0.732 ± 0.007, AUC std = 0.004, smallest among all the configurations). Compared to the distilled ViT approach of Zhou and Mosadegh (AUC = 0.972, ~86M parameters), MP-KD achieves competitive performance with 16× fewer parameters.
Conclusion: MP-KD demonstrates that ground-truth segmentation masks can be effectively integrated as spatial priors within the knowledge distillation pipeline, improving both the AUC and attention alignment of a compact student model compared to standard KD. The proposed MWAT mechanism offers a practical and interpretable solution for deploying breast cancer classification in resource-constrained clinical environments and is generalizable to other medical imaging tasks where segmentation annotations are available.
Keywords: Breast ultrasound; Knowledge distillation; Attention transfer; Binary classification; EfficientNet; Segmentation mask; Deep learning; Medical imaging.
Received: April 14, 2026; Revised: July 14, 2026; Accepted: July 29, 2026; Prepublished online: November 8, 2026
References
- Al-Dhabyani, W., Gomaa, M., Khaled, H., & Fahmy, A. (2020). Dataset of breast ultrasound images. Data in Brief, 28, 104863. https://doi.org/10.1016/j.dib.2019.104863
Go to original source... - Ali, M. D., Saleem, A., Elahi, H., Khan, M. A., Khan, M. I., Yaqoob, M. M., Farooq Khattak, U., & Al-Rasheed, A. (2023). Breast Cancer Classification through Meta-Learning Ensemble Technique Using Convolution Neural Networks. Diagnostics, 13(13), 2242. https://doi.org/10.3390/diagnostics13132242
Go to original source... - Aumente-Maestro, C., Díez, J., & Remeseiro, B. (2025). A multi-task framework for breast cancer segmentation and classification in ultrasound imaging. Computer Methods and Programs in Biomedicine, 260, 108540. https://doi.org/10.1016/j.cmpb.2024.108540
Go to original source... - Burnside, E. S., Sickles, E. A., Bassett, L. W., Rubin, D. L., Lee, C. H., Ikeda, D. M., Mendelson, E. B., Wilcox, P. A., Butler, P. F., & D'Orsi, C. J. (2009). The ACR BI-RADS® Experience: Learning from History. Journal of the American College of Radiology, 6(12), 851-860. https://doi.org/10.1016/j.jacr.2009.07.023
Go to original source... - Buslaev, A., Iglovikov, V. I., Khvedchenya, E., Parinov, A., Druzhinin, M., & Kalinin, A. A. (2020). Albumentations: Fast and Flexible Image Augmentations. Information, 11(2), 125. https://doi.org/10.3390/info11020125
Go to original source... - Dan, Q., Xu, Z., Burrows, H., Bissram, J., Stringer, J. S. A., & Li, Y. (2024). Diagnostic performance of deep learning in ultrasound diagnosis of breast cancer: a systematic review. Npj Precision Oncology, 8(1), 21. https://doi.org/10.1038/s41698-024-00514-z
Go to original source... - Gheflati, B., & Rivaz, H. (2022). Vision Transformers for Classification of Breast Ultrasound Images. In 2022 44th Annual International Conference of the IEEE Engineering in Medicine & Biology Society (EMBC), (pp. 480-483). IEEE. https://doi.org/10.1109/EMBC48229.2022.9871809
Go to original source... - Gou, J., Yu, B., Maybank, S. J., & Tao, D. (2021). Knowledge Distillation: A Survey. International Journal of Computer Vision, 129(6), 1789-1819. https://doi.org/10.1007/s11263-021-01453-z
Go to original source... - He, K., Zhang, X., Ren, S., & Sun, J. (2016). Deep Residual Learning for Image Recognition. In 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), (pp. 770-778). IEEE. https://doi.org/10.1109/CVPR.2016.90
Go to original source... - Hinton, G., Vinyals, O., & Dean, J. (2015). Distilling the Knowledge in a Neural Network. arXiv:1503.02531. https://doi.org/10.48550/arXiv.1503.02531
Go to original source... - Jabeen, K., Khan, M. A., Alhaisoni, M., Tariq, U., Zhang, Y.-D., Hamza, A., Mickus, A., & Damaąevičius, R. (2022). Breast Cancer Classification from Ultrasound Images Using Probability-Based Optimal Deep Learning Feature Fusion. Sensors, 22(3), 807. https://doi.org/10.3390/s22030807
Go to original source... - Latha, M., Kumar, P. S., Chandrika, R. R., Mahesh, T. R., Kumar, V. V., & Guluwadi, S. (2024). Revolutionizing breast ultrasound diagnostics with EfficientNet-B7 and Explainable AI. BMC Medical Imaging, 24(1), 230. https://doi.org/10.1186/s12880-024-01404-3
Go to original source... - Liu, L., Wang, Z., Phan, M. H., Zhang, B., Ge, J., & Liu, Y. (2024). BPKD: Boundary Privileged Knowledge Distillation for Semantic Segmentation. In 2024 IEEE/CVF Winter Conference on Applications of Computer Vision (WACV). IEEE. https://doi.org/10.1109/WACV57701.2024.00110
Go to original source... - Liu, Y., Chen, K., Liu, C., Qin, Z., Luo, Z., & Wang, J. (2019). Structured Knowledge Distillation for Semantic Segmentation. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, (pp. 2604-2613). IEEE. https://doi.org/10.1109/CVPR.2019.00271
Go to original source... - Mansourian, A. M., Jalali, A., Ahmadi, R., & Kasaei, S. (2025). Attention-guided Feature Distillation for Semantic Segmentation. arXiv:2403.05451. https://doi.org/10.48550/arXiv.2403.05451
Go to original source... - Noothout, J. M. H., Lessmann, N., van Eede, M. C., van Harten, L. D., Sogancioglu, E., Heslinga, F. G., Veta, M., van Ginneken, B., & Iągum, I. (2022). Knowledge distillation with ensembles of convolutional neural networks for medical image segmentation. Journal of Medical Imaging, 9(5), 052407. https://doi.org/10.1117/1.JMI.9.5.052407
Go to original source... - Park, W., Kim, D., Lu, Y., & Cho, M. (2019). Relational Knowledge Distillation. In 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition, (pp. 3962-3971). IEEE. https://doi.org/10.1109/CVPR.2019.00409
Go to original source... - Pawłowska, A., Karwat, P., & Żołek, N. (2023). Letter to the Editor. Re: "[Dataset of breast ultrasound images by W. Al-Dhabyani, M. Gomaa, H. Khaled & A. Fahmy, Data in Brief, 2020, 28, 104863]." Data in Brief, 48, 109247. https://doi.org/10.1016/j.dib.2023.109247
Go to original source... - Qin, D., Bu, J.-J., Liu, Z., Shen, X., Zhou, S., Gu, J.-J., Wang, Z.-H., Wu, L., & Dai, H.-F. (2021). Efficient Medical Image Segmentation Based on Knowledge Distillation. IEEE Transactions on Medical Imaging, 40(12), 3820-3831. https://doi.org/10.1109/TMI.2021.3098703
Go to original source... - Romero, A., Ballas, N., Kahou, S. E., Chassang, A., Gatta, C., & Bengio, Y. (2015). FitNets: Hints for Thin Deep Nets. In ICLR 2015, 3rd International Conference on Learning Representations. ICLR.
- Selvaraju, R. R., Cogswell, M., Das, A., Vedantam, R., Parikh, D., & Batra, D. (2017). Grad-CAM: Visual Explanations from Deep Networks via Gradient-Based Localization. In 2017 IEEE International Conference on Computer Vision (ICCV), (pp. 618-626). IEEE. https://doi.org/10.1109/ICCV.2017.74
Go to original source... - Sepahvand, M., & Abdali-Mohammadi, F. (2023). Joint learning method with teacher-student knowledge distillation for on-device breast cancer image classification. Computers in Biology and Medicine, 155, 106476. https://doi.org/10.1016/j.compbiomed.2022.106476
Go to original source... - Stavros, A. T., Thickman, D., Rapp, C. L., Dennis, M. A., Parker, S. H., & Sisney, G. A. (1995). Solid breast nodules: use of sonography to distinguish between benign and malignant lesions. Radiology, 196(1), 123-134. https://doi.org/10.1148/radiology.196.1.7784555
Go to original source... - Sung, H., Ferlay, J., Siegel, R. L., Laversanne, M., Soerjomataram, I., Jemal, A., & Bray, F. (2021). Global Cancer Statistics 2020: GLOBOCAN Estimates of Incidence and Mortality Worldwide for 36 Cancers in 185 Countries. CA: A Cancer Journal for Clinicians, 71(3), 209-249. https://doi.org/10.3322/caac.21660
Go to original source... - Tan, M., & Le, Q. V. (2019). EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks. In Proceedings of the 36th International Conference on Machine Learning, (pp. 6105-6114). PMLR.
- Vallez, N., Bueno, G., Deniz, O., Rienda, M. A., & Pastor, C. (2025). BUS-UCLM: Breast ultrasound lesion segmentation dataset. Scientific Data, 12(1), 242. https://doi.org/10.1038/s41597-025-04562-3
Go to original source... - Vapnik, V., & Vashist, A. (2009). A new learning paradigm: Learning using privileged information. Neural Networks, 22(5-6), 544-557. https://doi.org/10.1016/j.neunet.2009.06.042
Go to original source... - Zagoruyko, S., & Komodakis, N. (2017). Paying More Attention to Attention: Improving the Performance of Convolutional Neural Networks via Attention Transfer. In ICLR 2017, 5th International Conference on Learning Representations. ICLR.
- Zhang, J., Wu, J., Zhou, X. S., Shi, F., & Shen, D. (2023). Recent advancements in artificial intelligence for breast cancer: Image augmentation, segmentation, diagnosis, and prognosis approaches. Seminars in Cancer Biology, 96, 11-25. https://doi.org/10.1016/j.semcancer.2023.09.001
Go to original source... - Zhao, L., Qian, X., Guo, Y., Song, J., Hou, J., & Gong, J. (2023). MSKD: Structured knowledge distillation for efficient medical image segmentation. Computers in Biology and Medicine, 164, 107284. https://doi.org/10.1016/j.compbiomed.2023.107284
Go to original source... - Zhou, G., & Mosadegh, B. (2024). Distilling Knowledge from an Ensemble of Vision Transformers for Improved Classification of Breast Ultrasound. Academic Radiology, 31(1), 104-120. https://doi.org/10.1016/j.acra.2023.08.006
Go to original source...
This is an open access article distributed under the terms of the Creative Commons Attribution 4.0 International License (CC BY 4.0), which permits use, distribution, and reproduction in any medium, provided the original publication is properly cited. No use, distribution or reproduction is permitted which does not comply with these terms.

ORCID...