Multi-Output Tree Chaining: An Interpretative Modelling and Lightweight Multi-Target Approach

Carregando...
Imagem de Miniatura

Data

2018-05-05

Autores

Mastelini, Saulo Martiello
da Costa, Victor Guilherme Turrisi
Santana, Everton Jose
Nakano, Felipe Kenji
Guido, Rodrigo Capobianco [UNESP]
Cerri, Ricardo
Barbon, Sylvio

Título da Revista

ISSN da Revista

Título de Volume

Editor

Resumo

Multi-target regression (MTR) regards predictive problems with multiple numerical targets. To solve this, machine learning techniques can model solutions treating each target as a separated problem based only on the input features. Nonetheless, modelling inter-target correlation can improve predictive performance. When performing MTR tasks using the statistical dependencies of targets, several approaches put aside the evaluation of each pair-wise correlation between those targets, which may differ for each problem. Besides that, one of the main drawbacks of the current leading MTR method is its high memory cost. In this paper, we propose a novel MTR method called Multi-output Tree Chaining (MOTC) to overcome the mentioned disadvantages. Our method provides an interpretative internal tree-based structure which represents the relationships between targets denominated Chaining Trees (CT). Different from the current techniques, we compute the outputs dependencies, one-by-one, based on the Random Forest importance metric. Furthermore, we proposed a memory friendly approach which reduces the number of required regression models when compared to a leading method, reducing computational cost. We compared the proposed algorithm against three MTR methods (Single-target - ST; Multi-Target Regressor Stacking - MTRS; and Ensemble of Regressor Chains - ERC) on 18 benchmark datasets with two base regression algorithms (Random Forest and Support Vector Regression). The obtained results show that our method is superior to the ST approach regarding predictive performance, whereas, having no significant difference from ERC and MTRS. Moreover, the interpretative tree-based structures built by MOTC pose as great insight on the relationships among targets. Lastly, the proposed solution used significantly less memory than ERC being very similar in predictive performance.

Descrição

Palavras-chave

Interpretative tree structure, Machine learning, Memory-friendly algorithm, Multi-output, Multi-target regression

Como citar

Journal of Signal Processing Systems, p. 1-25.