激活函数在深度学习中扮演着非常重要的角色,它给网络赋予了非线性,从而使得神经网络能够拟合任意复杂的函数。
如果没有激活函数,无论多复杂的网络,都等价于单一的线性变换,无法对非线性函数进行拟合。
目前,深度学习中最流行的激活函数为 relu,但也有些新推出的激活函数,例如 swish、GELU 据称效果优于relu激活函数。
https://zhuanlan.zhihu.com/p/98472075
https://zhuanlan.zhihu.com/p/98863801
一,常用激活函数
- tf.nn.tanh:将实数压缩到-1到1之间,输出期望为0。主要缺陷为存在梯度消失问题,计算复杂度高。
- tf.nn.leaky_relu:对修正线性单元的改进,解决了死亡relu问题。
- tf.nn.elu:指数线性单元。对relu的改进,能够缓解死亡relu问题。
- tf.nn.selu:扩展型指数线性单元。在权重用tf.keras.initializers.lecun_normal初始化前提下能够对神经网络进行自归一化。不可能出现梯度爆炸或者梯度消失问题。需要和Dropout的变种AlphaDropout一起使用。
二,在模型中使用激活函数
在keras模型中使用激活函数一般有两种方式,一种是作为某些层的activation参数指定,另一种是显式添加layers.Activation激活层。
import numpy as np pandas as pd tensorflow as tf from tensorflow.keras layers,models tf.keras.backend.clear_session() model = models.Sequential() model.add(layers.Dense(32,input_shape = (None,16),activation = tf.nn.relu)) #通过activation参数指定 model.add(layers.Dense(10)) model.add(layers.Activation(tf.nn.softmax)) 显式添加layers.Activation激活层 model.summary()
参考:
开源电子书地址:https://lyhue1991.github.io/eat_tensorflow2_in_30_days/
GitHub 项目地址:https://github.com/lyhue1991/eat_tensorflow2_in_30_days
原文链接:/tensorflow/991517.html