arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2501.03095cs.CVcs.NE
收藏

一种用于深度神经网络权重共享压缩的新型结构无关多目标方法

A Novel Structure-Agnostic Multi-Objective Approach for Weight-Sharing Compression in Deep Neural Networks

Brock University(布洛克大学)

Rasa Khosrowshahli, Shahryar Rahnamayan, Beatrice Ombuki-Berman

AI总结:

提出一种结构无关的 MOEA 权重共享压缩框架,通过均匀码本量化、Pareto 优化与相邻区间迭代合并,在多数据集上实现最高约15倍内存压缩。

摘要:

深度神经网络在训练后需要在内存中存储数百万乃至数十亿个权重,这使得内存密集型模型难以部署在嵌入式设备上。权重共享技术是流行的压缩方法之一,它使用较少的权重值,并在网络中的特定连接之间共享这些值。本文提出一种基于多目标进化算法(MOEA)的压缩框架,该框架独立于神经网络架构、维度、任务和数据集。我们使用大小均匀的区间,将网络权重量化到单个码本(查找表)中,以实现高效的权重表示。利用 MOEA,我们通过优化两个目标来搜索 Pareto 最优的 k 个区间。随后,我们对非支配 Pareto 前沿解应用迭代合并技术,在不降低性能的情况下合并相邻区间,从而减少区间数量并提高压缩率。我们的方法独立于模型和层,这意味着来自任意层的权重会混合在聚类中;并且本文采用的均匀量化方法具有 O(N) 复杂度,而不是像 k-means 这类具有 O(Nkt) 复杂度的非均匀量化方法。此外,我们使用聚类中心作为共享权重值,而不是对共享权重进行重新训练,因为后者在计算上代价高昂。使用进化多目标优化的优势在于,它能够获得关于性能和共享权重的非支配 Pareto 前沿解。实验结果表明,我们能够在 CIFAR-10 上将神经网络内存降低 13.72~14.98 倍,在 CIFAR-100 上降低 11.61~12.99 倍,并在 ImageNet 上降低 7.44~8.58 倍,展示了所提出的深度神经网络压缩框架的有效性。

英文摘要:

Deep neural networks suffer from storing millions and billions of weights in memory post-training, making challenging memory-intensive models to deploy on embedded devices. The weight-sharing technique is one of the popular compression approaches that use fewer weight values and share across specific connections in the network. In this paper, we propose a multi-objective evolutionary algorithm (MOEA) based compression framework independent of neural network architecture, dimension, task, and dataset. We use uniformly sized bins to quantize network weights into a single codebook (lookup table) for efficient weight representation. Using MOEA, we search for Pareto optimal $k$ bins by optimizing two objectives. Then, we apply the iterative merge technique to non-dominated Pareto frontier solutions by combining neighboring bins without degrading performance to decrease the number of bins and increase the compression ratio. Our approach is model- and layer-independent, meaning the weights are mixed in the clusters from any layer, and the uniform quantization method used in this work has $O(N)$ complexity instead of non-uniform quantization methods such as k-means with $O(Nkt)$ complexity. In addition, we use the center of clusters as the shared weight values instead of retraining shared weights, which is computationally expensive. The advantage of using evolutionary multi-objective optimization is that it can obtain non-dominated Pareto frontier solutions with respect to performance and shared weights. The experimental results show that we can reduce the neural network memory by $13.72 \sim14.98 \times$ on CIFAR-10, $11.61 \sim 12.99\times$ on CIFAR-100, and $7.44 \sim 8.58\times$ on ImageNet showcasing the effectiveness of the proposed deep neural network compression framework.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