Multi-stage multi-task feature learning

Pinghua Gong, Jieping Ye, Changshui Zhang

Research output: Contribution to journalArticlepeer-review

62 Scopus citations

Abstract

Multi-task sparse feature learning aims to improve the generalization performance by exploiting the shared features among tasks. It has been successfully applied to many applications including computer vision and biomedical informatics. Most of the existing multi-task sparse feature learning algorithms are formulated as a convex sparse regularization problem, which is usually suboptimal, due to its looseness for approximating an ?0-type regularizer. In this paper, we propose a non-convex formulation for multi-task sparse feature learning based on a novel non-convex regularizer. To solve the non-convex optimization problem, we propose a Multi-Stage Multi-Task Feature Learning (MSMTFL) algorithm; we also provide intuitive interpretations, detailed convergence and reproducibility analysis for the proposed algorithm. Moreover, we present a detailed theoretical analysis showing thatMSMTFL achieves a better parameter estimation error bound than the convex formulation. Empirical studies on both synthetic and real-world data sets demonstrate the effectiveness of MSMTFL in comparison with the state of the art multi-task sparse feature learning algorithms.

Original languageEnglish (US)
Pages (from-to)2979-3010
Number of pages32
JournalJournal of Machine Learning Research
Volume14
StatePublished - Oct 2013

Keywords

  • Multi-stage
  • Multi-task learning
  • Non-convex
  • Sparse learning

ASJC Scopus subject areas

  • Software
  • Artificial Intelligence
  • Control and Systems Engineering
  • Statistics and Probability

Fingerprint

Dive into the research topics of 'Multi-stage multi-task feature learning'. Together they form a unique fingerprint.

Cite this