arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.08232cs.LG

SwishReLU:一种用于提升深度神经网络性能的激活函数统一方法

SwishReLU: A Unified Approach to Activation Functions for Enhanced Deep Neural Networks Performance

Jiangsu University(江苏大学);Abdus Salam School of Mathematical Sciences, Govt College University(政府学院大学阿卜杜斯·萨拉姆数学科学学院);Metaverse Research Institute, School of Computer Science and Cyber Engineering, Guangzhou University(广州大学计算机科学与网络工程学院元宇宙研究院)

Jamshaid Ul Rahman, Rubiqa Zulfiqar, Asad Khan, Nimra

AI总结:

针对ReLU激活函数的死亡神经元问题及现有改进版本的不足,本文提出融合ReLU与Swish特性的SwishReLU激活函数,在多个数据集上验证其性能更优、计算成本更低,在VGG16模型的CIFAR-10任务上准确率提升6%。

摘要:

ReLU是深度神经网络中常用的激活函数,易出现“死亡ReLU”问题。已有ELU、SeLU、Swish等若干增强版本被提出,但应用相对较少。不过由于这些改进激活函数的优势并不稳定,替换ReLU存在一定难度。Swish虽然具备类似ReLU的平滑过渡特性,但通常计算开销比ReLU更高。本文提出SwishReLU,这是一种融合ReLU与Swish特性的新型激活函数。研究结果表明,SwishReLU性能优于ReLU,且计算成本低于Swish。本文对不同类型的ReLU变体与SwishReLU展开检验与对比,具体在CIFAR-10、CIFAR-100和MNIST三个数据集上,将ELU、SeLU以及Tanh与SwishReLU进行比较。值得注意的是,在算法2所述的VGG16模型中应用SwishReLU,在CIFAR-10数据集上实现了6%的准确率提升。

英文摘要:

ReLU, a commonly used activation function in deep neural networks, is prone to the issue of "Dying ReLU". Several enhanced versions, such as ELU, SeLU, and Swish, have been introduced and are considered to be less commonly utilized. However, replacing ReLU can be somewhat challenging due to its inconsistent advantages. While Swish offers a smoother transition similar to ReLU, its utilization generally incurs a greater computational burden compared to ReLU. This paper proposes SwishReLU, a novel activation function combining elements of ReLU and Swish. Our findings reveal that SwishReLU outperforms ReLU in performance with a lower computational cost than Swish. This paper undertakes an examination and comparison of different types of ReLU variants with SwishReLU. Specifically, we compare ELU and SeLU along with Tanh on three datasets: CIFAR-10, CIFAR-100 and MNIST. Notably, applying SwishReLU in the VGG16 model described in Algorithm 2 yields a 6% accuracy improvement on the CIFAR-10 dataset.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