MfdcModel: A Novel Classification Model for Classification of Benign and Malignant Breast Tumors in Ultrasound Images

Abstract

Automatic classification of benign and malignant breast ultrasound images is an important and challenging task to improve the efficiency and accuracy of clinical diagnosis of breast tumors and reduce the rate of missed and misdiagnosis. The task often requires a large amount of data to train. However, it is difficult to obtain medical images, which contradicts the large amount of data needed to obtain good diagnostic models for training. In this paper, a novel classification model for the classification of breast tumors is proposed to improve the performance of diagnosis models trained by small datasets. The method integrates three features from medical features extracted from segmented images, features selected from the pre-trained ResNet101 output by principal component analysis (PCA), and texture features. Among the medical features that are used to train the naive Bayes (NB) classifier, and the PCA-selected features are used to train the support vector machine (SVM) classifier. Subsequently, the final results of boosting are obtained by weighting the classifiers. A five-fold cross-validation experiment yields an average accuracy of 89.17%, an average precision of 90.00%, and an average AUC value of 0.95. According to the experimental results, the proposed method has better classification accuracy compared to the accuracy obtained by other models trained on only small datasets. This approach can serve as a reliable second opinion for radiologists, and it can also provide useful advice for junior radiologists who do not have sufficient clinical experience.