Accelerated Gradient Method for Multi-Task Sparse Learning Problem

被引:126
作者
Chen, Xi [1 ]
Pan, Weike [2 ]
Kwok, James T. [2 ]
Carbonell, Jaime G. [1 ]
机构
[1] Carnegie Mellon Univ, Sch Comp Sci, Pittsburgh, PA 15213 USA
[2] Hong Kong Univ Sci & Technol, Dept Comp Sci & Technol, Hong Kong, Peoples R China
来源
2009 9TH IEEE INTERNATIONAL CONFERENCE ON DATA MINING | 2009年
关键词
multi-task learning; L-1-infinity regularization; optimal method; gradient descend; SHRINKAGE;
D O I
10.1109/ICDM.2009.128
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Many real world learning problems can be recast as multi-task learning problems which utilize correlations among different tasks to obtain better generalization performance than learning each task individually. The feature selection problem in multi-task setting has many applications in fields of computer vision, text classification and bio-informatics. Generally, it can be realized by solving a L-1-infinity regularized optimization problem. And the solution automatically yields the joint sparsity among different tasks. However, due to the nonsmooth nature of the L-1-infinity norm, there lacks an efficient training algorithm for solving such problem with general convex loss functions. In this paper, we propose an accelerated gradient method based on an "optimal" first order black-box method named after Nesterov and provide the convergence rate for smooth convex loss functions. For nonsmooth convex loss functions, such as hinge loss, our method still has fast convergence rate empirically. Moreover, by exploiting the structure of the L-1-infinity ball, we solve the black-box oracle in Nesterov's method by a simple sorting scheme. Our method is suitable for large-scale multi-task learning problem since it only utilizes the first order information and is very easy to implement. Experimental results show that our method significantly outperforms the most state-of-the-art methods in both convergence speed and learning accuracy.
引用
收藏
页码:746 / +
页数:2
相关论文
共 26 条
  • [1] Ando RK, 2005, J MACH LEARN RES, V6, P1817
  • [2] [Anonymous], P INT C MACH LEARN
  • [3] [Anonymous], 2003, P INT C MACH LEARN
  • [4] [Anonymous], 2007, 200776 CORE
  • [5] [Anonymous], 2003, INTRO LECT CONVEX OP
  • [6] [Anonymous], 2006, MULTITASK FEATURE SE
  • [7] [Anonymous], 2008, P INT C MACH LEARN
  • [8] Convex multi-task feature learning
    Argyriou, Andreas
    Evgeniou, Theodoros
    Pontil, Massimiliano
    [J]. MACHINE LEARNING, 2008, 73 (03) : 243 - 272
  • [9] A Fast Iterative Shrinkage-Thresholding Algorithm for Linear Inverse Problems
    Beck, Amir
    Teboulle, Marc
    [J]. SIAM JOURNAL ON IMAGING SCIENCES, 2009, 2 (01): : 183 - 202
  • [10] Bertsekas D.P., 1999, NONLINEAR PROGRAMMIN