Neural Random Subspace

Yun-Hao Cao,Jianxin Wu,Hanchen Wang,Joan Lasenby
DOI: https://doi.org/10.1016/j.patcog.2020.107801
IF: 8
2021-01-01
Pattern Recognition
Abstract:•End-to-end trainable ensemble method in a single model. We implement the random subspace method using existing CNN layers and it achieves both higher accuracy and faster inference speed than conventional forest methods. We show that such designs are attractive for many real-world tasks dealing with vector inputs.•Efficient and effective non-linear transformation. NRS can be seamlessly installed at a CNN as a non-linear component. NRS is more efficient than previous higher-order pooling methods and achieves higher accuracy than standard GAP with negligible additional cost. We demonstrate its effectiveness on both 2D image and 3D object recognition tasks under various CNN architectures.•Good generalization ability and transferability. The proposed method achieves better performance on various tasks when compared with different algorithms. Our method is easy to implement and can be easily extended to other deep learning libraries and deployed to different devices.
What problem does this paper attempt to address?