批量梯度下降,随机梯度下降,小批量梯度下降

梯度下降法是机器学习常用优化算法,有批量梯度下降(BGD)、随机梯度下降(SGD)和小批量梯度下降(MBGD)三种形式。BGD每次迭代用所有样本更新梯度,SGD每次用一个样本,MBGD则是每次用batch_size个样本。文中分析了三种方法的优缺点及收敛情况。
部署运行你感兴趣的模型镜像

https://www.cnblogs.com/lliuye/p/9451903.html

批量梯度下降(BGD)、随机梯度下降(SGD)以及小批量梯度下降(MBGD)的理解

  梯度下降法作为机器学习中较常使用的优化算法,其有着三种不同的形式:批量梯度下降(Batch Gradient Descent)、随机梯度下降(Stochastic Gradient Descent)以及小批量梯度下降(Mini-Batch Gradient Descent)。其中小批量梯度下降法也常用在深度学习中进行模型的训练。接下来,我们将对这三种不同的梯度下降法进行理解。
  为了便于理解,这里我们将使用只含有一个特征的线性回归来展开。此时线性回归的假设函数为:

hθ(x(i))=θ1x(i)+θ0hθ(x(i))=θ1x(i)+θ0


  其中 i=1,2,...,mi=1,2,...,m 表示样本数。
  对应的目标函数(代价函数)即为:

J(θ0,θ1)=12m∑i=1m(hθ(x(i))−y(i))2J(θ0,θ1)=12m∑i=1m(hθ(x(i))−y(i))2


  下图为 J(θ0,θ1)J(θ0,θ1) 与参数 θ0,θ1θ0,θ1 的关系的图:

 


 


1、批量梯度下降(Batch Gradient Descent,BGD)

  批量梯度下降法是最原始的形式,它是指在每一次迭代时使用所有样本来进行梯度的更新。从数学上理解如下:
  (1)对目标函数求偏导:

ΔJ(θ0,θ1)Δθj=1m∑i=1m(hθ(x(i))−y(i))x(i)jΔJ(θ0,θ1)Δθj=1m∑i=1m(hθ(x(i))−y(i))xj(i)


  其中 i=1,2,...,mi=1,2,...,m 表示样本数, j=0,1j=0,1 表示特征数,这里我们使用了偏置项 x(i)0=1x0(i)=1 。
  (2)每次迭代对参数进行更新:

θj:=θj−α1m∑i=1m(hθ(x(i))−y(i))x(i)jθj:=θj−α1m∑i=1m(hθ(x(i))−y(i))xj(i)


  注意这里更新时存在一个求和函数,即为对所有样本进行计算处理,可与下文SGD法进行比较。
  伪代码形式为:
  repeat{
       θj:=θj−α1m∑mi=1(hθ(x(i))−y(i))x(i)jθj:=θj−α1m∑i=1m(hθ(x(i))−y(i))xj(i)
      (for j =0,1)
  }


  优点:
  (1)一次迭代是对所有样本进行计算,此时利用矩阵进行操作,实现了并行。
  (2)由全数据集确定的方向能够更好地代表样本总体,从而更准确地朝向极值所在的方向。当目标函数为凸函数时,BGD一定能够得到全局最优。
  缺点:
  (1)当样本数目 mm 很大时,每迭代一步都需要对所有样本计算,训练过程会很慢。
  从迭代的次数上来看,BGD迭代的次数相对较少。其迭代的收敛曲线示意图可以表示如下:

 


 


2、随机梯度下降(Stochastic Gradient Descent,SGD)

  随机梯度下降法不同于批量梯度下降,随机梯度下降是每次迭代使用一个样本来对参数进行更新。使得训练速度加快。
  对于一个样本的目标函数为:

J(i)(θ0,θ1)=12(hθ(x(i))−y(i))2J(i)(θ0,θ1)=12(hθ(x(i))−y(i))2


  (1)对目标函数求偏导:

ΔJ(i)(θ0,θ1)θj=(hθ(x(i))−y(i))x(i)jΔJ(i)(θ0,θ1)θj=(hθ(x(i))−y(i))xj(i)


  (2)参数更新:

