arXiv:2511.04494cs.LGcs.CV2025-11NeurIPS被引 2

用输入分布信息压缩卷积网络,无需微调也能保持高精度。

Distribution-Aware Tensor Decomposition for Compression of Convolutional Neural Networks

  • 基于输入协方差设计新误差度量,优化函数空间输出变化。
  • 在多个数据集上压缩后准确率损失小于1%,且无需微调。
  • 方法可跨数据集迁移,原训练数据不可用时仍有效。

神经网络广泛用于图像任务,但计算开销大。训练完成后可通过压缩降低内存和计算负担。本文聚焦于通过张量分解与低秩表示实现压缩。不同于传统方法在权重空间最小化弗罗贝尼乌斯范数,我们采用数据驱动的范数,直接最小化层输出分布的变化,表达为 $\lVert (W - \widetilde{W}) Σ^{1/2}\rVert_F$,其中 $Σ^{1/2}$ 是层输入的协方差矩阵平方根。针对 Tucker-2 与 CPD 两种主流张量分解,提出新的交替最小二乘算法以直接优化该范数。与常规压缩流程不同,本方法常无需微调即可达到竞争性精度。进一步表明,该协方差范数可在不同数据集间迁移,仅带来轻微精度下降,使原始数据不可用时仍可压缩。在 ResNet-18/50、GoogLeNet 及 ImageNet、FGVC-Aircraft、Cifar10、Cifar100 等多个架构与数据集上的实验验证了该方法的优势。

原文摘要 · Abstract (English)

Neural networks are widely used for image-related tasks but typically demand considerable computing power. Once a network has been trained, however, its memory- and compute-footprint can be reduced by compression. In this work, we focus on compression through tensorization and low-rank representations. Whereas classical approaches search for a low-rank approximation by minimizing an isotropic norm such as the Frobenius norm in weight-space, we use data-informed norms that measure the error in function space. Concretely, we minimize the change in the layer's output distribution, which can be expressed as $\lVert (W - \widetilde{W}) Σ^{1/2}\rVert_F$ where $Σ^{1/2}$ is the square root of the covariance matrix of the layer's input and $W$, $\widetilde{W}$ are the original and compressed weights. We propose new alternating least square algorithms for the two most common tensor decompositions (Tucker-2 and CPD) that directly optimize the new norm. Unlike conventional compression pipelines, which almost always require post-compression fine-tuning, our data-informed approach often achieves competitive accuracy without any fine-tuning. We further show that the same covariance-based norm can be transferred from one dataset to another with only a minor accuracy drop, enabling compression even when the original training dataset is unavailable. Experiments on several CNN architectures (ResNet-18/50, and GoogLeNet) and datasets (ImageNet, FGVC-Aircraft, Cifar10, and Cifar100) confirm the advantages of the proposed method.

模型压缩张量分解低秩近似分布感知

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