Neural Forest Learning.

Yun-Hao Cao,Jianxin Wu,Hanchen Wang,Joan Lasenby
2019-01-01
Abstract:The random subspace method, known as the pillar of random forests, is good atmaking precise and robust predictions. However, there is not a straightforwardway yet to combine it with deep learning. In this paper, we therefore proposeNeural Random Subspace (NRS), a novel deep learning based random subspacemethod. In contrast to previous forest methods, NRS enjoys the benefits ofend-to-end, data-driven representation learning, as well as pervasive supportfrom deep learning software and hardware platforms, hence achieving fasterinference speed and higher accuracy. Furthermore, as a non-linear component tobe encoded into Convolutional Neural Networks (CNNs), NRS learns non-linearfeature representations in CNNs more efficiently than previous higher-orderpooling methods, producing good results with negligible increase in parameters,floating point operations (FLOPs) and real running time. Compared with randomsubspaces, random forests and gradient boosting decision trees (GBDTs), NRSachieves superior performance on 35 machine learning datasets. Moreover, onboth 2D image and 3D point cloud recognition tasks, integration of NRS with CNNarchitectures achieves consistent improvements with minor extra cost. Code isavailable at https://github.com/CupidJay/NRS_pytorch.
What problem does this paper attempt to address?