Accepted for/Published in: JMIR Medical Informatics
Date Submitted: Dec 10, 2019
Date Accepted: Apr 19, 2020
Warning: This is an author submission that is not peer-reviewed or edited. Preprints - unless they show as "accepted" - should not be relied on to guide clinical practice or health-related behavior and should not be reported in news media as established information.
Predicting breast cancer in Chinese women using machine-learning algorithms
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.
Citation
Request queued. Please wait while the file is being generated. It may take some time.
Copyright
© The authors. All rights reserved. This is a privileged document currently under peer-review/community review (or an accepted/rejected manuscript). Authors have provided JMIR Publications with an exclusive license to publish this preprint on it's website for review and ahead-of-print citation purposes only. While the final peer-reviewed paper may be licensed under a cc-by license on publication, at this stage authors and publisher expressively prohibit redistribution of this draft paper other than for review purposes.