Mini-batch梯度下降法


在这里插入图片描述

在大数据时代,数据量很大,如果我们每次都进行基于整个训练集的batch gradient descent 也就是批梯度下降会使得计算时间变得很长

所以 我们研究出了一种新的算法 mini-batch algorithm 其简单来说,就是将训练集拆分成n个mini-batch再进行分别处理,而不再是同时处理X和Y集

对训练集的拆分用 X { t } X^{\{t\}} X{t} 表示
在这里插入图片描述

经过拆分后,一次遍历训练集可以做t次梯度下降.

其与传统batch gradient descent的比较

在这里插入图片描述

可以看到,mini-batch的迭代会出现更多的noisy
后面博客会讲到可以使用动量梯度下降优化这个问题

选择mini-batch的size

在这里插入图片描述

极端情况

  1. 若将size设为m 那么就是 batch gradient descent

  2. 若将size设为1 那么就是随机梯度下降

    将mini-batch size的值设在两者的中间会使优化速度达到最快,因为既利用了向量化的优势,训练集也相对来说没有那么大,能够在一个训练集中实现 m/size 次梯度下降

在这里插入图片描述

一般来说,我们的选择都是2的n次方作为mini-batch 的size ,根据实验证明,这样的选择会更加具有普适性,但一般来说,size的选择需要匹配cpu/Gpu的内存,否则算法性能也会表现的很差.

  • 3
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值