Machine learning and SHAP value interpretation for predicting comorbidity of cardiovascular disease and cancer with dietary antioxidants
Objective: To develop and validate a machine learning model incorporating dietary antioxidants to predict cardiovascular disease (CVD)-cancer comorbidity and to elucidate the role of antioxidants in disease prediction. Methods: Data were sourced from the National Health and Nutrition Examination Sur...
Saved in:
| Main Authors: | , , , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
Elsevier
2025-02-01
|
| Series: | Redox Biology |
| Subjects: | |
| Online Access: | http://www.sciencedirect.com/science/article/pii/S2213231724004488 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1841542509821427712 |
|---|---|
| author | Xiangjun Qi Shujing Wang Caishan Fang Jie Jia Lizhu Lin Tianhui Yuan |
| author_facet | Xiangjun Qi Shujing Wang Caishan Fang Jie Jia Lizhu Lin Tianhui Yuan |
| author_sort | Xiangjun Qi |
| collection | DOAJ |
| description | Objective: To develop and validate a machine learning model incorporating dietary antioxidants to predict cardiovascular disease (CVD)-cancer comorbidity and to elucidate the role of antioxidants in disease prediction. Methods: Data were sourced from the National Health and Nutrition Examination Survey. Antioxidants, including vitamins, minerals, and polyphenols, were selected as key features. Additionally, demographic, lifestyle, and health condition features were incorporated to improve model accuracy. Feature preprocessing included removing collinear features, addressing class imbalance, and normalizing data. Models constructed within the mlr3 framework included recursive partitioning and regression trees, random forest, kernel k-nearest neighbors, naïve bayes, and light gradient boosting machine (LightGBM). Benchmarking provided a systematic approach to evaluating and comparing model performance. SHapley Additive exPlanation (SHAP) values were calculated to determine the prediction role of each feature in the model with the highest predictive performance. Results: This analysis included 10,064 participants, with 353 identified as having comorbid CVD and cancer. After excluding collinear features, the machine learning model retained 29 dietary antioxidant features and 9 baseline features. LightGBM achieved the highest predictive accuracy at 87.9 %, a classification error rate of 12.1 %, and the top area under the receiver operating characteristic curve (0.951) and the precision‐recall curve (0.930). LightGBM also demonstrated balanced sensitivity and specificity, both close to 88 %. SHAP analysis indicated that naringenin, magnesium, theaflavin, kaempferol, hesperetin, selenium, malvidin, and vitamin C were the most influential contributors. Conclusion: LightGBM exhibited the best performance for predicting CVD-cancer comorbidity. SHAP values highlighted the importance of antioxidants, with naringenin and magnesium emerging as primary factors in this model. |
| format | Article |
| id | doaj-art-ee16eda3596c48f998f147e3edd98e8d |
| institution | Kabale University |
| issn | 2213-2317 |
| language | English |
| publishDate | 2025-02-01 |
| publisher | Elsevier |
| record_format | Article |
| series | Redox Biology |
| spelling | doaj-art-ee16eda3596c48f998f147e3edd98e8d2025-01-14T04:12:11ZengElsevierRedox Biology2213-23172025-02-0179103470Machine learning and SHAP value interpretation for predicting comorbidity of cardiovascular disease and cancer with dietary antioxidantsXiangjun Qi0Shujing Wang1Caishan Fang2Jie Jia3Lizhu Lin4Tianhui Yuan5The First Clinical Medical College, Guangzhou University of Chinese Medicine, Guangzhou, 510000, ChinaThe First Clinical Medical College, Guangzhou University of Chinese Medicine, Guangzhou, 510000, ChinaHospital of Chengdu University of Traditional Chinese Medicine, Chengdu University of Traditional Chinese Medicine, Chengdu, 610031, China; Yong Loo Lin School of Medicine, National University of Singapore, 117597, SingaporeThe First Affiliated Hospital of Guangzhou University of Chinese Medicine, Guangzhou, 510405, China; Guangdong Clinical Research Academy of Chinese Medicine, Guangzhou, 510405, China; Kolling Institute of Medical Research, University of Sydney, Sydney, NSW, 2065, AustraliaThe First Affiliated Hospital of Guangzhou University of Chinese Medicine, Guangzhou, 510405, China; Guangdong Clinical Research Academy of Chinese Medicine, Guangzhou, 510405, China; Corresponding author. The First Affiliated Hospital of Guangzhou University of Chinese Medicine, No.12, Ji Chang Road, Baiyun District, Guangzhou, 510405, China.The First Affiliated Hospital of Guangzhou University of Chinese Medicine, Guangzhou, 510405, China; Guangdong Clinical Research Academy of Chinese Medicine, Guangzhou, 510405, China; Corresponding author. The First Affiliated Hospital of Guangzhou University of Chinese Medicine, No.12, Ji Chang Road, Baiyun District, Guangzhou, 510405, China.Objective: To develop and validate a machine learning model incorporating dietary antioxidants to predict cardiovascular disease (CVD)-cancer comorbidity and to elucidate the role of antioxidants in disease prediction. Methods: Data were sourced from the National Health and Nutrition Examination Survey. Antioxidants, including vitamins, minerals, and polyphenols, were selected as key features. Additionally, demographic, lifestyle, and health condition features were incorporated to improve model accuracy. Feature preprocessing included removing collinear features, addressing class imbalance, and normalizing data. Models constructed within the mlr3 framework included recursive partitioning and regression trees, random forest, kernel k-nearest neighbors, naïve bayes, and light gradient boosting machine (LightGBM). Benchmarking provided a systematic approach to evaluating and comparing model performance. SHapley Additive exPlanation (SHAP) values were calculated to determine the prediction role of each feature in the model with the highest predictive performance. Results: This analysis included 10,064 participants, with 353 identified as having comorbid CVD and cancer. After excluding collinear features, the machine learning model retained 29 dietary antioxidant features and 9 baseline features. LightGBM achieved the highest predictive accuracy at 87.9 %, a classification error rate of 12.1 %, and the top area under the receiver operating characteristic curve (0.951) and the precision‐recall curve (0.930). LightGBM also demonstrated balanced sensitivity and specificity, both close to 88 %. SHAP analysis indicated that naringenin, magnesium, theaflavin, kaempferol, hesperetin, selenium, malvidin, and vitamin C were the most influential contributors. Conclusion: LightGBM exhibited the best performance for predicting CVD-cancer comorbidity. SHAP values highlighted the importance of antioxidants, with naringenin and magnesium emerging as primary factors in this model.http://www.sciencedirect.com/science/article/pii/S2213231724004488Machine learningSHAPCardiovascular diseaseCancerDietary antioxidants |
| spellingShingle | Xiangjun Qi Shujing Wang Caishan Fang Jie Jia Lizhu Lin Tianhui Yuan Machine learning and SHAP value interpretation for predicting comorbidity of cardiovascular disease and cancer with dietary antioxidants Redox Biology Machine learning SHAP Cardiovascular disease Cancer Dietary antioxidants |
| title | Machine learning and SHAP value interpretation for predicting comorbidity of cardiovascular disease and cancer with dietary antioxidants |
| title_full | Machine learning and SHAP value interpretation for predicting comorbidity of cardiovascular disease and cancer with dietary antioxidants |
| title_fullStr | Machine learning and SHAP value interpretation for predicting comorbidity of cardiovascular disease and cancer with dietary antioxidants |
| title_full_unstemmed | Machine learning and SHAP value interpretation for predicting comorbidity of cardiovascular disease and cancer with dietary antioxidants |
| title_short | Machine learning and SHAP value interpretation for predicting comorbidity of cardiovascular disease and cancer with dietary antioxidants |
| title_sort | machine learning and shap value interpretation for predicting comorbidity of cardiovascular disease and cancer with dietary antioxidants |
| topic | Machine learning SHAP Cardiovascular disease Cancer Dietary antioxidants |
| url | http://www.sciencedirect.com/science/article/pii/S2213231724004488 |
| work_keys_str_mv | AT xiangjunqi machinelearningandshapvalueinterpretationforpredictingcomorbidityofcardiovasculardiseaseandcancerwithdietaryantioxidants AT shujingwang machinelearningandshapvalueinterpretationforpredictingcomorbidityofcardiovasculardiseaseandcancerwithdietaryantioxidants AT caishanfang machinelearningandshapvalueinterpretationforpredictingcomorbidityofcardiovasculardiseaseandcancerwithdietaryantioxidants AT jiejia machinelearningandshapvalueinterpretationforpredictingcomorbidityofcardiovasculardiseaseandcancerwithdietaryantioxidants AT lizhulin machinelearningandshapvalueinterpretationforpredictingcomorbidityofcardiovasculardiseaseandcancerwithdietaryantioxidants AT tianhuiyuan machinelearningandshapvalueinterpretationforpredictingcomorbidityofcardiovasculardiseaseandcancerwithdietaryantioxidants |