Hierarchical Multi-label Classification using Fully Associative Ensemble Learning

L. Zhang, S. K. Shah, I. A. Kakadiaris

Research output: Contribution to journalArticlepeer-review

63 Scopus citations

Abstract

Traditional flat classification methods (e.g., binary or multi-class classification) neglect the structural information between different classes. In contrast, Hierarchical Multi-label Classification (HMC) considers the structural information embedded in the class hierarchy, and uses it to improve classification performance. In this paper, we propose a local hierarchical ensemble framework for HMC, Fully Associative Ensemble Learning (FAEL). We model the relationship between each class node's global prediction and the local predictions of all the class nodes as a multi-variable regression problem with Frobenius norm or l1 norm regularization. It can be extended using the kernel trick, which explores the complex correlation between global and local prediction. In addition, we introduce a binary constraint model to restrict the optimal weight matrix learning. The proposed models have been applied to image annotation and gene function prediction datasets with tree structured class hierarchy and large scale visual recognition dataset with Direct Acyclic Graph (DAG) structured class hierarchy. The experimental results indicate that our models achieve better performance when compared with other baseline methods.

Original languageEnglish (US)
Pages (from-to)89-103
Number of pages15
JournalPattern Recognition
Volume70
DOIs
StatePublished - Oct 2017

Keywords

  • Ensemble learning
  • Hierarchical multi-label classification
  • Ridge regression

ASJC Scopus subject areas

  • Software
  • Signal Processing
  • Computer Vision and Pattern Recognition
  • Artificial Intelligence

Fingerprint

Dive into the research topics of 'Hierarchical Multi-label Classification using Fully Associative Ensemble Learning'. Together they form a unique fingerprint.

Cite this