随机梯度下降(Stochastic Gradient Descent, SGD)是一种常用的优化算法,主要用于训练机器学习模型,尤其是神经网络。是训练优化神经网络的常用方法。
它的基本思想是基于单个样本或小批量样本来更新模型参数,从而加速优化过程。
简介
SGD的基本思想是通过逐个样本或小批量样本来更新模型参数,而不是使用整个数据集。这种方法大大提高了计算效率,特别是在处理大规模数据集时。
原理
SGD 的原理可以分为以下几个步骤:
这是一般的梯度下降算法的原理示意图:其中L函数是基于最小二乘描述拟合状态的损失函数,然后对于该函数对角度θ求偏导,再求平均。学习率是用来人为控制学习效率的。
在现实过程中,如果数据点足够多,那么再一一计算损失函数就会变得不现实,那么在每次计算时就会随机选取其中的某些点来计算损失函数,这样虽然难免会受到某些噪音的影响,但是通过多次计算,总朝着正确的方向收敛,这种影响是可以忽视的。
以上是简单的来源过程,下面会分布介绍:
- 初始化模型参数:随机选择初始参数值。
- 随机选择样本:从训练数据集中随机选择一个样本或一个小批量样本。
- 计算梯度:计算目标函数(例如损失函数)关于模型参数的梯度。
- 更新参数:根据梯度和学习率更新参数。公式如下: θ = θ − η ∇ θ J ( θ ; x i , y i ) 其中, ( θ ) 是模型参数, ( η ) 是学习率, ( ∇ θ J ( θ ; x i , y i ) ) 是损失函数关于参数的梯度。 \theta = \theta - \eta \nabla_{\theta} J(\theta; x_i, y_i) \其中,(\theta) 是模型参数,(\eta) 是学习率,(\nabla_{\theta} J(\theta; x_i, y_i)) 是损失函数关于参数的梯度。 θ=θ−η∇θJ(θ;xi,yi)其中,(θ)是模型参数,(η)是学习率,(∇θJ(θ;xi,yi))是损失函数关于参数的梯度。
- 重复:重复步骤2-4,直到达到停止条件(例如达到最大迭代次数或损失小于某个阈值)。
优劣分析
优点:
- 计算效率高:每次更新只使用一个样本或一个小批量样本,计算速度快,适合大规模数据集。
- 在线学习:SGD可以很容易地应用于在线学习,即通过连续获取数据流实时更新模型。
- 更好的模型泛化性:由于参数更新有一定的随机性,SGD有助于避免陷入局部最优解,从而获得更好的模型泛化性。
缺点:
- 收敛不稳定:由于每次只使用一个样本计算梯度,参数更新路径非常不稳定,可能导致优化过程中的振荡。
- 需要调整学习率:学习率的选择非常关键且敏感,通常需要仔细调整以获得最佳效果。
- 局部解问题:尽管随机性有助于避免陷入局部解,但它不总是能够找到全局最优解。
使用步骤
- 导入数据和库: 开始时,需要导入必要的库和数据集。例如,如果使用Python进行实现,可以使用如下代码:
import numpy as npimport matplotlib.pyplot as plt
- 初始化模型参数: 为模型参数赋初始值。假设我们要训练一个简单的线性回归模型 ( y = w x + b ) ,初始参数可以设为0或随机值。
w = np.random.randn()b = np.random.randn()
- 设置学习率和超参数: 设定学习率和其他超参数。例如:
learning_rate =0.01num_epochs =1000
- 定义损失函数: 定义我们要最小化的损失函数,比如均方误差(MSE)。
defcompute_loss(y_true, y_pred):return np.mean((y_true - y_pred)**2)
- 定义梯度计算: 根据损失函数定义梯度的计算方法。
defcompute_gradients(x, y, w, b): y_pred = w * x + b dw =-2* np.mean(x *(y - y_pred)) db =-2* np.mean(y - y_pred)return dw, db
- SGD更新步骤: 根据随机选择的样本计算梯度并更新模型参数。以下是循环内的实现方式:
for epoch inrange(num_epochs):# 随机选择一个样本 idx = np.random.randint(len(x_train)) x_sample = x_train[idx] y_sample = y_train[idx]# 计算梯度 dw, db = compute_gradients(x_sample, y_sample, w, b)# 更新参数 w = w - learning_rate * dw b = b - learning_rate * db # 打印损失信息if epoch %100==0: y_pred = w * x_train + b loss = compute_loss(y_train, y_pred)print(f'Epoch {epoch}, Loss: {loss}')
- 模型验证和评估: 在训练完成后,可以使用验证集或测试集来评估模型的性能。例如:
y_test_pred = w * x_test + btest_loss = compute_loss(y_test, y_test_pred)print(f'Test Loss: {test_loss}')
示例代码
以下是一个完整的示例代码,用于训练一个简单的线性回归模型,相信初学者可以对随机梯度下降法(SGD)有一个全面而深入的理解:
import numpy as np
import matplotlib.pyplot as plt
# 生成一些随机数据
np.random.seed(42)
x_train =2* np.random.rand(100,1)
y_train =4+3* x_train + np.random.randn(100,1)# 初始化参数
w = np.random.randn()
b = np.random.randn()# 超参数设置
learning_rate =0.01
num_epochs =1000# 定义损失函数defcompute_loss(y_true, y_pred):return np.mean((y_true - y_pred)**2)# 定义梯度计算defcompute_gradients(x, y, w, b):
y_pred = w * x + b
dw =-2* np.mean(x *(y - y_pred))
db =-2* np.mean(y - y_pred)return dw, db
# 训练过程for epoch inrange(num_epochs):# 随机选择一个样本
idx = np.random.randint(len(x_train))
x_sample = x_train[idx]
y_sample = y_train[idx]# 计算梯度
dw, db = compute_gradients(x_sample, y_sample, w, b)# 更新参数
w = w - learning_rate * dw
b = b - learning_rate * db
# 打印损失信息if epoch %100==0:
y_pred = w * x_train + b
loss = compute_loss(y_train, y_pred)print(f'Epoch {epoch}, Loss: {loss}')# 模型验证和评估
x_test = np.array([[1],[2]])
y_test =4+3* x_test
y_test_pred = w * x_test + b
test_loss = compute_loss(y_test, y_test_pred)print(f'Test Loss: {test_loss}')# 绘制拟合结果
plt.scatter(x_train, y_train, color='blue', label='Training data')
plt.plot(x_test, y_test_pred, color='red', label='Fitted line')
plt.legend()
plt.show()
改进:动量随机梯度下降
改进:学习率的自动调整
adagrad算法
RMSPROP算法
Adam算法
版权归原作者 We1ky 所有, 如有侵权,请联系我们删除。