arXiv:2411.09475cs.LGcs.AI2024-11

改进残差连接的特征复用,提升深层网络训练效果

ResidualDroppath: Enhancing Feature Reuse over Residual Connections

  • 引入DropPath迭代与分阶段训练,增强特征复用机会
  • 在图像分类任务中,部分模型准确率显著提升
  • 适合研究深层网络优化与残差结构改进的读者

残差连接是缓解梯度消失、促进深层网络训练的关键组件。现有分析表明,其有效性部分源于促进特征复用。然而,我们发现原始残差连接在特征复用方面存在局限。为此,提出两种训练策略:一是采用DropPath,在训练中随机丢弃部分层以强制特征复用;二是冻结未丢弃层,专门训练被丢弃部分,使其学习依赖于已有特征的复用模式。实验表明,该方法在多个图像分类任务中提升了残差网络性能,验证了特征复用机制的强化对深层网络训练的积极作用。

原文摘要 · Abstract (English)

Residual connections are one of the most important components in neural network architectures for mitigating the vanishing gradient problem and facilitating the training of much deeper networks. One possible explanation for how residual connections aid deeper network training is by promoting feature reuse. However, we identify and analyze the limitations of feature reuse with vanilla residual connections. To address these limitations, we propose modifications in training methods. Specifically, we provide an additional opportunity for the model to learn feature reuse with residual connections through two types of iterations during training. The first type of iteration involves using droppath, which enforces feature reuse by randomly dropping a subset of layers. The second type of iteration focuses on training the dropped parts of the model while freezing the undropped parts. As a result, the dropped parts learn in a way that encourages feature reuse, as the model relies on the undropped parts with feature reuse in mind. Overall, we demonstrated performance improvements in models with residual connections for image classification in certain cases.

残差网络特征复用深度学习

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