arXiv Machine Learning

PruneShift: A Framework for Evaluating Decision Reliability in Structured Pruning

arXiv Machine Learning
Aug 4

When May a Model Replace the Experiment? Audits, Licenses, and the Price of Trust in Surrogate-Driven Design

arXiv:2608. 01378v1 Announce Type: new Abstract: Design campaigns in chemistry, materials science, and machine learning share a bottleneck: determining how good a candidate truly is requires an expensive evaluation - an experiment, a first-principles simulation, or a full training run.

By Shuangxiu (Max), Ma (Zachary), Wenhe (Zachary), Zhao
arXiv AI
Sep 10

PAC-CF: Calibrating Irreversible Frontier Pruning in LLM-Guided Search

PAC-CF introduces a method called Probably Approximately Correct Conformal Filtering to calibrate irreversible frontier pruning in large language model (LLM)-guided search. The approach treats tree pruning as a PAC‑guaranteed decision problem, deriving a conformal margin from score deficits on held‑out Native traces and applying a direct score‑gap filtering rule during deployment. Experiments across multiple domains and controllers show that PAC‑CF improves utility at various budgets and reduces workload metrics, notably boosting equal‑domain utility by 4.38 points and cutting physical requests by 18.95% with a 23.76% token reduction on a pruning‑aware ToolTree under a 100‑request budget.

By Tianhao Qian, Jiayu Chen, Lixu Wang
arXiv AI
Sep 10

Damage-Aware Bandit Pruning for Vision and Language Transformers

The paper introduces a structured post‑training pruning method for vision and language transformers called Damage‑Aware Bandit Pruning. It treats the selection of functional units (attention heads and MLP channel groups) as a multi‑armed bandit problem, using paired damage (masked loss minus base loss) as a reward to guide either UCB or Thompson Sampling policies. Experiments on a range of models (GPT‑2, OPT, Pythia, Qwen2.5, SmolLM2, ViT‑B/16, DeiT‑Tiny, Swin‑Tiny) show that the bandit approaches generally reduce degradation compared to budgeted‑greedy baselines, with statistically significant improvements in most comparisons.

By Salem Ameen, Sunil Vadera
arXiv Machine Learning
Oct 3

SimplexUQ: An Evaluation Framework and Benchmark for Conformal Uncertainty on Simplex-Valued Predictions

SimplexUQ introduces the first benchmark and reproducible protocol for evaluating how conformal prediction wrappers allocate coverage across simplex‑valued predictions. The framework, called SimplexTasks‑12, combines six synthetic regimes and six real tasks (e.g., class probabilities, topic mixtures, spectral abundances) to compare existing wrappers on metrics such as marginal coverage, worst‑stratum coverage, max disparity, and computational cost. Empirical results show that no single wrapper consistently dominates, with Mondrian and BatchMVP performing best in different settings, and that removing predictor bias only partially mitigates disparity.

By Liang You, Hengyu Shi, Dongwen Ou