Skip to main navigation Skip to search Skip to main content

BVDT: A Boosted Vector Decision Tree Algorithm for Multi-Class Classification Problems

  • Beihang University

Research output: Contribution to journalArticlepeer-review

Abstract

In this paper, we propose a powerful weak learner (Vector Decision Tree (VDT)) and a new Boosted Vector Decision Tree (BVDT) algorithm framework for the task of multi-class classification. Unlike the traditional scalar valued boosting algorithms, the BVDT algorithm directly maps the feature space to the decision space in the multi-class setting, which facilitates convenient implementations of the multi-class classification algorithms using diverse loss functions. By viewing the explicit hard threshold on the leaf node value applied in the LogitBoost as a constraint optimization problem, we further develop two new variants of the BVDT algorithm: the L1-BVDT and the L2-BVDT. The performance of the proposed algorithm is evaluated on different datasets and compared with three state-of-the-art boosting algorithms, k-Nearest Neighbor (KNN) and Support Vector Machine (SVM). The results show that the performance of the proposed algorithm ranks first in all but one dataset and reduces the test error rate by 4% up to 58% with respect to the state-of-the-art boosting algorithms based on the scalar-valued weak learner. Furthermore, we present a case study on the Abalone dataset by designing a new loss function that combines the negative log-likelihood loss function of classification problem and square loss function of regression problem.

Original languageEnglish
Article number1750016
JournalInternational Journal of Pattern Recognition and Artificial Intelligence
Volume31
Issue number5
DOIs
StatePublished - 1 May 2017

Keywords

  • Multi-class classification
  • constraint optimization
  • gradient boosting
  • vector decision tree

Fingerprint

Dive into the research topics of 'BVDT: A Boosted Vector Decision Tree Algorithm for Multi-Class Classification Problems'. Together they form a unique fingerprint.

Cite this