arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2406.12837cs.LGcs.CV

LayerMerge:通过层剪枝与层合并实现神经网络深度压缩

LayerMerge: Neural Network Depth Compression through Layer Pruning and Merging

Seoul National University(首尔大学);Neural Processing Research Center(神经处理研究中心);Samsung - SAIT AI Lab, Montreal(三星SAIT人工智能实验室蒙特利尔分部);Google(谷歌)

Jinuk Kim, Marwa El Halabi, Mingi Ji, Hyun Oh Song

AI总结:

针对现有深度压缩方法合并层后核尺寸增大抵消延迟收益的问题,提出LayerMerge方法,通过动态规划求解代理优化问题联合剪枝卷积层与激活层,在多任务多架构上性能优于现有方法。

摘要:

现有研究表明,减少卷积神经网络的层数可在维持网络性能的同时提升效率。已有的深度压缩方法通过移除冗余非线性激活函数,将连续卷积层合并为单层来实现压缩,但这类方法存在关键缺陷:合并后层的卷积核尺寸会增大,大幅抵消了网络深度减少带来的延迟降低收益。本文证明,联合剪枝卷积层与激活函数可解决该问题。为此,我们提出LayerMerge(一种新型深度压缩方法),它能选择待移除的激活层与卷积层,在实现目标推理加速的同时最小化性能损失。由于对应的选择问题存在指数级搜索空间,我们构建了全新的代理优化问题,并通过动态规划高效求解。实验结果显示,在图像分类与生成任务的多种网络架构上,该方法的性能始终优于现有深度压缩与层剪枝方法。我们已在https://github.com/snu-mllab/LayerMerge 发布代码。

英文摘要:

Recent works show that reducing the number of layers in a convolutional neural network can enhance efficiency while maintaining the performance of the network. Existing depth compression methods remove redundant non-linear activation functions and merge the consecutive convolution layers into a single layer. However, these methods suffer from a critical drawback; the kernel size of the merged layers becomes larger, significantly undermining the latency reduction gained from reducing the depth of the network. We show that this problem can be addressed by jointly pruning convolution layers and activation functions. To this end, we propose LayerMerge, a novel depth compression method that selects which activation layers and convolution layers to remove, to achieve a desired inference speed-up while minimizing performance loss. Since the corresponding selection problem involves an exponential search space, we formulate a novel surrogate optimization problem and efficiently solve it via dynamic programming. Empirical results demonstrate that our method consistently outperforms existing depth compression and layer pruning methods on various network architectures, both on image classification and generation tasks. We release the code at https://github.com/snu-mllab/LayerMerge.

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

↑