A shared-subspace learning framework for multi-label classification

Shuiwang Ji, Lei Tang, Shipeng Yu, Jieping Ye

Research output: Contribution to journalArticle

134 Scopus citations

Abstract

Multi-label problems arise in various domains such as multi-topic document categorization, protein function prediction, and automatic image annotation. One natural way to deal with such problems is to construct a binary classifier for each label, resulting in a set of independent binary classification problems. Since multiple labels share the same input space, and the semantics conveyed by different labels are usually correlated, it is essential to exploit the correlation information contained in different labels. In this paper, we consider a general framework for extracting shared structures in multi-label classification. In this framework, a common subspace is assumed to be shared among multiple labels. We show that the optimal solution to the proposed formulation can be obtained by solving a generalized eigenvalue problem, though the problem is nonconvex. For high-dimensional problems, direct computation of the solution is expensive, and we develop an efficient algorithm for this case. One appealing feature of the proposed framework is that it includes several well-known algorithms as special cases, thus elucidating their intrinsic relationships. We further show that the proposed framework can be extended to the kernel-induced feature space. We have conducted extensive experiments on multi-topic web page categorization and automatic gene expression pattern image annotation tasks, and results demonstrate the effectiveness of the proposed formulation in comparison with several representative algorithms.

Original languageEnglish (US)
Article number8
JournalACM Transactions on Knowledge Discovery from Data
Volume4
Issue number2
DOIs
StatePublished - May 1 2010

Keywords

  • Gene expression pattern image annotation
  • Kernel methods
  • Least squares loss
  • Multi-label classification
  • Shared subspace
  • Singular value decomposition
  • Web page categorization

ASJC Scopus subject areas

  • Computer Science(all)

Fingerprint Dive into the research topics of 'A shared-subspace learning framework for multi-label classification'. Together they form a unique fingerprint.

  • Cite this