Abstract:Ultrasonography is a widely used medical imaging technique for detecting breast cancer. While manual diagnostic methods are subject to variability and time-consuming, computer-aided diagnostic (CAD) methods have proven to be more efficient. However, current CAD approaches neglect the impact of noise and artifacts on the accuracy of image analysis. To enhance the precision of breast ultrasound image analysis for identifying tissues, organs and lesions, we propose a novel approach for improved tumor classification through a dual-input model and global average pooling (GAP)-guided attention loss function. Our approach leverages a convolutional neural network with transformer architecture and modifies the single-input model for dual-input. This technique employs a fusion module and GAP operation-guided attention loss function simultaneously to supervise the extraction of effective features from the target region and mitigate the effect of information loss or redundancy on misclassification. Our proposed method has three key features: (i) ResNet and MobileViT are combined to enhance local and global information extraction. In addition, a dual-input channel is designed to include both attention images and original breast ultrasound images, mitigating the impact of noise and artifacts in ultrasound images. (ii) A fusion module and GAP operation-guided attention loss function are proposed to improve the fusion of dual-channel feature information, as well as supervise and constrain the weight of the attention mechanism on the fused focus region. (iii) Using the collected uterine fibroid ultrasound dataset to train ResNet18 and load the pre-trained weights, our experiments on the BUSI and BUSC public datasets demonstrate that the proposed method outperforms some state-of-the-art methods. The code will be publicly released at https://github.com/425877/Improved-Breast-Ultrasound-Tumor-Classification.

Segmentation Information with Attention Integration for Classification of Breast Tumor in Ultrasound Image

Deep Fusion of Human-Machine Knowledge with Attention Mechanism for Breast Cancer Diagnosis

A Hybrid Attentional Guidance Network for Tumors Segmentation of Breast Ultrasound Images.

ACL-DUNet: A tumor segmentation method based on multiple attention and densely connected breast ultrasound images

Breast Tumor Classification in Ultrasound Images by Fusion of Deep Convolutional Neural Network and Shallow LBP Feature

Tumor segmentation in automated whole breast ultrasound using bidirectional LSTM neural network and attention mechanism

Effective hybrid attention network based on pseudo-color enhancement in ultrasound image segmentation

ACA U-Net: Asymmetric Channel Attention Mechanism for Semantic Segmentation of Breast Ultrasound Images

Attention Enriched Deep Learning Model for Breast Tumor Segmentation in Ultrasound Images

Improved breast ultrasound tumor classification using dual-input CNN with GAP-guided attention loss

Automated Classification and Segmentation and Feature Extraction from Breast Imaging Data

Multi-task Learning for Segmentation and Classification of Tumors in 3D Automated Breast Ultrasound Images

Automatic Breast Ultrasound Tumor Segmentation Via One-Stage Hierarchical Target Activation Network

Image Segmentation of Triple-Negative Breast Cancer by Incorporating Multiscale and Parallel Attention Mechanisms

Using Segment-Level Attention to Guide Breast Ultrasound Video Classification

DAU-Net: Dual attention-aided U-Net for segmenting tumor in breast ultrasound images

Multi-task learning for segmentation and classification of breast tumors from ultrasound images

DBL-Net: A dual-branch learning network with information from spatial and frequency domains for tumor segmentation and classification in breast ultrasound image

Attention guided neural ODE network for breast tumor segmentation in medical images

Weakly-supervised Deep Learning for Breast Tumor Segmentation in Ultrasound Images

Tumor Segmentation in Breast Ultrasound Image by Means of Res Path Combined with Dense Connection Neural Network