從隨機(jī)梯度下降到Mini-Batch

梯度下降法(gradient descent)是最小化目標(biāo)函數(shù)時(shí)最容易想到的方法馒吴,但是其缺點(diǎn)也很明顯:非常的慢扎运。原因在于,在運(yùn)行梯度下降時(shí)饮戳,需要遍歷整個(gè)訓(xùn)練集豪治,才能進(jìn)行一步梯度下降,為了避免目標(biāo)函數(shù)的振蕩扯罐,學(xué)習(xí)率被限制在一個(gè)很小的范圍內(nèi)负拟,所以每一步梯度下降參數(shù)的增量非常有限,最后的結(jié)果就是算法運(yùn)行起來(lái)很慢歹河。

隨機(jī)梯度下降

一種更快的梯度下降法被稱(chēng)作隨機(jī)梯度下降(stochastic gradient descent)掩浙,對(duì)原始梯度下降做了一些改進(jìn)使得算法運(yùn)行得更快。

設(shè)目標(biāo)函數(shù)(cost function)為

f(\dot{\vec{w}}) = \sum_{i=1}^{m}f_i(\dot{\vec{w}},\vec{x_i},\vec{y_i})

其中\dot{\vec{w}}為權(quán)重張量秸歧,(\dot{\vec{x_i}},\dot{\vec{y_i}})為某一個(gè)訓(xùn)練樣本點(diǎn)厨姚。原始梯度下降的權(quán)重更新公式為

\Delta \dot{\vec{w}} = - \alpha \nabla_{\dot{\vec{w}}}f=-\alpha \sum_{i=1}^{m}\nabla_{\dot{\vec{w}}}f_i(\dot{\vec{w}},\vec{x_i},\vec{y_i})

隨機(jī)梯度下降的基本原理是用某個(gè)隨機(jī)的 \nabla_{\dot{\vec{w}}}f_i來(lái)替代整個(gè) \nabla_{\dot{\vec{w}}}f,設(shè)為那么原公式就變得簡(jiǎn)單起來(lái)

\Delta \dot{\vec{w}} = - \alpha \nabla_{\dot{\vec{w}}}f =  - \alpha \nabla_{\dot{\vec{w}}}f_{rand}

也就是說(shuō)只用計(jì)算一個(gè)訓(xùn)練樣本的梯度就能進(jìn)行一步梯度更新键菱,效率大大提高谬墙。下面一個(gè)很自然的問(wèn)題就是:該算法是否能收斂到最小值(附近)?答案是肯定的经备。由于

E(\nabla_{\dot{\vec{w}}}f_{rand}) = E(\nabla_{\dot{\vec{w}}}f)

那么當(dāng)學(xué)習(xí)率\alpha 取某些值的時(shí)候拭抬,在期望的意義下是收斂的,更加精細(xì)的證明可以看 這篇論文侵蒙。

即便隨機(jī)梯度下降在期望意義下收斂造虎,但在極小點(diǎn)附近

\nabla_{\dot{\vec{w}}}f_{rand}(\vec{w^*}) \neq0

這降低了梯度下降的精度,所以后來(lái)衍生了SAG纷闺,SVRG算凿,SDCA算法(先在這里挖個(gè)坑),根本目的就在于降低訓(xùn)練集方差導(dǎo)致的梯度方差急但,從而提升精度澎媒。

Mini-Batch梯度下降

Mini-Batch梯度下降也叫小批量梯度下降,基本原理是結(jié)合了原始的梯度下降(批量梯度)和隨機(jī)梯度下降的一種折中方案波桩。

具體來(lái)說(shuō)戒努,該算法將訓(xùn)練集分成若干個(gè)Mini-Batch(設(shè)為n),每個(gè)Mini-Batch含有相同的樣本數(shù)量(設(shè)為k),計(jì)算過(guò)程變?yōu)椋好勘闅v一個(gè)Mini-Batch更新一次梯度储玫。

\begin{aligned} for\space i \space from \space 1 \space to \space n&:\\\quad \Delta \dot{\vec{w}} &= - \alpha \nabla_{\dot{\vec{w}}}f=-\alpha \sum_{i=n_1}^{n_k}\nabla_{\dot{\vec{w}}}f_i(\dot{\vec{w}},\vec{x_i},\vec{y_i})\\ \dot{\vec{w}}&=\dot{\vec{w}}-  \Delta \dot{\vec{w}}\end{aligned}

