A Comprehensive Framework for Multi-Class Breast Lesion Classification Using a Gaussian Process-Optimized VGG16-Capsule Hybrid Model on Ultrasound Radio Frequency B-mode Images
Author(s):
Article Type:
Research/Original Article (دارای رتبه معتبر)
Abstract:
Effective breast cancer screening is essential for early detection and treatment. Ultrasound (US) radio frequency (RF) data offers a novel, equipment-independent approach. However, class imbalance and limited interpretability hinder its application in clinical practice. This study proposes a hybrid deep learning model combining a pre-trained convolutional neural network (CNN) based on VGG16 and capsule neural networks (CapsNets) to classify breast lesions. The model was evaluated using an RFTSBU dataset, comprising 220 data points from 118 patients, acquired on the SuperSonic Imagine Aixplorer® system with a linear transducer. To address data imbalance, the synthetic minority over-sampling technique (SMOTE) was employed to generate synthetic samples while preserving data distribution. Furthermore, Gaussian process (GP) was applied to fine-tune CapsNet hyperparameters, improving classification performance. Three experiments were conducted to classify breast lesions into two, three, and four classes: (I) CapsNet with balanced datasets based on class weight, (II) CapsNet with balanced datasets using SMOTE, and (III) CapsNet with hyperparameters optimized using GP on SMOTE-balanced datasets. The proposed model achieved average accuracies of 98.81%, 97.89%, and 95.94% for two-, three-, and four-class classifications, respectively. The hybrid VGG16-CapsNet model effectively addresses class imbalance and captures critical lesion attributes such as size, perspective, and orientation. Integrating GP optimization achieves superior accuracy in multi-class breast lesion classification. The proposed approach can serve as a valuable aid in breast tumor classification using US RF B-mode images. Its enhanced interpretability and efficiency enable clinicians to move beyond binary classification, facilitating identifying and differentiating a broader spectrum of breast lesions.
Keywords:
Language:
English
Published:
Journal of Electrical Engineering, Volume:57 Issue: 2, Summer-Autumn 2025
Pages:
317 to 332
https://www.magiran.com/p2865514
سامانه نویسندگان
مقالات دیگری از این نویسنده (گان)
-
Selection of Effective Features from Raw US RF Signals to Enhance Intelligent Breast Lesion Classification using Machine Learning
Mahsa Arab, *, Saeid Rashidi, Maryam Mehdizadeh Dastjerdi, Nasrin Ahmadinejad
Iranian Journal of Biomedical Engineering, -
Brain Connectivity Estimation Pitfall in Multiple Trials of Electroencephalography Data
Vida Mehdizadehfar*, Farnaz Ghassemi,
Basic and Clinical Neuroscience, Jul-Aug 2023