arXiv:2511.19566cs.LGstat.ML2025-11NeurIPS被引 1

不依赖梯度和损失函数,用分布数据识别模型关键组件。

ModHiFi: Identifying High Fidelity predictive components for Model Modification

  • 基于局部重构误差衡量组件重要性,提出新指标Subset Fidelity。
  • Pruning与去学习任务中,速度提升11%,性能媲美现有方法。
  • 适用于资源受限场景,无需训练数据或标签,适合模型优化研究者。

开放权重模型普遍缺乏训练数据或损失函数访问权限,导致剪枝、去学习等模型修改任务难以开展。现有方法通常依赖梯度或真实标签,在计算资源受限环境下不可行。本文提出在仅能获取分布数据(如合成数据)的条件下,识别对模型预测性能至关重要的组件。理论证明:对于Lipschitz连续网络(如CNN与充分训练的Transformer),全局误差可被局部重构误差线性约束。据此设计基于子集重构能力的评估指标——子集保真度(Subset Fidelity)。在特征无关设定下,依据该分数选择组件即为最优。基于此提出ModHiFi算法,无需训练数据或损失函数即可实现模型修改。ModHiFi-P在ImageNet模型上实现11%加速,语言模型表现良好;ModHiFi-U在CIFAR-10上实现完全去学习且无需微调,于Swin Transformer上表现竞争力。

原文摘要 · Abstract (English)

Open weight models, which are ubiquitous, rarely provide access to their training data or loss function. This makes modifying such models for tasks such as pruning or unlearning, which are constrained by this unavailability, an active area of research. Existing techniques typically require gradients or ground-truth labels, rendering them infeasible in settings with limited computational resources. In this work, we investigate the fundamental question of identifying components that are critical to the model's predictive performance, without access to either gradients or the loss function, and with only distributional access such as synthetic data. We theoretically demonstrate that the global error is linearly bounded by local reconstruction errors for Lipschitz-continuous networks such as CNNs and well-trained Transformers (which, contrary to existing literature, we find exhibit Lipschitz continuity). This motivates using the locally reconstructive behavior of component subsets to quantify their global importance, via a metric that we term Subset Fidelity. In the uncorrelated features setting, selecting individual components based on their Subset Fidelity scores is optimal, which we utilize to propose ModHiFi, an algorithm for model modification that requires neither training data nor access to a loss function. ModHiFi-P, for structured pruning, achieves an 11\% speedup over the current state of the art on ImageNet models and competitive performance on language models. ModHiFi-U, for classwise unlearning, achieves complete unlearning on CIFAR-10 without fine-tuning and demonstrates competitive performance on Swin Transformers.

模型剪枝去学习无监督优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。