Few-shot classification of ultrasound breast cancer images using meta-learning algorithms

Gültekin Işık,İshak Paçal
DOI: https://doi.org/10.1007/s00521-024-09767-y
2024-04-18
Neural Computing and Applications
Abstract:Abstract Medical datasets often have a skewed class distribution and a lack of high-quality annotated images. However, deep learning methods require a large amount of labeled data for classification. In this study, we present a few-shot learning approach for the classification of ultrasound breast cancer images using meta-learning methods. We used prototypical networks and model agnostic meta-learning (MAML) algorithms as meta-learning methods. The breast ultrasound images (BUSI) dataset, which has three classes and is difficult to use in meta-learning, was used for meta-testing in a cross-domain approach along with other datasets for meta-training. Our proposed approach yielded an accuracy range of 0.882–0.889, achieved by implementing the ResNet50 backbone with ProtoNet in a 10-shot setting. These results represent a significant improvement ranging from 6.27 to 7.10% over the baseline accuracy of 0.831. The results showed that ProtoNet outperformed the MAML method for all k-shot settings. In addition, the use of ResNet models as the backbone network for feature extraction was found to be more successful than the use of a four-layer convolutional model. Our proposed method is the first attempt to apply meta-learning for few-shot classification in the BUSI dataset while providing higher accuracy compared to deep learning methods for medical images with small-scale datasets and few classes. The methodology used in this study can be adapted to other datasets with similar problems.
computer science, artificial intelligence
What problem does this paper attempt to address?
### Problems the Paper Attempts to Solve This paper aims to address the issue of small sample classification in medical imaging datasets, particularly in the application of breast cancer ultrasound image classification. Specifically: 1. **Data Scarcity**: Medical imaging datasets often suffer from imbalanced class distribution and a lack of high-quality annotated images. However, deep learning methods require a large amount of annotated data for classification. This paper proposes a meta-learning-based approach to solve this problem. 2. **Small Sample Classification**: Traditional machine learning and deep learning methods often perform poorly when the data is limited. This study achieves high-precision classification tasks with a small number of annotated samples by using meta-learning techniques. 3. **Breast Cancer Detection**: One of the main challenges in breast cancer detection is the lack of comprehensive and diverse datasets, which hinders the development of accurate classification models. This paper attempts to improve the classification accuracy of breast cancer ultrasound images using meta-learning techniques. 4. **Cross-Domain Strategy**: Since the breast ultrasound image dataset (BUSI) has only 3 categories, it is not suitable for standard small sample classification methods. Therefore, this paper proposes a cross-domain strategy, where other datasets are used for model training during the meta-training phase, and the BUSI dataset is specifically used for performance evaluation during the meta-testing phase. Through these means, this paper aims to develop a small sample classification method suitable for small-scale medical imaging datasets and achieve higher accuracy in breast cancer diagnosis.