Improved prediction rule ensembling through model-based data generation

Benny Markovitch, Marjolein Fokkema

Onderzoeksoutput: Bijdrage aan tijdschriftTijdschriftartikelAcademic

28 Downloads (Pure)

Samenvatting

Prediction rule ensembles (PRE) provide interpretable prediction models with relatively high accuracy.PRE obtain a large set of decision rules from a (boosted) decision tree ensemble, and achieves sparsitythrough application of Lasso-penalized regression. This article examines the use of surrogate modelsto improve performance of PRE, wherein the Lasso regression is trained with the help of a massivedataset generated by the (boosted) decision tree ensemble. This use of model-based data generationmay improve the stability and consistency of the Lasso step, thus leading to improved overallperformance. We propose two surrogacy approaches, and evaluate them on simulated and existingdatasets, in terms of sparsity and predictive accuracy. The results indicate that the use of surrogacymodels can substantially improve the sparsity of PRE, while retaining predictive accuracy, especiallythrough the use of a nested surrogacy approach.
Originele taal-2Engels
Artikelnummer2109.13672
Aantal pagina's16
TijdschriftarXiv
Volume2021
DOI's
StatusGepubliceerd - 28 sep. 2021

Trefwoorden

  • stat.ML
  • cs.LG

Vingerafdruk

Duik in de onderzoeksthema's van 'Improved prediction rule ensembling through model-based data generation'. Samen vormen ze een unieke vingerafdruk.

Citeer dit