arXiv:2505.22074cs.LGcs.AI2025-05被引 9

给经典ReLU装上平滑梯度,让它不再‘死亡’,还能提升模型表现。

The Resurrection of the ReLU

  • 用平滑替代梯度,让ReLU在反向传播中不丢失更新能力。
  • 在VGG-16、ResNet-18等网络上显著提升泛化性能。
  • 适用于现代架构如Swin Transformer,效果媲美甚至优于GELU。

深度学习中的复杂激活函数(如GELU、SELU、SiLU)因平滑梯度和更好收敛性成为主流。然而,经典ReLU因其简洁性、稀疏性和拓扑优势仍具吸引力。其主要缺陷是‘死区问题’——神经元一旦失活便无法恢复。本文提出一种名为SUGAR的新型可插拔正则化方法:前向传播保持标准ReLU,反向传播使用平滑替代梯度,避免梯度归零。实验表明,SUGAR在VGG-16和ResNet-18等卷积网络上显著提升泛化性能,同时保持更稀疏激活并有效唤醒死掉的ReLU。进一步在Conv2NeXt与Swin Transformer等现代结构中替换GELU为SUGAR,仍取得竞争性甚至更优表现。结果挑战了‘先进激活函数才有效’的普遍认知,表明通过恰当梯度处理,传统ReLU可在多种视觉模型中焕发新生。

原文摘要 · Abstract (English)

Modeling sophisticated activation functions within deep learning architectures has evolved into a distinct research direction. Functions such as GELU, SELU, and SiLU offer smooth gradients and improved convergence properties, making them popular choices in state-of-the-art models. Despite this trend, the classical ReLU remains appealing due to its simplicity, inherent sparsity, and other advantageous topological characteristics. However, ReLU units are prone to becoming irreversibly inactive - a phenomenon known as the dying ReLU problem - which limits their overall effectiveness. In this work, we introduce surrogate gradient learning for ReLU (SUGAR) as a novel, plug-and-play regularizer for deep architectures. SUGAR preserves the standard ReLU function during the forward pass but replaces its derivative in the backward pass with a smooth surrogate that avoids zeroing out gradients. We demonstrate that SUGAR, when paired with a well-chosen surrogate function, substantially enhances generalization performance over convolutional network architectures such as VGG-16 and ResNet-18, providing sparser activations while effectively resurrecting dead ReLUs. Moreover, we show that even in modern architectures like Conv2NeXt and Swin Transformer - which typically employ GELU - substituting these with SUGAR yields competitive and even slightly superior performance. These findings challenge the prevailing notion that advanced activation functions are necessary for optimal performance. Instead, they suggest that the conventional ReLU, particularly with appropriate gradient handling, can serve as a strong, versatile revived classic across a broad range of deep learning vision models.

ReLU激活函数梯度优化模型泛化

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