arXiv:2605.19533cs.CV2026-05

用替换块减少训练参数,提升效率且不损失性能

Replacement Learning: Training Neural Networks with Fewer Parameters

论文配图:Replacement Learning: Training Neural Networks with Fewer Parameters
图 1 · 摘自论文原文
  • 用轻量层替代删除的网络块,融合前后块参数生成新算子
  • 在多个数据集上减少参数量、显存和训练时间,效果不降反升
  • 适合需要高效训练的模型优化场景,兼容多种任务与压缩技术

端到端全深度反向传播仍是优化深度神经网络的主流范式,但随着模型加深,其效率下降。由于每个模块必须在统一目标下执行并求导,全深度反向传播导致大量参数冗余、激活内存开销和训练延迟,尤其当相邻层学习模式高度相关时。直接跳过或移除层可降低开销,但常削弱表征能力或需特定架构设计。本文提出训练时的替换学习(RepL),通过替换而非简单丢弃选定模块来减少全深度冗余。对于每个被移除的模块,RepL插入一个轻量计算层,通过可学习变换从其前后相邻模块的参数中合成一个代理算子,并作用于前一激活。这样既保持局部上下文连续性,又避免不必要的全层计算。我们在CNN和ViT上构建了适配的参数融合模块,处理卷积通道、特征分辨率及Transformer子模块。在CIFAR-10、SVHN、STL-10、ImageNet、COCO和CityScapes上的大量实验表明,RepL在分类、检测和分割任务中均显著减少可训练参数、GPU内存占用和训练时间,同时达到或超过标准端到端训练性能。额外实验在WikiText-2、迁移学习、推理吞吐率、检查点、随机深度和INT8量化方面进一步验证其通用性与兼容性。

原文摘要 · Abstract (English)

End-to-end training with full-depth backpropagation remains the dominant paradigm for optimizing deep neural networks, but its efficiency deteriorates as models grow deeper. Since every block must be executed and differentiated under a single global objective, full-depth BP introduces substantial parameter redundancy, activation-memory cost, and training latency, especially when neighboring layers exhibit highly correlated learning patterns. Directly skipping or removing layers can reduce cost, but often weakens representation capacity or requires architecture-specific reuse designs. In this paper, we propose Replacement Learning (RepL), a training-time paradigm that reduces full-depth redundancy by replacing selected blocks rather than simply discarding them. For each removed block, RepL inserts a lightweight computing layer that synthesizes a surrogate operator from the parameters of its adjacent preceding and succeeding blocks through a learnable transformation, and applies the synthesized operator to the preceding activation. In this way, RepL preserves local contextual continuity while avoiding unnecessary full-layer computation. We instantiate RepL for CNNs and ViTs with tailored parameter-fusion blocks that handle convolutional channels, feature resolutions, and transformer submodules. Extensive experiments on CIFAR-10, SVHN, STL-10, ImageNet, COCO, and CityScapes show that RepL reduces trainable parameters, GPU memory usage, and training time while matching or surpassing standard end-to-end training across classification, detection, and segmentation. Additional results on WikiText-2, transfer learning, inference throughput, checkpointing, stochastic depth, and INT8 quantization further demonstrate its generality and compatibility.

神经网络训练参数压缩模型效率深度学习

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