Ziheng Chen, Hongshik Ahn. Item Response Theory Based Ensemble in Machine Learning[J]. Machine Intelligence Research, 2020, 17(5): 621-636. DOI: 10.1007/s11633-020-1239-y
Citation: Ziheng Chen, Hongshik Ahn. Item Response Theory Based Ensemble in Machine Learning[J]. Machine Intelligence Research, 2020, 17(5): 621-636. DOI: 10.1007/s11633-020-1239-y

Item Response Theory Based Ensemble in Machine Learning

  • 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.
  • loading

Catalog

    /

    DownLoad:  Full-Size Img  PowerPoint
    Return
    Return