# Research on the establishment of a four-classification model for breast mass ultrasound images based on transfer learning

## TL;DR

This study uses transfer learning with deep neural networks to classify breast masses in ultrasound images into four categories, achieving higher accuracy than expert radiologists.

## Contribution

The paper introduces an optimized transfer learning approach for four-class breast mass classification in ultrasound images, outperforming human experts.

## Key findings

- DenseNet121 and GoogLeNet achieved 91.2% and 92.6% accuracy in four-class breast mass classification.
- The models outperformed radiologists with a statistically significant difference (p < 0.01).
- Optimized transfer learning improved class-specific accuracy for invasive ductal carcinoma, mucinous carcinoma, and inflammatory mass.

## Abstract

Breast mass classification via ultrasound is critical for early diagnosis but remains challenging due to overlapping morphological features. This study evaluates the efficacy of optimized transfer learning (TL) based on deep convolutional neural networks (DCNNs) in distinguishing four breast mass categories: invasive ductal carcinoma (IDC), fibroadenoma (FA), mucinous carcinoma (MC), and inflammatory mass (IM). Our approach comprises four key steps: (1) applying extensive data augmentation to a retrospective dataset of 346 ultrasound images from 294 patients (November 2021-March 2023) classified by pathological confirmation; (2) implementing transfer learning with five pretrained DCNN models (ResNet18, ResNet50, DenseNet121, MobileNetV2, GoogLeNet); (3) optimizing model hyperparameters for four-class classification; (4) comparing performance metrics (accuracy, precision, recall, F1-score, AUC-ROC) against two senior radiologists (8- and 10-years’ experience) on a held-out test set. Results: The DenseNet121 and GoogLeNet models demonstrated superior overall accuracy (0.912 and 0.926, respectively), significantly outperforming the radiologists’ consensus (0.691) (p < 0.01). In class-specific analysis, the optimized models achieved notably higher accuracy for IDC, MC, and IM. Statistical testing confirmed the significant performance improvement of the top models over human experts. Optimized TL with DCNNs, particularly DenseNet121 and GoogLeNet, enables highly accurate four-class breast mass classification in ultrasound images, surpassing expert radiologist performance with statistical significance. This approach holds promise for clinical decision support. Code is publicly available at https://github.com/jinlao777/BCC.