写在前面
一直不太理解梯度下降算法是什么意思,今天我们就解开它神秘的面纱
写在中间
线性回归方程
如果要求出一条直线,我们只需知道直线上的两个不重合的点,就可以通过解方程组来求出直线
但是,如果我们选取的这两个点不在直线上,而是存在误差(暂且称作观测误差),这样求出的直线就会和原直线相差很大,我们应该怎样做呢?首先肯定不能只通过两个点,就武断地求出这条直线。
我们通常尽可能多地使用分布在直线周围的点,也可能不存在一条直线完美的穿过所有采样点。那么,退而求其次,我们希望能找到一条比较“好”的位于采样点中间的直线。那么怎么衡量“好”与“不好”呢?一个很自然的想法就是,求出当前模型的所有采样点上的预测值𝑤𝑥(𝑖) + 𝑏与真实值𝑦(𝑖)之间的差的平方和作为总误差 L \mathcal{L} L,然后搜索一组参数 w ∗ , b ∗ w^{*},b^{*} w∗,b∗使得 L \mathcal{L} L最小,对应的直线就是我们要寻找的最优直线。
w ∗ , b ∗ = arg min w , b 1 n ∑ i = 1 n ( w x ( i ) + b − y ( i ) ) 2 w^*,b^*=\arg\min_{w,b}\frac{1}{n}\sum_{i=1}^{n}\bigl(wx^{(i)}+b-y^{(i)}\bigr)^2 w∗,b∗=argminw,bn1∑i=1n(wx(i)+b−y(i))2
最后再通过梯度下降法来不断优化参数 w ∗ , b ∗ w^{*},b^{*} w∗,b∗
有基础的小伙伴们可能知道求误差的方法其实就是均方误差函数,不懂得可以看这篇文章补充养分《误差函数》 ,我们这篇文章就侧重梯度下降。
梯度下降
函数的梯度定义为函数对各个自变量的偏导数组成的向量。不会的话,翻翻高等数学下册书。
举个例子,对于曲面函数𝑧 = 𝑓(𝑥, 𝑦),函数对自变量𝑥的偏导数记为 ∂ z ∂ x \frac{\partial z}{\partial x} ∂x∂z,函数对自变量𝑦的偏导数记为 ∂ z ∂ y \frac{\partial z}{\partial y} ∂y∂z,则梯度∇𝑓为向量 ( ∂ z ∂ x , ∂ z ∂ y ) ({\frac{\partial z}{\partial x}},{\frac{\partial z}{\partial y}}) (∂x∂z,∂y∂z),梯度的方向总是指向当前位置函数值增速最大的方向,函数曲面越陡峭,梯度的模也越大。
函数在各处的梯度方向∇𝑓总是指向函数值增大的方向,那么梯度的反方向−∇𝑓应指向函数值减少的方向。利用这一性质,我们只需要按照下式来更新参数,,其中𝜂用来缩放梯度向量,一般设置为某较小的值,如 0.01、0.001 等。
x ′ = x − η ⋅ d y d x x'=x-\eta\cdot\frac{\mathrm{d}y}{\mathrm{d}x} x′=x−η⋅dxdy
结合上面的回归方程,我们就可对误差函数求偏导,以循环的方式更新参数 w , b w,b w,b:
w ′ = w − η ∂ L ∂ w b ′ = b − η ∂ L ∂ b \begin{aligned}w'&=w-\eta\frac{\partial\mathcal{L}}{\partial w}\\\\b'&=b-\eta\frac{\partial\mathcal{L}}{\partial b}\end{aligned} w′b′=w−η∂w∂L=b−η∂b∂L
函数实现
计算过程都需要包裹在 with tf.GradientTape() as tape
上下文中,使得前向计算时能够保存计算图信息,方便自动求导操作。通过tape.gradient()
函数求得网络参数到梯度信息,结果保存在 grads 列表变量中。
GradientTape()函数
GradientTape(persistent=False, watch_accessed_variables=True)
-
persistent
: 布尔值,用来指定新创建的gradient
tape是否是可持续性的。默认是False,意味着只能够调用一次GradientTape()函数,再次使用会报错 -
watch_accessed_variables
:布尔值,表明GradientTape()函数是否会自动追踪任何能被训练的变量。默认是True。要是为False的话,意味着你需要手动去指定你想追踪的那些变量。
tape.watch()函数
tape.watch()用于跟踪指定类型的tensor变量。
- 由于GradientTape()默认只对
tf.Variable
类型的变量进行监控。如果需要监控的变量是tensor
类型,则需要tape.watch()
来监控,否则输出结果将是None
tape.gradient()函数
tape.gradient(target, source)
-
target
:求导的因变量 -
source
:求导的自变量
import tensorflow as tf
w = tf.constant(1.)
x = tf.constant(2.)
y = x * w
with tf.GradientTape() as tape:
tape.watch([w])
y = x * w
grads = tape.gradient(y, [w])
print(grads)
写在最后
👍🏻点赞,你的认可是我创作的动力!
⭐收藏,你的青睐是我努力的方向!
✏️评论,你的意见是我进步的财富!