Distributed Representation Learning with Skip-Gram Model for Trained Random Forests

Chao Ma,Tianjun Wang,Le Zhang,Zhiguang Cao,Yue Huang,Xinghao Ding
DOI: https://doi.org/10.1016/j.neucom.2023.126434
IF: 6
2023-06-09
Neurocomputing
Abstract:The random forest family has been extensively studied due to its wide applications in machine learning and data analytics. However, the representation abilities of forests have not been explored yet. The existing forest representation is mainly based on feature hashing on the indices of leaf nodes. Feature hashing typically disregards the information from tree structures, i.e., the relationships between leaf nodes. Furthermore, the visualisation abilities of feature hashing are limited. On the contrary, the Skip-Gram model has been widely explored in word and node embedding due to its excellent representation ability. This paper proposes distributed representation learning for trained forests (DRL-TF) to extract co-occurrence relationships of samples and tree structures, and further boost the representation abilities of the trained forest using the Skip-Gram model. The experimental results demonstrate that the proposed DRL-TF outperforms the challenging baselines. To the best of the authors' knowledge, the visualisation by DRL-TF is the first tool to analyse the trained forests. The code is available at: (https://github.com/machao199271/DRL-TF).
computer science, artificial intelligence
What problem does this paper attempt to address?