JMIR Publications

ABSTRACT

Background:

Risk-based breast cancer screening is a cost-effective intervention for controlling breast cancer in China. But the successfully implementation of such intervention requires an accurate breast cancer prediction model for Chinese women. This study aims to evaluate and compare the performance of four different machine-learning algorithms on predicting breast cancer among Chinese women, using 10 breast cancer risk factors.

Objective:

This study aims to evaluate and compare the performance of four different machine-learning algorithms on predicting breast cancer among Chinese women, using 10 breast cancer risk factors.

Methods:

A dataset consisting of 7,127 breast cancer cases and 7,127 matched healthy controls was used for model training and testing. We utilized repeated 5-fold cross-validation and calculated AUC, sensitivity, specificity, and accuracy as the measures of the model performance.

Results:

The three novel machine-learning algorithms (XGBoost, Random Forest and Deep Neural Network) all achieved significantly higher AUCs, sensitivity and accuracy than Logistic Regression. Among the three novel machine-learning algorithms, XGBoost (AUC=0.742) outperformed Deep Neural Network (0.728) and Random Forest (AUC=0.728). Main residence, number of live births, menopause status, age and age at first birth were presented as top ranked variables in the three novel machine-learning algorithms.

Conclusions:

The novel machine-learning algorithms, especially XGBoost can be used to develop breast cancer prediction models to help identify women at high risk of breast cancer in developing countries.