English

Riemann-Lebesgue Forest for Regression

Machine Learning 2024-05-13 v3 Machine Learning

Abstract

We propose a novel ensemble method called Riemann-Lebesgue Forest (RLF) for regression. The core idea in RLF is to mimic the way how a measurable function can be approximated by partitioning its range into a few intervals. With this idea in mind, we develop a new tree learner named Riemann-Lebesgue Tree (RLT) which has a chance to perform Lebesgue type cutting,i.e splitting the node from response YY at certain non-terminal nodes. We show that the optimal Lebesgue type cutting results in larger variance reduction in response YY than ordinary CART \cite{Breiman1984ClassificationAR} cutting (an analogue of Riemann partition). Such property is beneficial to the ensemble part of RLF. We also generalize the asymptotic normality of RLF under different parameter settings. Two one-dimensional examples are provided to illustrate the flexibility of RLF. The competitive performance of RLF against original random forest \cite{Breiman2001RandomF} is demonstrated by experiments in simulation data and real world datasets.

Keywords

Cite

@article{arxiv.2402.04550,
  title  = {Riemann-Lebesgue Forest for Regression},
  author = {Tian Qin and Wei-Min Huang},
  journal= {arXiv preprint arXiv:2402.04550},
  year   = {2024}
}
R2 v1 2026-06-28T14:41:01.858Z