θj:=θj−α(hθ(x(i))−y(i))x(i)jθj:=θj−α(hθ(x(i))−y(i))xj(i)


  注意,这里不再有求和符号
  伪代码形式为:
  repeat{
    for i=1,...,m{
       θj:=θj−α(hθ(x(i))−y(i))x(i)jθj:=θj−α(hθ(x(i))−y(i))xj(i)
      (for j =0,1)
    }
  }


  优点:
  (1)由于不是在全部训练数据上的损失函数,而是在每轮迭代中,随机优化某一条训练数据上的损失函数,这样每一轮参数的更新速度大大加快。
  缺点:
  (1)准确度下降。由于即使在目标函数为强凸函数的情况下,SGD仍旧无法做到线性收敛。
  (2)可能会收敛到局部最优,由于单个样本并不能代表全体样本的趋势。
  (3)不易于并行实现。


  解释一下为什么SGD收敛速度比BGD要快:
  答:这里我们假设有30W个样本,对于BGD而言,每次迭代需要计算30W个样本才能对参数进行一次更新,需要求得最小值可能需要多次迭代(假设这里是10);而对于SGD,每次更新参数只需要一个样本,因此若使用这30W个样本进行参数更新,则参数会被更新(迭代)30W次,而这期间,SGD就能保证能够收敛到一个合适的最小值上了。也就是说,在收敛时,BGD计算了 10×30W10×30W次,而SGD只计算了 1×30W1×30W 次。


  从迭代的次数上来看,SGD迭代的次数较多,在解空间的搜索过程看起来很盲目。其迭代的收敛曲线示意图可以表示如下:

 


 


3、小批量梯度下降(Mini-Batch Gradient Descent, MBGD)

  小批量梯度下降,是对批量梯度下降以及随机梯度下降的一个折中办法。其思想是:每次迭代 使用 ** batch_size** 个样本来对参数进行更新。
  这里我们假设 batchsize=10batchsize=10 ,样本数 m=1000m=1000 。
  伪代码形式为:
  repeat{
    for i=1,11,21,31,...,991{
       θj:=θj−α110∑(i+9)k=i(hθ(x(k))−y(k))x(k)jθj:=θj−α110∑k=i(i+9)(hθ(x(k))−y(k))xj(k)
      (for j =0,1)
    }
  }


  优点:
  (1)通过矩阵运算,每次在一个batch上优化神经网络参数并不会比单个数据慢太多。
  (2)每次使用一个batch可以大大减小收敛所需要的迭代次数,同时可以使收敛到的结果更加接近梯度下降的效果。(比如上例中的30W,设置batch_size=100时,需要迭代3000次,远小于SGD的30W次)
  (3)可实现并行化。
  缺点:
  (1)batch_size的不当选择可能会带来一些问题。


  batcha_size的选择带来的影响:
  (1)在合理地范围内,增大batch_size的好处:
    a. 内存利用率提高了,大矩阵乘法的并行化效率提高。
    b. 跑完一次 epoch(全数据集)所需的迭代次数减少,对于相同数据量的处理速度进一步加快。
    c. 在一定范围内,一般来说 Batch_Size 越大,其确定的下降方向越准,引起训练震荡越小。
  (2)盲目增大batch_size的坏处:
    a. 内存利用率提高了,但是内存容量可能撑不住了。
    b. 跑完一次 epoch(全数据集)所需的迭代次数减少,要想达到相同的精度,其所花费的时间大大增加了,从而对参数的修正也就显得更加缓慢。
    c. Batch_Size 增大到一定程度,其确定的下降方向已经基本不再变化。


  下图显示了三种梯度下降算法的收敛过程:


您可能感兴趣的与本文相关的镜像

Dify

Dify

AI应用
Agent编排

Dify 是一款开源的大语言模型(LLM)应用开发平台,它结合了 后端即服务(Backend as a Service) 和LLMOps 的理念,让开发者能快速、高效地构建和部署生产级的生成式AI应用。 它提供了包含模型兼容支持、Prompt 编排界面、RAG 引擎、Agent 框架、工作流编排等核心技术栈,并且提供了易用的界面和API,让技术和非技术人员都能参与到AI应用的开发过程中

