激活函数介绍

Title 激活函数介绍
Framework PyTorch
User wy8817399@vip.qq.com
Id 35
Created 1/31/26, 6:01 PM
Modified 1/31/26, 6:01 PM
Published Yes
Content

    激活函数的目的:

        给模型增加非线性功能, 让模型(神经元)既可以做分类, 还可以做回归问题.

    激活函数的分类:

        Sigmoid:

        ReLU:

        Tanh:

        Softmax:

 

    Sigmoid激活函数:

        主要应用于 二分类的输出层, 且适用于 浅层神经网络(不超过5层).

        数据在 [-6, 6]之间有效果, 在[-3, 3]之间效果明显, 会将数据值映射到: [0, 1]

        求导后范围在 [0, 0.25]

 

    Tanh:

        主要应用于 隐藏层, 且适用于 浅层神经网络(不超过5层).

        数据在 [-3, 3]之间有效果, 在[-1, 1]之间效果明显, 会将数据值映射到: [-1, 1]

        求导后范围在 [0, 1], 较之于Sigmoid, 收敛速度快.

 

    ReLU:

        计算公式为: max(0, x), 计算量相对较小, 训练成本低. 多应用于 隐藏层, 且适合 深层神经网络.

        求导后, 值要么是0, 要么是1, 较之于Tanh, 收敛速度更快.

        默认情况下ReLU只考虑 正样本, 可以使用LeakyReLU, PReLU 来考虑 正负样本.

 

 

    Softmax:

        将多分类的结果以概率的形式展示, 且概率和相加为1, 最终选取概率值最大的分类 作为最终结果.

 

记忆: 如何选择激活函数

    隐藏层:

        ReLU > Leaky ReLU > PReLU > Tanh > Sigmoid

    输出层:

        二分类: Sigmoid

        多分类: Softmax

        回归问题: identity