Search papers, labs, and topics across Lattice.
The paper introduces CSA-MoE-Net, a novel deep learning architecture for breast ultrasound image classification, designed to improve feature representation and classification accuracy. The network uses a Cross-Stage Attention module within a ResNet-18 backbone to enhance key tumor features and a Mixture of Experts (MoE) block to learn complementary features from different regions of the tumor. Experiments on a dataset of 2,129 breast ultrasound images demonstrate that CSA-MoE-Net achieves state-of-the-art performance, significantly outperforming ResNet-18 in accuracy, precision, recall, F1-score, and AUC.
A new deep learning architecture leverages cross-stage attention and a mixture of experts to push breast ultrasound image classification accuracy to 96.33%, potentially improving early breast cancer diagnosis.
Breast ultrasound imaging is an important noninvasive method for early breast cancer diagnosis, but automatic benign/malignant classification remains challenging due to tumor heterogeneity, blurred boundaries, and data imbalance. To improve feature representation and classification accuracy, this paper proposes the Cross-Stage Attention Mixture-of-Experts Network (CSA-MoE-Net). It adopts a Cross-Stage Attention-enhanced ResNet-18 as the backbone, in which the Cross-Stage Attention module adaptively recalibrates multi-level features, thereby enhancing key tumor features and suppressing redundancy. A three-branch Mixture of Experts (MoE) Block learns complementary features from the Whole Tumor Image, Tumor Core, and Boundary, and an Adaptive Gating Network fuses them to capture morphological, textural, and contextual information. The fused features are denoted as Fused Expert Feature (FEF) in the architecture. Experiments on a balanced dataset of 2,129 breast ultrasound images show that, averaged over 20 independent runs, the model achieves an accuracy of 96.33\%, precision of 94.09\%, recall of 98.53\%, F1-score of 96.25\%, and AUC of 99.50\%. Compared to the baseline ResNet-18, these metrics improve by 3.01, 0.70, 5.37, 2.98, and 5.42 percentage points, respectively. The proposed mechanism requires no invasive modification and can be seamlessly embedded into VGG-16, DenseNet-121, etc., yielding stable performance gains, thus providing reliable support for computer-aided diagnosis.