← Search

Zahra Atashgahi

5 accepted papers

2024

Supervised Feature Selection via Ensemble Gradient Information from Sparse Neural Networks

AISTATS 2024poster

Feature selection algorithms aim to select a subset of informative features from a dataset to reduce the data dimensionality, consequently saving resource consumption and improving the model’s performance and interpretability. In recent years, feature selection based on neural networks has become a…

2022

Deep Ensembling with No Overhead for either Training or Testing: The All-Round Blessings of Dynamic Sparsity

ICLR 2022poster

The success of deep ensembles on improving predictive performance, uncertainty estimation, and out-of-distribution robustness has been extensively studied in the machine learning literature. Albeit the promising results, naively training multiple deep neural networks and combining their predictions…

2022

Where to Pay Attention in Sparse Training for Feature Selection?

NeurIPS 2022accept

A new line of research for feature selection based on neural networks has recently emerged. Despite its superiority to classical methods, it requires many training iterations to converge and detect the informative features. For datasets with a large number of samples or a very high dimensional featu…

2021

Sparse Training via Boosting Pruning Plasticity with Neuroregeneration

NeurIPS 2021poster

Works on lottery ticket hypothesis (LTH) and single-shot network pruning (SNIP) have raised a lot of attention currently on post-training pruning (iterative magnitude pruning), and before-training pruning (pruning at initialization). The former method suffers from an extremely large computation cost…