Rishit Sheth

Differentiable Feature Selection by Discrete Relaxation

2020 · International Conference on Artificial Intelligence and Statistics

In this paper, we introduce Differentiable Feature Selection, a gradient-based search algorithm for feature selection. Our approach extends a recent result on the estimation of learnability in the sublinear data regime by showing that the …