用两个参数替换冻结层,大幅减少模型参数量且性能更优
Replacement Learning: Training Vision Tasks with Fewer Learnable Parameters
- 用两个可学习参数替代冻结层的全部参数
- 在多个数据集上参数减少、训练更快、内存更低
- 适合资源受限场景下的高效模型训练
传统端到端深度学习模型通过增加网络深度和复杂度提升特征表示能力,但导致参数冗余和资源效率低下,尤其在深层网络中更为严重。现有方法虽尝试跳过冗余层缓解问题,但仍面临性能差、计算复杂度高和内存使用低效等挑战。为此,我们提出一种名为替换学习(Replacement Learning)的新训练方法,通过将所有冻结层的参数完全替换为仅两个可学习参数,有效解决上述问题。具体而言,该方法选择性地冻结某些层参数,利用相邻层参数并通过由两个可学习参数控制的参数融合机制进行更新。该策略充分利用邻近结构信息,降低计算开销,节省GPU内存,并在历史上下文与新输入间保持平衡,最终提升整体模型性能。我们在CIFAR-10、STL-10、SVHN和ImageNet四个基准数据集上,采用CNNs和ViTs等多种架构验证了该方法的有效性。实验结果表明,该方法显著减少了参数数量、训练时间和内存消耗,同时全面超越端到端训练的性能表现。
原文摘要 · Abstract (English)
Traditional end-to-end deep learning models often enhance feature representation and overall performance by increasing the depth and complexity of the network during training. However, this approach inevitably introduces issues of parameter redundancy and resource inefficiency, especially in deeper networks. While existing works attempt to skip certain redundant layers to alleviate these problems, challenges related to poor performance, computational complexity, and inefficient memory usage remain. To address these issues, we propose an innovative training approach called Replacement Learning, which mitigates these limitations by completely replacing all the parameters of the frozen layers with only two learnable parameters. Specifically, Replacement Learning selectively freezes the parameters of certain layers, and the frozen layers utilize parameters from adjacent layers, updating them through a parameter integration mechanism controlled by two learnable parameters. This method leverages information from surrounding structures, reduces computation, conserves GPU memory, and maintains a balance between historical context and new inputs, ultimately enhancing overall model performance. We conducted experiments across four benchmark datasets, including CIFAR-10, STL-10, SVHN, and ImageNet, utilizing various architectures such as CNNs and ViTs to validate the effectiveness of Replacement Learning. Experimental results demonstrate that our approach reduces the number of parameters, training time, and memory consumption while completely surpassing the performance of end-to-end training.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。