从零开始认识AI:梯度下降法
创作时间:
作者:
@小白创作中心
从零开始认识AI:梯度下降法
引用
CSDN
1.
https://blog.csdn.net/u014147522/article/details/139933573
梯度下降法是机器学习和深度学习中一种常用的优化算法,用于最小化目标函数。本文将从原理到实践,详细介绍梯度下降法的工作机制,并通过一个具体的Python代码示例帮助读者理解其应用。
1. 原理介绍
梯度下降法(Gradient Descent)是一种用于优化函数的迭代算法,广泛应用于机器学习和深度学习中,用来最小化一个目标函数。该目标函数通常代表模型误差或损失。
基本思想是从一个初始点出发,通过不断沿着目标函数的负梯度方向更新参数,逐步逼近函数的局部最小值(或者全局最小值)。梯度是目标函数相对于参数的导数,因此,负梯度方向是函数值下降最快的方向。
具体步骤如下:
- 初始化参数:随机选择模型初始参数 ( \theta_0 )。
- 计算梯度:计算目标函数关于当前参数的梯度 ( \nabla J(\theta) ),这里 ( J(\theta) ) 是目标函数。
- 更新参数:根据梯度和学习率 ( \alpha ) 更新参数:
[
\theta := \theta - \alpha \nabla J(\theta)
]
其中,学习率 ( \alpha ) 是一个预先设置的超参数,决定了每一步更新的大小。 - 重复:重复步骤 2 和 3 直到收敛,即参数不再发生显著变化或者达到预设的迭代次数。
变种
梯度下降法有几种常见变种:
- 批量梯度下降(Batch Gradient Descent):每一步更新使用整个训练数据集计算梯度。这对于大规模数据集可能会非常耗时。
- 随机梯度下降(Stochastic Gradient Descent, SGD):每一步更新使用单个样本计算梯度,通过不断地用单个样本更新,效率较高但引入了较大的波动。
- 小批量梯度下降(Mini-batch Gradient Descent):每一步更新使用一个小批量的样本来计算梯度,结合了批量梯度下降和随机梯度下降的优点。
优化
为了提高梯度下降的效率和效果,可以结合一些优化方法,如:
- 动量法:在每一步更新中添加动量,帮助跳出局部最小值。
- AdaGrad、RMSprop、Adam:这些算法通过自适应调整学习率,以适应不同参数和不同迭代阶段。
梯度下降法是机器学习和深度学习中的关键技术之一,通过梯度下降可以有效地训练模型并优化目标函数。
2. 代码实现
以下用梯度下降法求解函数的最小值:
import numpy as np
import matplotlib.pyplot as plt
def f(x):
return x ** 2 + 5 * np.sin(x)
def df(x):
return 2 * x + 5 * np.cos(x)
# initialize parameter and learning rate
x = 4
lr = 0.1
epochs = 1000
history = [x]
for _ in range(epochs):
x = x - lr * df(x)
history.append(x)
xs = np.linspace(-5, 5, 200)
ys = f(xs)
plt.plot(xs, ys, label="f(x)")
plt.xlabel("x")
plt.ylabel("f(x)")
plt.scatter(history, f(np.array(history)), c="g", alpha=0.5, label="Gradient Descent Point")
plt.scatter(history[-1], f(history[-1]), c="r", label="Minimal Point")
plt.legend()
plt.show()
热门推荐
桂花酱的营养价值和桂花酱的食疗功效 怎么吃好
佛山机场新航站楼即将启用:全球首个模块化建设航站楼
韩亚空难后续:波音公司的责任大揭秘
阿塞拜疆、韩国空难背后的大国博弈
托马斯·杰斐逊:美国两党制的奠基者
冰糖桂花制作详解:从选材到烹饪,全方位教你打造美味甜品
冬至游台儿庄:一碗黄花牛肉面里的南北文化交融
台儿庄古城非遗美食节:创新演绎“非遗+”模式,打造文旅融合新典范
秋冬必打卡:台儿庄古城兰祺酒店&回味老街美食攻略
台儿庄古城必吃攻略:正宗枣庄辣子鸡在哪里?
托马斯·杰斐逊:美国民主的奠基人
如何保持良好的生活习惯?
郑州烩面:一碗汤鲜面筋的非遗美食
新能源公交司机必学:车辆维护秘籍
北京公交集团:如何守护司机职业健康?
揭秘公交司机为何不系安全带却不受罚?
马拉松期间,如何保障公交安全?
吕梁交警&狄晓丽教你冬季公交安全驾驶
公交车突发事故,这些逃生技巧你必须知道!
周末打卡:从龙阳路到上海海昌海洋公园
申沃客车教你公交车安全自救
地震来袭!如何冷静应对并自救?
秋冬练热瑜伽,舒筋活血效果杠杠滴!
八段锦:舒筋活血的最佳姿势
葡萄和山楂:舒筋活血的最佳拍档
【客家年俗】年初三,送穷鬼
客车自燃引发保险赔偿纠纷,法院判决保险公司全额赔付
跟Roy Wang学光绘摄影入门技巧
客车着火?教你秒变消防达人!
云茂高速客车起火事件:48名乘客安全疏散,这些自救知识请牢记!