Classification Prediction of Breast Cancer Based on Machine Learning

Abstract

Breast cancer is the most common and deadly type of cancer in the world. Based on machine learning algorithms such as XGBoost, random forest, logistic regression, and K-nearest neighbor, this paper establishes different models to classify and predict breast cancer, so as to provide a reference for the early diagnosis of breast cancer. Recall indicates the probability of detecting malignant cancer cells in medical diagnosis, which is of great significance for the classification of breast cancer, so this article takes recall as the primary evaluation index and considers the precision, accuracy, and F1-score evaluation indicators to evaluate and compare the prediction effect of each model. In order to eliminate the influence of different dimensional concepts on the effect of the model, the data are standardized. In order to find the optimal subset and improve the accuracy of the model, 15 features were screened out as input to the model through the Pearson correlation test. The K-nearest neighbor model uses the cross-validation method to select the optimal k value by using recall as an evaluation index. For the problem of positive and negative sample imbalance, the hierarchical sampling method is used to extract the training set and test set proportionally according to different categories. The experimental results show that under different dataset division (8 : 2 and 7 : 3), the prediction effect of the same model will have different changes. Comparative analysis shows that the XGBoost model established in this paper (which divides the training set and test set by 8 : 2) has better effects, and its recall, precision, accuracy, and F1-score are 1.00, 0.960, 0.974, and 0.980, respectively.

A Robust Lightweight Compound Emotion Recognition Approach Using Depthwise Separable CNN

“…The alternate term used instead of F1-score is Dice Similarity Coefficient (DSC). The mathematical formula for the F1-score is shown in Equation 9 [49,50].…”

Performance enhancement of machine learning algorithm for breast cancer diagnosis using hyperparameter optimization

“…Precision demonstrates a model's performance based on the correct diagnosis, which represents the percentage of really positive samples among positive predicted samples [29]. The recall is the ratio of the correctly predicted positive samples to all predicted positive samples in the original class, which is also called the rate of true positive [10], [15]. F1-score is the harmonic average of the pre and rec index, which is always less than the value of accuracy as it is calculated with precision and recall value [10], [17].…”

Development of an Artificial Intelligence-Supported Hybrid Data Management Platform for Monitoring Depression and Anxiety Symptoms in the Perinatal Period: Pilot-Scale Study

“…On the other hand, machine learning is a branch of artificial intelligence in which algorithms discover connections between input and output data [19]. Machine learning algorithms have been used in various branches of healthcare [20] [21]. To calculate the probability of developing acute graft-host illness, Arai et al used a Japanese population of 26,695 patients.…”