English

Feature Selection and Junta Testing are Statistically Equivalent

Machine Learning 2025-07-23 v2 Computational Complexity Data Structures and Algorithms Machine Learning

Abstract

For a function f ⁣:{0,1}n{0,1}f \colon \{0,1\}^n \to \{0,1\}, the junta testing problem asks whether ff depends on only kk variables. If ff depends on only kk variables, the feature selection problem asks to find those variables. We prove that these two tasks are statistically equivalent. Specifically, we show that the ``brute-force'' algorithm, which checks for any set of kk variables consistent with the sample, is simultaneously sample-optimal for both problems, and the optimal sample size is Θ(1ε(2klog(nk)+log(nk))). \Theta\left(\frac 1 \varepsilon \left( \sqrt{2^k \log {n \choose k}} + \log {n \choose k}\right)\right).

Cite

@article{arxiv.2505.04604,
  title  = {Feature Selection and Junta Testing are Statistically Equivalent},
  author = {Lorenzo Beretta and Nathaniel Harms and Caleb Koch},
  journal= {arXiv preprint arXiv:2505.04604},
  year   = {2025}
}

Comments

32 pages

R2 v1 2026-06-28T23:24:46.383Z