English

Floodgate: inference for model-free variable importance

Methodology 2022-09-13 v5

Abstract

Many modern applications seek to understand the relationship between an outcome variable YY and a covariate XX in the presence of a (possibly high-dimensional) confounding variable ZZ. Although much attention has been paid to testing \emph{whether} YY depends on XX given ZZ, in this paper we seek to go beyond testing by inferring the \emph{strength} of that dependence. We first define our estimand, the minimum mean squared error (mMSE) gap, which quantifies the conditional relationship between YY and XX in a way that is deterministic, model-free, interpretable, and sensitive to nonlinearities and interactions. We then propose a new inferential approach called \emph{floodgate} that can leverage any working regression function chosen by the user (allowing, e.g., it to be fitted by a state-of-the-art machine learning algorithm or be derived from qualitative domain knowledge) to construct asymptotic confidence bounds, and we apply it to the mMSE gap. \acc{We additionally show that floodgate's accuracy (distance from confidence bound to estimand) is adaptive to the error of the working regression function.} We then show we can apply the same floodgate principle to a different measure of variable importance when YY is binary. Finally, we demonstrate floodgate's performance in a series of simulations and apply it to data from the UK Biobank to infer the strengths of dependence of platelet count on various groups of genetic mutations.

Cite

@article{arxiv.2007.01283,
  title  = {Floodgate: inference for model-free variable importance},
  author = {Lu Zhang and Lucas Janson},
  journal= {arXiv preprint arXiv:2007.01283},
  year   = {2022}
}
R2 v1 2026-06-23T16:48:36.576Z