arXiv:2505.17469cs.LGcs.AI2025-05被引 1

通过稀疏优化提升模型压缩,实现更优泛化性能。

Efficient compression of neural networks and datasets

  • 基于ℓ₀正则化与参数稀疏性近似描述长度
  • 压缩率高且精度损失小,在图像与文本任务上均有效
  • 适合追求高效模型部署与样本效率的研究者

压缩与泛化在所罗门诺夫归纳和最小描述长度原理(MDL)下本质相关,该原理预测:当数据来自低复杂度分布时,越简单的模型越能更好泛化。本文结合算法信息论与神经网络剪枝技术,提出一种通过识别最优数据压缩方法来提升模型泛化的思路。由于精确的MDL优化不可行,我们将其转化为ℓ₀正则化学习,并解释为何参数稀疏性可作为模型描述长度的有效可计算近似。为寻找最佳实用方法,我们系统比较并改进互补的稀疏优化策略:通过无需蒙特卡洛采样的新流程改进概率剪枝,用二分搜索减少平滑ℓ₀近似的超参数复杂度。在卷积网络和Transformer于图像与文本数据集上的实验表明,改进方法优于原有方案,实现显著模型压缩且精度损失极小,同时获得较短的数据描述长度。最后,我们在教师-学生设定中验证了所罗门诺夫归纳的预测:压缩模型具有更高的样本效率与更强的泛化能力。

原文摘要 · Abstract (English)

Compression and generalization are fundamentally related through Solomonoff induction and the minimum description length principle (MDL), which predict that simpler models generalize better when data arises from low-complexity distributions. In this article, we combine insights from algorithmic information theory and techniques from neural network pruning to improve model generalization by identifying the most effective data compression method. Since exact MDL optimization is intractable, we cast it as $\ell_0$ regularized learning and explain why parameter sparsity provides an effective computable approximation of model description length. To identify the best practical approach, we systematically compare and refine complementary sparse optimization methods. In particular, we improve probabilistic pruning through a procedure that does not require Monte Carlo sampling and refine smooth $\ell_0$ approximations with a binary search routine that reduces hyperparameter complexity. Across convolutional networks and transformers evaluated on image and text datasets, our refined methods improve upon their predecessors, achieve substantial model compression with minimal accuracy loss, and yield short data description lengths. Finally, we use these methods in a controlled teacher-student setting to empirically verify the prediction of Solomonoff induction that compressed models learn more sample-efficiently and generalize better.

模型压缩稀疏优化泛化能力ℓ₀正则化

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