English

Pruning has a disparate impact on model accuracy

Machine Learning 2022-10-14 v3

Abstract

Network pruning is a widely-used compression technique that is able to significantly scale down overparameterized models with minimal loss of accuracy. This paper shows that pruning may create or exacerbate disparate impacts. The paper sheds light on the factors to cause such disparities, suggesting differences in gradient norms and distance to decision boundary across groups to be responsible for this critical issue. It analyzes these factors in detail, providing both theoretical and empirical support, and proposes a simple, yet effective, solution that mitigates the disparate impacts caused by pruning.

Keywords

Cite

@article{arxiv.2205.13574,
  title  = {Pruning has a disparate impact on model accuracy},
  author = {Cuong Tran and Ferdinando Fioretto and Jung-Eun Kim and Rakshit Naidu},
  journal= {arXiv preprint arXiv:2205.13574},
  year   = {2022}
}

Comments

NeurIPS 2022

R2 v1 2026-06-24T11:30:04.528Z