تعداد نشریات | 7 |
تعداد شمارهها | 399 |
تعداد مقالات | 5,389 |
تعداد مشاهده مقاله | 5,288,171 |
تعداد دریافت فایل اصل مقاله | 4,882,904 |
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 | ||
AUT Journal of Electrical Engineering | ||
مقالات آماده انتشار، پذیرفته شده، انتشار آنلاین از تاریخ 02 دی 1403 اصل مقاله (1.85 M) | ||
نوع مقاله: Research Article | ||
شناسه دیجیتال (DOI): 10.22060/eej.2024.23377.5609 | ||
نویسندگان | ||
Mahsa Arab1؛ Ali Fallah* 1؛ Saeid Rashidi2؛ Maryam Mehdizadeh Dastjerdi1؛ Nasrin Ahmadinejad3 | ||
1Biomedical Engineering Department, Amirkabir University of Technology, Tehran, Iran | ||
2Medical Sciences and Technologies Department, Islamic Azad University, Science and Research Branch, Tehran, Iran | ||
3Radiology-Medical Imaging Center, Cancer Research Institute, Imam Khomeini Hospital Advanced Diagnostic and Interventional Radiology Research Center, Tehran University of Medical Sciences, Tehran, Iran | ||
چکیده | ||
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. | ||
کلیدواژهها | ||
ultrasound؛ radio frequency؛ breast cancer classification؛ capsule neural networks؛ Gaussian process | ||
آمار تعداد مشاهده مقاله: 27 تعداد دریافت فایل اصل مقاله: 26 |