BiasFairnessExplainabilitycs.LG
Simon Says: Evaluating and Mitigating Bias in Pruned Neural Networks with Knowledge Distillation
Cody Blakeney, Nathaniel Huish, Yan Yan, Ziliang Zong
June 15, 2021
arXiv.org
20 citations
1 influential
Key Takeaways
-
Neural network pruning can inadvertently create bias: The paper identifies that when AI models are compressed through pruning (a technique that removes parts of neural networks to make them faster), this process can introduce or amplify bias—a problem that hasn't been well-studied until now.
-
Knowledge distillation offers a practical solution: The researchers demonstrate that knowledge distillation (a technique where a smaller model learns from a larger one) can effectively reduce bias in pruned models, even when training data is imbalanced, providing practitioners with a concrete mitigation strategy.
-
New metrics enable bias measurement in compressed models: The paper proposes two new measurement tools—Combined Error Variance (CEV) and Symmetric Distance Error (SDE)—that allow organizations to quantitatively assess whether their pruned AI systems are treating different groups fairly, addressing a gap in current evaluation methods.
-
Model similarity predicts where bias will occur: The research reveals that how similar a pruned model is to the original model correlates strongly with bias levels, offering an explainable way to understand why bias emerges during compression—crucial for building trustworthy AI systems.
-
Bias in compressed AI is harder to detect than human bias: The authors emphasize that algorithmic bias from model compression is "more abstract and unintuitive" than traditional discrimination, making it essential to develop specialized evaluation tools before deploying these optimized models in real-world applications.
Abstract
In recent years the ubiquitous deployment of AI has posed great concerns in regards to algorithmic bias, discrimination, and fairness. Compared to traditional forms of bias or discrimination caused by humans, algorithmic bias generated by AI is more abstract and unintuitive therefore more difficult to explain and mitigate. A clear gap exists in the current literature on evaluating and mitigating bias in pruned neural networks. In this work, we strive to tackle the challenging issues of evaluating, mitigating, and explaining induced bias in pruned neural networks. Our paper makes three contributions. First, we propose two simple yet effective metrics, Combined Error Variance (CEV) and Symmetric Distance Error (SDE), to quantitatively evaluate the induced bias prevention quality of pruned models. Second, we demonstrate that knowledge distillation can mitigate induced bias in pruned neural networks, even with unbalanced datasets. Third, we reveal that model similarity has strong correlations with pruning induced bias, which provides a powerful method to explain why bias occurs in pruned neural networks. Our code is available at https://github.com/codestar12/pruning-distilation-bias
Paper Info
arXiv ID2106.07849
PublishedJune 15, 2021
Citations
20
Influential
1
Relevance82%
VenuearXiv.org
TypeJournalArticle
Categorycs.LG
Responsible AI Topics
BiasFairnessExplainability
Links
Citation data from Semantic Scholar