Bi-Directional Representation Learning for Multi-Label Classification
Abstract
Multi-label classification is a central problem in many application domains. In this paper, we present a novel supervised bi-directional model that learns a low-dimensional mid-level representation for multi-label classification. Unlike traditional multi-label learning methods which identify intermediate representations from either the input space or the output space but not both, the mid-level representation in our model has two complementary parts that capture intrinsic information of the input data and the output labels respectively under the autoencoder principle while augmenting each other for the target output label prediction. The resulting optimization problem can be solved efficiently using an iterative procedure with alternating steps, while closed-form solutions exist for one major step. Our experiments conducted on a variety of multi-label data sets demonstrate the efficacy of the proposed bi-directional representation learning model for multi-label classification.
Cite
Text
Li and Guo. "Bi-Directional Representation Learning for Multi-Label Classification." European Conference on Machine Learning and Principles and Practice of Knowledge Discovery in Databases, 2014. doi:10.1007/978-3-662-44851-9_14Markdown
[Li and Guo. "Bi-Directional Representation Learning for Multi-Label Classification." European Conference on Machine Learning and Principles and Practice of Knowledge Discovery in Databases, 2014.](https://mlanthology.org/ecmlpkdd/2014/li2014ecmlpkdd-bidirectional/) doi:10.1007/978-3-662-44851-9_14BibTeX
@inproceedings{li2014ecmlpkdd-bidirectional,
title = {{Bi-Directional Representation Learning for Multi-Label Classification}},
author = {Li, Xin and Guo, Yuhong},
booktitle = {European Conference on Machine Learning and Principles and Practice of Knowledge Discovery in Databases},
year = {2014},
pages = {209-224},
doi = {10.1007/978-3-662-44851-9_14},
url = {https://mlanthology.org/ecmlpkdd/2014/li2014ecmlpkdd-bidirectional/}
}