PruneShift: A Framework for Evaluating Decision Reliability in Structured Pruning
Not on the map. Why?
Abstract (arXiv)
Structured pruning uses surrogate objectives because direct task evaluation over every feasible mask is too expensive. Most evaluations report average surrogate error or rank correlation on broadly sampled masks. These summaries do not directly test the mask chosen by the surrogate. We introduce PruneShift, an evaluation framework that separates broad predictive fidelity, fidelity near selector outputs, and the quality of the selected pruning decision. We first prove that Spearman and Kendall agreement can approach one while normalized selection regret remains maximal. We then derive sufficient conditions based on uniform error, selector suboptimality, decision margin, density ratio, and comparison mass. The analysis also yields a finite pool certificate with an explicit excess cost bound. Four studies test different links in this argument. External TextbookQA confirmation is heterogeneous: 7 of 20 simultaneous intervals favor the surrogate-selected mask, 6 favor its fixed comparator, and 7 cross zero. On a fixed Natural Questions pool, strict improvement holds in one of four settings. A controlled QQP experiment supports the proposed coverage mechanism in all 16 prespecified endpoints, although the sufficient bounds are conservative. Finally, a restricted OSSCAR reconstruction study on OPT-125M shows better local than broad fidelity in 68 of 75 primary endpoints. Independent fixed-mask confirmation is inconclusive in 24 of 25 endpoints and favors the comparator in one. These results show why predictive fit, decision reliability, and pruning method quality require separate evidence.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 33 of its 48 references that are in the map. First posted on arXiv 30 Aug 2026.
Builds on
33 of 48 references.
References · 33
- 2017Attention Is All You Needcited by 60,361
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,372
- 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,899
- 2022OPT: Open Pre-trained Transformer Language Modelscited by 3,512
- 2018The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networkscited by 2,262
- 2018Know What You Don't Know: Unanswerable Questions for SQuADcited by 1,554
- 2019Analyzing Multi-Head Self-Attention: Specialized Heads Do the Heavy Lifting, the Rest Can Be Prunedcited by 856
- 2019Are Sixteen Heads Really Better than One?cited by 816
- 2020Movement Pruning: Adaptive Sparsity by Fine-Tuningcited by 403
- 2017Smart "Predict, then Optimize"cited by 339
21 more
- 2018Melding the Data-Decisions Pipeline: Decision-Focused Learning for Combinatorial Optimizationcited by 227
- 2019MRQA 2019 Shared Task: Evaluating Generalization in Reading Comprehensioncited by 184
- 2021Block Pruning For Faster Transformerscited by 177
- 2022Structured Pruning Learns Compact and Accurate Modelscited by 170
- 2022A Fast Post-Training Pruning Framework for Transformerscited by 162
- 2021Conservative Objective Models for Effective Offline Model-Based Optimizationcited by 88
- 2021Decision-Focused Learning: Through the Lens of Learning to Rankcited by 62
- 2023ZipLM: Inference-Aware Structured Pruning of Language Modelscited by 49
- 2023The LLM Surgeoncited by 38
- 2021SOSP: Efficiently Capturing Global Correlations by Second-Order Structured Pruningcited by 30
- 2022SurCo: Learning Linear Surrogates For Combinatorial Nonlinear Optimization Problemscited by 21
- 2024OSSCAR: One-Shot Structured Pruning in Vision and Language Models with Combinatorial Optimizationcited by 18
- 2024Offline Model-Based Optimization by Learning to Rankcited by 13
- 2022Shapley Head Pruning: Identifying and Removing Interference in Multilingual Transformerscited by 9
- 2025From Local to Global: Revisiting Structured Pruning Paradigms for Large Language Modelscited by 6
- 2025SparseSwaps: Tractable LLM Pruning Mask Refinement at Scalecited by 4
- 2026Deterministic Differentiable Structured Pruning for Large Language Modelscited by 2
- 2026GRASPrune: Global Gating for Budgeted Structured Pruning of Large Language Modelscited by 2
- 2026CoCurve: Cross-Module Co-Pruning Curvature for Structured LLM Pruningcited by 2
- 2025Decision-aware training of spatiotemporal forecasting models to select a top K subset of sites for interventioncited by 1
- 2026Decision-Aware Evaluation of Physics-Informed Surrogatescited by 1
Built on by
Not cited yet.