### 批量梯度下降随机梯度下降小批量梯度下降的区别 #### 定义与区别 - **批量梯度下降 (Batch Gradient Descent)** 这种方法在每次更新模型参数时都会使用整个数据集来计算损失函数的梯度。这种方法的优点在于它能够找到全局最小值,尤其是在凸优化问题中表现优异[^3]。然而,当数据集非常庞大时,这种做法可能会变得极其耗时且占用大量内存资源。 - **随机梯度下降 (Stochastic Gradient Descent, SGD)** 随机梯度下降通过每次仅选取一个样本进行梯度估计并据此调整权重的方式工作。相比批量梯度下降而言,SGD 的收敛速度更快,在处理大规模数据集时尤其如此[^3]。不过,由于其固有的高波动特性,可能导致最终解并非绝对意义上的最优解,而是接近于该点附近震荡的结果[^2]。 - **小批量梯度下降 (Mini-Batch Gradient Descent)** 小批量梯度下降综合了前两者的优点:既不像BGD那样需要等待全部数据完成一轮迭代才能前进一小步;也不像SGD那般剧烈摇摆不定。通常情况下会选择一个小规模的数据子集(称为minibatch),一般介于数十至数百个样本之间来进行每轮更新操作[^1]。此策略不仅保留了一定程度上的平稳性以便更有效地逼近目标极值点,同时也维持较高的运行效率以及较低的时间成本开销。 #### 优缺点分析 | 方法 | 计算代价 | 收敛行为 | 存储需求 | |-----------------------|---------------|------------------|--------------| | 批量梯度下降 | 极高 | 缓慢但精确 | 大 | | 随机梯度下降 | 较低 | 波动较大但仍可接受 | 小 | | 小批量梯度下降 | 中等偏高 | 更加稳定和平滑 | 可调节范围广 | ##### 批量梯度下降(BGD) - **优势**: 提供了一个较为稳定的路径朝向最低点移动; 对小型或中型尺寸的数据集合来说效果很好. - **劣势**: 当面对海量级别的输入资料时显得力不从心 – 更新过程缓慢而且耗费巨大. ##### 随机梯度下降(SGD) - **优势**: 显著减少了每次迭代所需时间; 能够快速逃离局部最优点从而有助于发现更好的整体解决方案. - **劣势**: 参数变化轨迹存在较多噪音干扰, 导致难以精准定位理想终点位置. ##### 小批量梯度下降(Mini-Batch GD) - **优势**: 结合了两者长处–既能保持相对合理的执行速率又能展现足够的稳健特征; 是目前大多数深度学习框架默认采用的技术方案之一. - **劣势**: 如果选择不当(mini-batches太小), 则可能丧失部分由更大群体所提供的统计规律支持所带来的好处. #### 应用场景探讨 对于不同类型的机器学习项目应该依据具体情况挑选合适的梯度下降变体: - 若您的问题是线性的并且拥有较小数量级别的训练实例,则可以考虑运用批处理版本因为它们往往能提供更加可靠的解答途径。 - 在涉及超大型数据库或者实时在线预测服务等领域里头,鉴于性能考量因素的影响,推荐优先选用基于单一样本或是固定数目组群形式下的即时反馈机制即所谓的“随机”模式及其衍生品们。 - 实际开发过程中最为常见的还是混合采用了多种技术手段的小批次方法论,因为它能够在保障一定质量水平的同时兼顾到运算效能方面的要求平衡点。 ```python import numpy as np def batch_gradient_descent(X, y, theta, alpha=0.01, iterations=100): m = len(y) for _ in range(iterations): error = np.dot(X, theta) - y gradient = (1/m)*np.dot(X.T, error) theta -= alpha * gradient return theta def stochastic_gradient_descent(X, y, theta, alpha=0.01, epochs=100): m = len(y) for epoch in range(epochs): for i in range(m): rand_index = np.random.randint(0,m) xi = X[rand_index,:].reshape(1,X.shape[1]) yi = y[rand_index] prediction = np.dot(xi,theta) error = prediction - yi gradient = xi.T*error theta -= alpha * gradient return theta def mini_batch_gradient_descent(X, y, theta, alpha=0.01, batch_size=32, epochs=100): m = len(y) for epoch in range(epochs): indices = list(range(m)) np.random.shuffle(indices) X_shuffled = X[indices] y_shuffled = y[indices] for i in range(0, m, batch_size): Xi = X_shuffled[i:i+batch_size] yi = y_shuffled[i:i+batch_size] error = np.dot(Xi, theta) - yi gradient = (1/batch_size)*(np.dot(Xi.T, error)) theta -= alpha * gradient return theta ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值