Computer and Modernization

Previous Articles     Next Articles

Multi-target Prediction Algorithm Based on AdaBoost Regression Tree

  

  1. 1. School of Computer and Information Technology, Beijing Jiaotong University, Beijing 100044, China; 
    2. Beijing Key Lab of Traffic Data Analysis and Mining, Beijing 100044, China
  • Received:2017-02-16 Online:2017-09-20 Published:2017-09-19

Abstract: Real word prediction problems typically involve the simultaneous prediction of multiple target variables using the same set of predictive variables. When the target variables are binary, the prediction task is called multi-label classification, and when the target variable is real-valued, it is called multi-target prediction. This paper puts forward two new multi-target regression algorithms: Multi-Target Stacking(MTS) and Ensemble of Regressor Chains(ERC) which are inspired by two popular multi-label classification methods. Both MTS and ERC have the same baseline method, which are based on the single-target (ST) prediction model that is established by using 100 regression tree AdaBoost iterative algorithms. However, MTS and ERC extend the input space of the second stage by adding the target prediction value of the first stage. Both methods take into account the dependencies between the target variables, besides, ERC takes into account the order selection between targets. In addition, we also summarize the shortcomings of MTS and ERC methods, and propose the corrected versions denoted as MTS Corrected (MTSC) and ERC Corrected (ERCC). Experimental results show that the modified regression chain ERCC performs best in multi-objective prediction problems.

Key words: multi-target prediction, multi-label classification, single-target prediction, regressor chains, stacking

CLC Number: