Pruning has a disparate impact on model accuracy
Cuong Tran, Ferdinando Fioretto, Jung-Eun Kim, Rakshit Naidu
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.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 661a439d-7ec4-4ed4-89b0-66cdf7d96c39Cited by top-tier papers15
- Sparse Model Soups: A Recipe for Improved Pruning via Model AveragingMax Zimmer, Christoph Spiegel, Sebastian PokuttaICLR 2024 · 22 citations
- Recall Distortion in Neural Network Pruning and the Undecayed Pruning AlgorithmAidan Good, Jiaqi Lin, Xin Yu, Hannah Sieg et al.NeurIPS 2022 · 15 citations
- Balancing Act: Constraining Disparate Impact in Sparse ModelsMeraj Hashemizadeh, Juan Ramirez, Rohan Sukumaran, Golnoosh Farnadi et al.ICLR 2024 · 9 citations
- LayerMerge: Neural Network Depth Compression through Layer Pruning and MergingJinuk Kim, Marwa El Halabi, Mingi Ji, Hyun Oh SongICML 2024 · 5 citations
- Impartial Adversarial Distillation: Addressing Biased Data-Free Knowledge Distillation via Adaptive Constrained OptimizationDongping Liao, Xitong Gao, Chengzhong XuAAAI 2024 · 5 citations
Builds on6
- Comparing Rewinding and Fine-tuning in Neural Network PruningAlex Renda, Jonathan Frankle, Michael CarbinICLR 2020 · 437 citations
- To be Robust or to be Fair: Towards Fairness in Adversarial TrainingHan Xu, Xiaorui Liu, Yaxin Li, Anil K. Jain et al.ICML 2021 · 218 citations
- Differentially Private Empirical Risk Minimization under the Fairness LensCuong Tran, My H. Dinh, Ferdinando FiorettoNeurIPS 2021 · 61 citations
- Bias and Variance of Post-processing in Differential PrivacyKeyu Zhu, Pascal Van Hentenryck, Ferdinando FiorettoAAAI 2021 · 46 citations
- The Rich Get Richer: Disparate Impact of Semi-Supervised LearningZhaowei Zhu, Tianyi Luo, Yang LiuICLR 2022 · 44 citations
Related papers
- Towards Fairness-aware Adversarial Network PruningLei Zhang, Zhibo Wang, Xiaowei Dong, Yunhe Feng et al.ICCV 2023 · 8 citations
- Disparate Impact in Differential Privacy from Gradient MisalignmentMaria S. Esipova, Atiyeh Ashari Ghomi, Yaqiao Luo, Jesse C. CresswellICLR 2023 · 7 citations
- Pruning's Effect on Generalization Through the Lens of Training and RegularizationTian Jin, Michael Carbin, Daniel M. Roy, Jonathan Frankle et al.NeurIPS 2022 · 40 citations
- The Generalization-Stability Tradeoff In Neural Network PruningBrian R. Bartoldson, Ari S. Morcos, Adrian Barbu, Gordon ErlebacherNeurIPS 2020 · 97 citations
- Membership Inference Attacks and Defenses in Neural Network PruningXiaoyong Yuan, Lan ZhangUSENIX Security 2022
