Interpretable machine learning models for predicting cognitive impairment using NHANES neuropsychological tests: nutritional and sociodemographic associations.
📄 Abstract
Early identification of individuals at risk for cognitive impairment is essential for timely intervention and public health planning. While sociodemographic and clinical predictors are well recognized, the role of nutrition and its interactions in cognitive health remains less explored. Using data from the 2011-2014 National Health and Nutrition Examination Survey (NHANES, Ensemble models demonstrated excellent predictive performance, consistently outperforming traditional classifiers. Key predictors included education, age, socioeconomic status, and chronic disease conditions. Among nutritional factors, vitamin B2 emerged as consistently associated with lower predicted cognitive impairment risk across all three models, with notable interactions observed with copper and vitamin E. Exploratory Interpretable machine learning models integrating cognitive tests with demographic, clinical, and nutritional variables can accurately predict cognitive impairment. Nutritional predictors, particularly vitamin B2 and its interactions, may contribute to model performance and biological plausibility, suggesting potential avenues for stratified monitoring strategies.