Use of random forest to estimate population attributable fractions from a case-control study of Salmonella enterica serotype Enteritidis infections

Epidemiol Infect. 2015 Oct;143(13):2786-94. doi: 10.1017/S095026881500014X. Epub 2015 Feb 12.

Abstract

To design effective food safety programmes we need to estimate how many sporadic foodborne illnesses are caused by specific food sources based on case-control studies. Logistic regression has substantive limitations for analysing structured questionnaire data with numerous exposures and missing values. We adapted random forest to analyse data of a case-control study of Salmonella enterica serotype Enteritidis illness for source attribution. For estimation of summary population attributable fractions (PAFs) of exposures grouped into transmission routes, we devised a counterfactual estimator to predict reductions in illness associated with removing grouped exposures. For the purpose of comparison, we fitted the data using logistic regression models with stepwise forward and backward variable selection. Our results show that the forward and backward variable selection of logistic regression models were not consistent for parameter estimation, with different significant exposures identified. By contrast, the random forest model produced estimated PAFs of grouped exposures consistent in rank order with results obtained from outbreak data, with egg-related exposures having the highest estimated PAF (22·1%, 95% confidence interval 8·5-31·8). Random forest might be structurally more coherent and efficient than logistic regression models for attributing Salmonella illnesses to sources involving many causal pathways.

Keywords: Causality; counterfactual; foodborne diseases; logistic regression; machine learning.

MeSH terms

  • Animals
  • Case-Control Studies
  • Decision Trees*
  • Disease Outbreaks
  • Food Microbiology
  • Humans
  • Logistic Models
  • Predictive Value of Tests
  • Risk Factors
  • Salmonella Food Poisoning / epidemiology*
  • Salmonella Food Poisoning / microbiology*
  • Salmonella enteritidis / isolation & purification*
  • Serogroup
  • United States / epidemiology