Sorry, you need to enable JavaScript to visit this website.

Group-wise Feature Selection for Supervised Learning

Citation Author(s):
Qi Xiao, Hebi Li, Jin Tian, Zhengdao Wang
Submitted by:
Qi Xiao
Last updated:
15 May 2022 - 1:31am
Document Type:
Qi Xiao
Paper Code:


Feature selection has been explored in two ways, global feature selection and instance-wise feature selection. Global feature selection picks the same feature selector for the entire dataset, while instance-wise feature selection allows different feature selectors for different data instances. We propose group-wise feature selection, a new setting that sits between global and instance-wise feature selections. In group-wise feature selection, we constrain the number of possible feature selectors to be a finite number K, which allows different feature selectors while regularizing the number of different selectors. This is for flexible trade-offs between expressiveness and model complexity. We propose two techniques to solve the problem: the first applies K-Means Clustering to the instance-wise feature selection algorithm; the second uses the mixture of experts model with Gumbel-Softmax to learn group membership and feature selector simultaneously. We evaluate our techniques and show promising results on both synthetic and real datasets.

0 users have voted: