Skip to main navigation Skip to search Skip to main content

Item Response Theory Based Ensemble in Machine Learning

  • Stony Brook University

Research output: Contribution to journalArticlepeer-review

34 Scopus citations

Abstract

In this article, we propose a novel probabilistic framework to improve the accuracy of a weighted majority voting algorithm. In order to assign higher weights to the classifiers which can correctly classify hard-to-classify instances, we introduce the item response theory (IRT) framework to evaluate the samples’ difficulty and classifiers’ ability simultaneously. We assigned the weights to classifiers based on their abilities. Three models are created with different assumptions suitable for different cases. When making an inference, we keep a balance between the accuracy and complexity. In our experiment, all the base models are constructed by single trees via bootstrap. To explain the models, we illustrate how the IRT ensemble model constructs the classifying boundary. We also compare their performance with other widely used methods and show that our model performs well on 19 datasets.

Original languageEnglish
Pages (from-to)621-636
Number of pages16
JournalInternational Journal of Automation and Computing
Volume17
Issue number5
DOIs
StatePublished - Oct 1 2020

Keywords

  • Classification
  • ensemble learning
  • expectation maximization (EM) algorithm
  • item response theory
  • machine learning

Fingerprint

Dive into the research topics of 'Item Response Theory Based Ensemble in Machine Learning'. Together they form a unique fingerprint.

Cite this