Collaboration Based Multi-Label Learning

Authors

  • Lei Feng Nanyang Technological University
  • Bo An Nanyang Technological University
  • Shuo He Southwestern University

DOI:

https://doi.org/10.1609/aaai.v33i01.33013550

Abstract

It is well-known that exploiting label correlations is crucially important to multi-label learning. Most of the existing approaches take label correlations as prior knowledge, which may not correctly characterize the real relationships among labels. Besides, label correlations are normally used to regularize the hypothesis space, while the final predictions are not explicitly correlated. In this paper, we suggest that for each individual label, the final prediction involves the collaboration between its own prediction and the predictions of other labels. Based on this assumption, we first propose a novel method to learn the label correlations via sparse reconstruction in the label space. Then, by seamlessly integrating the learned label correlations into model training, we propose a novel multi-label learning approach that aims to explicitly account for the correlated predictions of labels while training the desired model simultaneously. Extensive experimental results show that our approach outperforms the state-of-the-art counterparts.

Downloads

Published

2019-07-17

How to Cite

Feng, L., An, B., & He, S. (2019). Collaboration Based Multi-Label Learning. Proceedings of the AAAI Conference on Artificial Intelligence, 33(01), 3550-3557. https://doi.org/10.1609/aaai.v33i01.33013550

Issue

Section

AAAI Technical Track: Machine Learning