提出新型激活函数HeLU,解决ReLU死区问题且保持硬件高效。
Hysteresis Activation Function for Efficient Inference
- 采用可变阈值机制,训练时动态调整激活条件。
- 在多个数据集上提升模型泛化能力,性能媲美复杂函数。
- 适合追求高效推理的神经网络部署场景。
广泛使用的ReLU因推理时仅需一位符号判断而具备硬件高效性,但存在训练中神经元永久不激活的‘死区’问题。传统缓解方法常引入更复杂的激活函数,牺牲硬件友好性。本文提出滞回修正线性单元(HeLU),通过可变阈值机制优化反向传播,使简单激活函数在不增加复杂度或依赖归纳偏置的前提下,实现与复杂函数相当的性能。实证表明,HeLU显著提升多种数据集上的模型泛化能力,为各类神经网络架构提供高效可靠的推理解决方案。
原文摘要 · Abstract (English)
The widely used ReLU is favored for its hardware efficiency, {as the implementation at inference is a one bit sign case,} yet suffers from issues such as the ``dying ReLU'' problem, where during training, neurons fail to activate and constantly remain at zero, as highlighted by Lu et al. Traditional approaches to mitigate this issue often introduce more complex and less hardware-friendly activation functions. In this work, we propose a Hysteresis Rectified Linear Unit (HeLU), an efficient activation function designed to address the ``dying ReLU'' problem with minimal complexity. Unlike traditional activation functions with fixed thresholds for training and inference, HeLU employs a variable threshold that refines the backpropagation. This refined mechanism allows simpler activation functions to achieve competitive performance comparable to their more complex counterparts without introducing unnecessary complexity or requiring inductive biases. Empirical evaluations demonstrate that HeLU enhances model generalization across diverse datasets, offering a promising solution for efficient and effective inference suitable for a wide range of neural network architectures.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。