用稀疏形态层替代激活函数,提升模型可压缩性与收敛速度。
Sparse Hybrid Linear-Morphological Networks
- 在线性层间插入形态层,取代传统激活函数,实现结构化稀疏。
- 在MTAT和CIFAR-10上,稀疏初始化形态层比ReLU等表现更优且收敛更快。
- 方法天然促进L1剪枝,适合轻量化部署场景。
我们研究了混合线性-形态网络。近期研究表明形态层具有固有的可剪枝性,但训练困难。本文提出一种混合网络结构:将形态层置于网络的线性层之间,替代激活函数。实验包含三类形态层:1)maxout池化层(形态层特例),2)全连接密集形态层,3)新型稀疏初始化变体。在Magna-Tag-A-Tune(音乐自动标注)和CIFAR-10(图像分类)数据集上,将SOTA卷积网络的线性分类头替换为所提结构。结果表明,该结构能诱导线性层产生稀疏性,在L1无结构剪枝下更具可剪枝性。在MTAT任务中,稀疏初始化形态层性能略优于ReLU、maxout及密集初始化最大加法层,且初始收敛更快。
原文摘要 · Abstract (English)
We investigate hybrid linear-morphological networks. Recent studies highlight the inherent affinity of morphological layers to pruning, but also their difficulty in training. We propose a hybrid network structure, wherein morphological layers are inserted between the linear layers of the network, in place of activation functions. We experiment with the following morphological layers: 1) maxout pooling layers (as a special case of a morphological layer), 2) fully connected dense morphological layers, and 3) a novel, sparsely initialized variant of (2). We conduct experiments on the Magna-Tag-A-Tune (music auto-tagging) and CIFAR-10 (image classification) datasets, replacing the linear classification heads of state-of-the-art convolutional network architectures with our proposed network structure for the various morphological layers. We demonstrate that these networks induce sparsity to their linear layers, making them more prunable under L1 unstructured pruning. We also show that on MTAT our proposed sparsely initialized layer achieves slightly better performance than ReLU, maxout, and densely initialized max-plus layers, and exhibits faster initial convergence.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。