Neural Random Forest Imitation

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Reinders, Christoph, Rosenhahn, Bodo
Natura: Preprint
Pubblicazione: 2019
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866911826673926144
author Reinders, Christoph
Rosenhahn, Bodo
author_facet Reinders, Christoph
Rosenhahn, Bodo
contents We present Neural Random Forest Imitation - a novel approach for transforming random forests into neural networks. Existing methods propose a direct mapping and produce very inefficient architectures. In this work, we introduce an imitation learning approach by generating training data from a random forest and learning a neural network that imitates its behavior. This implicit transformation creates very efficient neural networks that learn the decision boundaries of a random forest. The generated model is differentiable, can be used as a warm start for fine-tuning, and enables end-to-end optimization. Experiments on several real-world benchmark datasets demonstrate superior performance, especially when training with very few training examples. Compared to state-of-the-art methods, we significantly reduce the number of network parameters while achieving the same or even improved accuracy due to better generalization.
format Preprint
id arxiv_https___arxiv_org_abs_1911_10829
institution arXiv
publishDate 2019
record_format arxiv
spellingShingle Neural Random Forest Imitation
Reinders, Christoph
Rosenhahn, Bodo
Machine Learning
We present Neural Random Forest Imitation - a novel approach for transforming random forests into neural networks. Existing methods propose a direct mapping and produce very inefficient architectures. In this work, we introduce an imitation learning approach by generating training data from a random forest and learning a neural network that imitates its behavior. This implicit transformation creates very efficient neural networks that learn the decision boundaries of a random forest. The generated model is differentiable, can be used as a warm start for fine-tuning, and enables end-to-end optimization. Experiments on several real-world benchmark datasets demonstrate superior performance, especially when training with very few training examples. Compared to state-of-the-art methods, we significantly reduce the number of network parameters while achieving the same or even improved accuracy due to better generalization.
title Neural Random Forest Imitation
topic Machine Learning
url https://arxiv.org/abs/1911.10829