arXiv:2602.14896cs.LG2026-02被引 1

用可复用模块组合压缩神经网络,让模型更简单且不丢性能。

Algorithmic Simplification of Neural Networks with Mosaic-of-Motifs

  • 将权重分成小块,从固定模板中选重复使用,降低复杂度。
  • 实验显示压缩后模型算法复杂度下降,性能几乎不变。
  • 适合关注模型轻量化与内在结构的算法研究者。

大规模深度学习模型易于压缩。在多种任务中,剪枝、量化和知识蒸馏等方法可在仅小幅损失性能的前提下大幅减少参数量。这引出核心问题:为何深度神经网络适合压缩?本文从算法复杂度视角解释这一现象,假设训练后模型参数更具结构,算法复杂度低于随机初始化时的权重。压缩方法正是利用了这种降低的复杂度。尽管无约束的权重参数化 $\mathbf{w} \in \mathbb{R}^n$ 可表示任意赋值,但训练所得解具有重复性和结构性,比简单程序更易实现。为此,我们通过 $\mathcal{K}(\mathbf{w})$ 定义权重的柯尔莫哥洛夫复杂度,并引入受限参数化 $\widehat{\mathbf{w}}$,将参数划分为大小为 $s$ 的块,每块从 $k$ 个可复用模板中选取,由重用模式(或拼图)决定。该方法称为 $\mathit{Mosaic\text{-}of\text{-}Motifs}$(MoMos),提供了理论上合理的参数化,引导优化向算法更简单的解收敛。多个实验表明,MoMos 在训练过程中持续降低神经网络的算法复杂度,同时保持原模型性能。结果表明,参数可压缩性不仅存在于训练后,还可通过优化域设计主动诱导。

原文摘要 · Abstract (English)

Large-scale deep learning models are well-suited for compression. Across a variety of tasks, methods like pruning, quantization, and knowledge distillation have been used to achieve massive reductions in model parameters with only marginal performance drops. This raises the central question: *Why are deep neural networks suited for compression?* In this work, we take up the perspective of algorithmic complexity to explain this behavior. We hypothesize that the parameters of trained models have more structure and, hence, exhibit lower algorithmic complexity compared to the weights at (random) initialization. Furthermore, model compression methods harness this reduced algorithmic complexity to compress models. Although an unconstrained parameterization of model weights, $\mathbf{w} \in \mathbb{R}^n$, can represent arbitrary weight assignments, the solutions found during training exhibit repeatability and structure, making them simpler to implement than a trivial program. To this end, we formalize the Kolmogorov complexity of $\mathbf{w}$ by $\mathcal{K}(\mathbf{w})$. We introduce a constrained parameterization $\widehat{\mathbf{w}}$ that partitions parameters into blocks of size $s$ and restricts each block to be selected from a set of $k$ reusable motifs, specified by a reuse pattern (or mosaic). The resulting method, $\mathit{Mosaic\text{-}of\text{-}Motifs}$ (MoMos), provides a theoretically justified parameterization that biases optimization toward algorithmically simpler solutions. Empirical evidence from multiple experiments shows that MoMos consistently lowers the algorithmic complexity of neural networks during training while preserving the performance of unconstrained models. These results suggest that parameter compressibility is not only observed after training, but can be induced from the optimization domain.

神经网络压缩算法复杂度可复用模块

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