On variance reduction in stochastic gradient descent and its asynchronous variants

1. Abstract

基于variancereduction(VR)的SGD算法,比SGD更好,不论是theoretically and empirically,但是异步版本没有被研究。本文为很多VR算法提供了一个unifying framework,然后在这个框架中提出了一种异步算法,并证明了很快的收敛速度。对于通用的稀疏的机器学习问题,能够达到线性的加速。

2. Introduction

在强凸假设下,VR随机算法比SGD的期望收敛速度更快,VR分析了problem structure,做了一些space-time的trade-off,能够减少因为随机梯度带来的varince。需要将同步的VR算法扩展到异步的并行和分布式环境。

3. Related work:

3.1. Primal VR方法

  • SAG(Minimizing Finite Sums with the StochasticAverage Gradient)

  • SAGA(SAGA: A fast incremental gradient methodwith support for non-strongly convex composite objectives)

  • SVRG(Accelerating stochastic gradient descent using predictive variance reduction)

  • S2GD(Semi-Stochastic Gradient Descent Methods)

3.2. Dual VR方法

  • SDCA(Stochastic dual coordinate ascent methods for regularized loss)

  • Finito(Finito: A faster, permutable incremental gradientmethod for big data problems)

3.3. 分析Dual方法和VR随机方法之间关系

  • New Optimization Methods for Machine Learning

3.4. VR的算法结构

VR的算法结构可以trace back to经典的非随机incremental梯度算法(Incremental gradient, subgradient, and proximal methods for convex optimization:A survey),但是现在被公认的是,随机性帮助取得更快的收敛

3.5. Proximal方法

  • A proximal stochastic gradient method with progressive variance reduction

3.6. 加速VR方法

  • Stochastic Proximal Gradient Descent with Acceleration Techniques

  • Accelerated mini-batch stochastic dual coordinate ascent)

3.7. 分析finite-sum问题lower-bound

  • A lower bound for the optimization of finite sums

3.8. 异步SGD算法

并行variants

  • Hogwild!: A Lock-Free Approach to Parallelizing Stochastic Gradient Descent

分布式variants

  • Distributed delayed stochastic optimization

  • Distributed asynchronous incremental subgradient methods

3.9. Coordinate descent方法的并行和分布式variants

  • Asynchronous stochastic coordinate descent: Parallelism and convergence properties

  • An asynchronous parallel stochastic coordinate descent algorithm

  • An asynchronous parallel stochastic coordinate descent algorithm

  • Iteration complexity of randomized block-coordinate descent methods for minimizing a composite function

3.10. mini-batch

  • Mini-Batch Semi-Stochastic Gradient Descent in the Proximal Setting. 将S2GD扩展到mini-batch上,因此允许并行运行,但是需要更多的同步,只能允许小的batch

4. VR随机算法的通用框架

4.1. 假设条件

  • L-Lipschitz条件
  • lamda-strongly凸函数(也可扩展到smooth convex函数)

4.2. 已有VR算法

x是参数,alpha是额外的参数,A是alpha的集合

  • SVRG每m轮iteration更新一次完整的A
  • SAGA每轮iteration更新A中的一个alpha
  • SAG也是每轮iteration更新一个alpha

4.3. 空间时间开销分析

  • m较大时,SVRG的计算开销小,但是收敛速度慢
  • SAG和SAGA更频繁地更新A,因此收敛速度更快,但是存储开销大

4.4. 通用算法

HSAG:结合不同VR算法的调度策略

5. 异步VR算法

类似HogWild!,单机多核环境,稀疏机器学习问题

6. 实验

  • l2-LR算法
  • 无锁的SVRG对训练步长(学习速度)和线程数的变化鲁棒性更强

7. 总结

  • a common platform
  • an synchronous algorithm
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 215,463评论 6 497
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,868评论 3 391
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 161,213评论 0 351
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,666评论 1 290
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,759评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,725评论 1 294
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,716评论 3 415
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,484评论 0 270
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,928评论 1 307
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,233评论 2 331
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,393评论 1 345
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,073评论 5 340
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,718评论 3 324
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,308评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,538评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,338评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,260评论 2 352

推荐阅读更多精彩内容

  • 周末还是算休息吧,不强求了……………
    鹤书嗣音阅读 187评论 0 0
  • 不说假话,福珠和俊亨是我近年来看过最美好的一对,离看要这剧已经三个月左右了吧,还是久久不能忘怀。所以今天把剧照设成...
    天涯路人各不相干阅读 820评论 0 0
  • 还记得二十年前,那部风靡全国的电视剧《渴望》吗? 记得那时候,电视还是一种稀罕物,没有电视的人家,每天搬个小马扎去...
    李在在阅读 682评论 3 3