Using Life's Essential 8 and heavy metal exposure to determine infertility risk in American women: a machine learning prediction model based on the SHAP method

Xiaoqing GuQianbing LiXiangfei Wang^*

• Wuhan Sports University, Wuhan, China

Background: Fertility status is a marker of future health, and female infertility has been shown to be an important medical and social problem. LE8 and heavy metal exposure have both been shown to be associated with infertility. However, the association between LE8 and heavy metal exposure and female infertility has not been investigated. The aim of this study was to develop a machine learning prediction model for LE8 and heavy metal exposure and the risk of female infertility in the United States.Methods: The National Health and Nutrition Examination Survey is a nationally representative program conducted by the National Center for Health Statistics to assess the health and nutritional status of the U.S. population. For this study, 873 women between the ages of 20 and 45 were selected from the 2013-2018 NHANES dataset. The association between LE8 and heavy metal exposure and risk of infertility was assessed using logistic regression analysis and six machine learning models and the SHAP algorithm was used to explain the model’s decision process.Results: Of the six machine learning models, the LGBM model has the best predictive performance, with an AUROC of 0.964 on the test set. SHAP analysis showed that LE8, body mass index, diet, Cadmium, Cesium, Molybdenum, Antimony, Tin, education level and pregnancy history were significantly associated with the risk of female infertility. Cd, BMI and LE8 are the variables that contribute most to the prediction of infertility risk. Among them, BMI and LE8 have a negative predictive effect on female infertility in the model, while Cd has a positive contribution to the prediction of female infertility. Further analysis showed that there was a significant interaction between heavy metals and LE8, which may have a synergistic effect on the risk of female infertility.Conclusions: This study used LE8 and heavy metal exposure to create a machine learning model that predicts the risk of female infertility. The model identified ten key factors. The model demonstrated high predictive accuracy and good clinical interpretability. In the future, LE8 and heavy metal exposure can be used to screen for female infertility early on.