Classification of breast lesions in ultrasound images using deep convolutional neural networks: transfer learning versus automatic architecture design
Journal article
Authors | AlZoubi, A., Lu, F., Zhu, Y., Ying, T., Ahmed, M. and Du, H. |
---|---|
Abstract | Deep convolutional neural networks (DCNNs) have demonstrated promising performance in classifying breast lesions in 2D ultrasound (US) images. Exiting approaches typically use pre-trained models based on architectures designed for natural images with transfer learning. Fewer attempts have been made to design customized architectures specifically for this purpose. This paper presents a comprehensive evaluation on transfer learning based solutions and automatically designed networks, analyzing the accuracy and robustness of different recognition models in three folds. First, we develop six different DCNN models (BNet, GNet, SqNet, DsNet, RsNet, IncReNet) based on transfer learning. Second, we adapt the Bayesian optimization method to optimize a CNN network (BONet) for classifying breast lesions. A retrospective dataset of 3034 US images collected from various hospitals is then used for evaluation. Extensive tests show that the BONet outperforms other models, exhibiting higher accuracy (83.33%), lower generalization gap (1.85%), shorter training time (66 min), and less model complexity (approximately 0.5 million weight parameters). We also compare the diagnostic performance of all models against that by three experienced radiologists. Finally, we explore the use of saliency maps to explain the classification decisions made by different models. Our investigation shows that saliency maps can assist in comprehending the classification decisions. |
Keywords | Breast cancer; Ultrasonography; Cancer recognition; Deep convolutional neural network; Transfer learning; Automatic architecture design; Bayesian optimization |
Year | 2023 |
Journal | Medical & Biological Engineering & Computing |
Journal citation | pp. 1-15 |
Publisher | Springer |
ISSN | 1741-0444 |
Digital Object Identifier (DOI) | https://doi.org/10.1007/s11517-023-02922-y |
Web address (URL) | https://link.springer.com/article/10.1007/s11517-023-02922-y |
Publisher's version | License File Access Level Open |
Output status | Published |
Publication dates | |
Online | 22 Sep 2023 |
Publication process dates | |
Deposited | 02 Nov 2023 |
https://repository.derby.ac.uk/item/q25zx/classification-of-breast-lesions-in-ultrasound-images-using-deep-convolutional-neural-networks-transfer-learning-versus-automatic-architecture-design
Download files
27
total views13
total downloads1
views this month0
downloads this month