TeLU激活函数让深度学习更快更稳,可直接替换ReLU
TeLU Activation Function for Fast and Stable Deep Learning
- 设计新激活函数TeLU,结合ReLU简单性与平滑性优势
- 在多个基准上提升收敛速度,如ResNet18在ImageNet表现更优
- 无需调参即可替代ReLU,适合追求稳定高效的模型开发
我们提出超双曲正切指数线性单元(TeLU),其定义为 TeLU(x) = x·tanh(exp(x))。TeLU的设计基于关键激活函数的核心原理,在活跃区域近似恒等函数以实现强收敛,在饱和区域有效缓解梯度消失问题。其简洁形式提升了计算效率,增强可扩展性与收敛速度。不同于多数现代激活函数,TeLU无缝融合ReLU的简洁性与深度网络学习所需的平滑性和解析特性。它能模仿ReLU的行为与最优超参数设置,同时引入平滑性和曲率优势,成为理想的即插即用替代方案。其解析性质使其具备强大的通用逼近能力,显著提升多种实验中的鲁棒性与泛化性能。通过理论分析与实验验证,TeLU在多个挑战性基准上表现优异:包括ImageNet上的ResNet18、Text8上的Dynamic-Pooling Transformer,以及Penn TreeBank上的RNN。结果表明,TeLU有潜力成为激活函数的新标准,推动深度神经网络更高效、更稳定的训练,加速各领域的科学发现。
原文摘要 · Abstract (English)
We propose the Hyperbolic Tangent Exponential Linear Unit (TeLU), a neural network hidden activation function defined as TeLU(x)=xtanh(exp(x)). TeLU's design is grounded in the core principles of key activation functions, achieving strong convergence by closely approximating the identity function in its active region while effectively mitigating the vanishing gradient problem in its saturating region. Its simple formulation enhances computational efficiency, leading to improvements in scalability and convergence speed. Unlike many modern activation functions, TeLU seamlessly combines the simplicity and effectiveness of ReLU with the smoothness and analytic properties essential for learning stability in deep neural networks. TeLU's ability to mimic the behavior and optimal hyperparameter settings of ReLU, while introducing the benefits of smoothness and curvature, makes it an ideal drop-in replacement. Its analytic nature positions TeLU as a powerful universal approximator, enhancing both robustness and generalization across a multitude of experiments. We rigorously validate these claims through theoretical analysis and experimental validation, demonstrating TeLU's performance across challenging benchmarks; including ResNet18 on ImageNet, Dynamic-Pooling Transformers on Text8, and Recurrent Neural Networks (RNNs) on the Penn TreeBank dataset. These results highlight TeLU's potential to set a new standard in activation functions, driving more efficient and stable learning in deep neural networks, thereby accelerating scientific discoveries across various fields.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。