Survey-aware Machine Learning: A Guideline for Valid Population Health Inference based on Scoping Review
Not on the map. Why?
Abstract (arXiv)
Machine Learning (ML) models trained on complex health surveys such as the National Health and Nutrition Examination Survey (NHANES) often ignore primary sampling units, stratification variables, and sampling weights. This practice violates the independence assumptions of standard evaluation methods. As a result, estimates become biased, uncertainty is underestimated, and fairness assessments fail to reflect population-level disparities. We propose Survey-aware Machine Learning (SaML), a nine-step guideline that incorporates survey design metadata across the ML lifecycle. Through a scoping review of 16 methodological papers, we summarize existing work on weighted model training, design-based cross-validation, and survey-adjusted performance evaluation. We also identify gaps in hyperparameter tuning and deployment. We provide task-specific guidance that clarifies which steps are required for different analytical objectives. SaML provides a checklist for valid population inference from survey data.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 14 of its 63 references that are in the map. First posted on arXiv 9 May 2026.
Builds on
14 of 63 references.
References · 14
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2016XGBoost: A Scalable Tree Boosting Systemcited by 5,166
- 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,899
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,926
- 2019A Survey on Bias and Fairness in Machine Learningcited by 1,917
- 2022On scientific understanding with artificial intelligencecited by 70
- 2021How to avoid machine learning pitfalls: a guide for academic researcherscited by 35
- 2021Improving the output quality of official statistics based on machine learning algorithmscited by 5
- 2014Building Consistent Regression Trees From Complex Sample Datacited by 4
- 2025Multi-View Contrastive Learning for Robust Domain Adaptation in Medical Time Series Analysiscited by 4
- 2017Estimating prediction error for complex samplescited by 2
- 2020Model-assisted estimation in high-dimensional settings for survey datacited by 2
2 more
- 2023Design-based conformal predictioncited by 2
- 2023Evaluating Binary Outcome Classifiers Estimated from Survey Datacited by 1
Built on by
Not cited yet.