arXiv:2506.09714cs.LG2025-06NeurIPS被引 4

用新架构让网络自动压缩计算量,提升效率与泛化能力

Auto-Compressing Networks

  • 用长连接替代残差连接,实现训练中的自动信息压缩
  • 减少18%遗忘率,压缩30%-80%结构,保持精度
  • 适合低数据、持续学习、抗噪等实际场景

深度神经网络中短残差连接虽成功,但增加深度常带来计算冗余而无表征提升。本文提出自压缩网络(ACNs),以每层到输出的长前馈连接取代传统短残差连接。分析表明,该设计引发一种称为“自压缩”的特性:网络在梯度下降训练中通过架构设计自发压缩信息,将关键信息动态集中于浅层,增强其表征能力,暴露深层冗余。理论证明此性质源于层间动态使用模式。实验显示,相比残差网络,ACNs具有更强噪声鲁棒性、低数据下更优性能、更好迁移学习能力,且显著缓解灾难性遗忘,最多降低18%遗忘率,实现30%-80%架构压缩,同时保持视觉变换器、MLP-mixer、BERT等模型的准确率。结果表明ACNs是高效自适应架构的实用方案,可自动调整计算开销以匹配任务复杂度,学习出适用于噪声现实与持续学习的鲁棒表示。

原文摘要 · Abstract (English)

Deep neural networks with short residual connections have demonstrated remarkable success across domains, but increasing depth often introduces computational redundancy without corresponding improvements in representation quality. We introduce Auto-Compressing Networks (ACNs), an architectural variant where additive long feedforward connections from each layer to the output replace traditional short residual connections. By analyzing the distinct dynamics induced by this modification, we reveal a unique property we coin as auto-compression, the ability of a network to organically compress information during training with gradient descent, through architectural design alone. Through auto-compression, information is dynamically "pushed" into early layers during training, enhancing their representational quality and revealing potential redundancy in deeper ones. We theoretically show that this property emerges from layer-wise training patterns present in ACNs, where layers are dynamically utilized during training based on task requirements. We also find that ACNs exhibit enhanced noise robustness compared to residual networks, superior performance in low-data settings, improved transfer learning capabilities, and mitigate catastrophic forgetting suggesting that they learn representations that generalize better despite using fewer parameters. Our results demonstrate up to 18% reduction in catastrophic forgetting and 30-80% architectural compression while maintaining accuracy across vision transformers, MLP-mixers, and BERT architectures. These findings establish ACNs as a practical approach to developing efficient neural architectures that automatically adapt their computational footprint to task complexity, while learning robust representations suitable for noisy real-world tasks and continual learning scenarios.

神经网络架构自压缩高效模型持续学习

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