Back

A Hybrid CNN-Transformer Deep Learning Model for Differentiating Benign and Malignant Breast Tumors Using Multi-View Ultrasound Images

qi, z.; Jianxing, Z.; Pan, T.; Miao, C.

2025-08-27 radiology and imaging
10.1101/2025.08.24.25334030 medRxiv
Show abstract

Breast cancer is a leading malignancy threatening womens health globally, making early and accurate diagnosis crucial. Ultrasound is a key screening and diagnostic tool due to its non-invasive, real-time, and cost-effective nature. However, its diagnostic accuracy is highly dependent on operator experience, and conventional single-image analysis often fails to capture the comprehensive features of a lesion. This study introduces a computer-aided diagnosis (CAD) system that emulates a clinicians multi-view diagnostic process. We developed a novel hybrid deep learning model that integrates a Convolutional Neural Network (CNN) with a Transformer architecture. The model uses a pretrained EfficientNetV2 to extract spatial features from multiple, unordered ultrasound images of a single lesion. These features are then processed by a Transformer encoder, whose self-attention mechanism globally models and fuses their intrinsic correlations. To prevent data leakage, a strict lesion-level data partitioning strategy ensured a rigorous evaluation. On an internal test set, our hybrid CNN-Transformer achieved an accuracy of 0.960, a sensitivity of 0.967, a specificity of 0.954, and an Area Under the Curve (AUC) of 0.9788. On an external dataset, it demonstrated an accuracy of 0.940, a sensitivity of 0.952, a specificity of 0.929, and an AUC of 0.9730. Furthermore, in a prospective validation on a newly collected independent dataset, the model maintained robust performance with an accuracy of 0.952 and an AUC of 0.9801. These results significantly outperform those of a baseline single-image model, which achieved accuracies of 0.88 and 0.89 and AUC of 0.95 and 0.94 on the internal and external dataset, respectively. This study demonstrates that combining CNN with Transformers yields a highly accurate and robust diagnostic system for breast ultrasound. By effectively fusing multi-view information, our model aligns with clinical logic and shows potential for improving diagnostic reliability.

Matching journals

The top 8 journals account for 50% of the predicted probability mass.

50% of probability mass above

"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.