arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1802.03604cs.LGcs.DCstat.ML
收藏

高维线性分类的分布式SVRG

Feature-Distributed SVRG for High-Dimensional Linear Classification

Nanjing University(南京大学)

Gong-Duo Zhang, Shen-Yi Zhao, Hao Gao, Wu-Jun Li

摘要:

线性分类在文本分类等高维应用中得到了广泛的应用。为了对大规模任务进行线性分类,我们通常需要在多台机器的集群上设计分布式学习方法。在本文中,我们提出了一种新的分布式学习方法,称为特征分布的随机方差约简梯度(FD-SVRG)的高维线性分类。与大多数现有的分布式学习方法(实例分布式)不同,FD-SVRG是特征分布式的。当数据维数大于数据实例数时,FD-SVRG具有比其他实例分布式方法更低的通信开销。实验结果表明,当样本维数大于训练数据的样本数时,FD-SVRG在通信开销和挂钟时间方面都优于其他分布式高维线性分类方法.

英文摘要:

Linear classification has been widely used in many high-dimensional applications like text classification. To perform linear classification for large-scale tasks, we often need to design distributed learning methods on a cluster of multiple machines. In this paper, we propose a new distributed learning method, called feature-distributed stochastic variance reduced gradient (FD-SVRG) for high-dimensional linear classification. Unlike most existing distributed learning methods which are instance-distributed, FD-SVRG is feature-distributed. FD-SVRG has lower communication cost than other instance-distributed methods when the data dimensionality is larger than the number of data instances. Experimental results on real data demonstrate that FD-SVRG can outperform other state-of-the-art distributed methods for high-dimensional linear classification in terms of both communication cost and wall-clock time, when the dimensionality is larger than the number of instances in training data.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