上述過(guò)程為遍歷一次訓(xùn)練集所進(jìn)行的梯度更新侍筛。顯然,當(dāng)Mini-Batch中batch size設(shè)置為1時(shí)撒穷,就是隨機(jī)梯度法匣椰,當(dāng)batch-size設(shè)置為m(訓(xùn)練集大小)時(shí)端礼,就是原始梯度(批量梯度)法禽笑。所以Mini-Batch是批量梯度和隨機(jī)梯度的一種折中方案。在精度和速度上都做了一些取舍蛤奥。

至此佳镜,我們已經(jīng)學(xué)習(xí)了隨機(jī)梯度和Mini-Batch梯度下降,實(shí)踐中凡桥,Mini-Batch梯度要用得更多一些蟀伸。

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個(gè)濱河市缅刽,隨后出現(xiàn)的幾起案子啊掏,更是在濱河造成了極大的恐慌,老刑警劉巖衰猛,帶你破解...
    沈念sama閱讀 211,948評(píng)論 6 492
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件迟蜜,死亡現(xiàn)場(chǎng)離奇詭異,居然都是意外死亡腕侄,警方通過(guò)查閱死者的電腦和手機(jī)小泉,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 90,371評(píng)論 3 385
  • 文/潘曉璐 我一進(jìn)店門(mén),熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái)冕杠,“玉大人,你說(shuō)我怎么就攤上這事酸茴》衷ぃ” “怎么了?”我有些...
    開(kāi)封第一講書(shū)人閱讀 157,490評(píng)論 0 348
  • 文/不壞的土叔 我叫張陵薪捍,是天一觀的道長(zhǎng)笼痹。 經(jīng)常有香客問(wèn)我,道長(zhǎng)酪穿,這世上最難降的妖魔是什么凳干? 我笑而不...
    開(kāi)封第一講書(shū)人閱讀 56,521評(píng)論 1 284
  • 正文 為了忘掉前任,我火速辦了婚禮被济,結(jié)果婚禮上救赐,老公的妹妹穿的比我還像新娘。我一直安慰自己只磷,他們只是感情好经磅,可當(dāng)我...
    茶點(diǎn)故事閱讀 65,627評(píng)論 6 386
  • 文/花漫 我一把揭開(kāi)白布泌绣。 她就那樣靜靜地躺著,像睡著了一般预厌。 火紅的嫁衣襯著肌膚如雪阿迈。 梳的紋絲不亂的頭發(fā)上,一...
    開(kāi)封第一講書(shū)人閱讀 49,842評(píng)論 1 290
  • 那天轧叽,我揣著相機(jī)與錄音苗沧,去河邊找鬼。 笑死炭晒,一個(gè)胖子當(dāng)著我的面吹牛待逞,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播腰埂,決...
    沈念sama閱讀 38,997評(píng)論 3 408
  • 文/蒼蘭香墨 我猛地睜開(kāi)眼飒焦,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼!你這毒婦竟也來(lái)了屿笼?” 一聲冷哼從身側(cè)響起牺荠,我...
    開(kāi)封第一講書(shū)人閱讀 37,741評(píng)論 0 268
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤,失蹤者是張志新(化名)和其女友劉穎驴一,沒(méi)想到半個(gè)月后休雌,有當(dāng)?shù)厝嗽跇?shù)林里發(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 44,203評(píng)論 1 303
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡肝断,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 36,534評(píng)論 2 327
  • 正文 我和宋清朗相戀三年杈曲,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片胸懈。...
    茶點(diǎn)故事閱讀 38,673評(píng)論 1 341
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡担扑,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出趣钱,到底是詐尸還是另有隱情涌献,我是刑警寧澤,帶...
    沈念sama閱讀 34,339評(píng)論 4 330
  • 正文 年R本政府宣布首有,位于F島的核電站燕垃,受9級(jí)特大地震影響,放射性物質(zhì)發(fā)生泄漏井联。R本人自食惡果不足惜卜壕,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 39,955評(píng)論 3 313
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望烙常。 院中可真熱鬧轴捎,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開(kāi)封第一講書(shū)人閱讀 30,770評(píng)論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)。三九已至跃洛,卻和暖如春率触,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背汇竭。 一陣腳步聲響...
    開(kāi)封第一講書(shū)人閱讀 32,000評(píng)論 1 266
  • 我被黑心中介騙來(lái)泰國(guó)打工葱蝗, 沒(méi)想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人细燎。 一個(gè)月前我還...
    沈念sama閱讀 46,394評(píng)論 2 360
  • 正文 我出身青樓两曼,卻偏偏與公主長(zhǎng)得像,于是被迫代替她去往敵國(guó)和親玻驻。 傳聞我的和親對(duì)象是個(gè)殘疾皇子悼凑,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 43,562評(píng)論 2 349