Constrained optimization of objective functions determined from random forests

• 成果类型: Article
• 署名作者: Biggs, Max; Hariss, Rim; Perakis, Georgia
• 署名单位: University of Virginia; McGill University; Massachusetts Institute of Technology (MIT)
• 刊物名称: PRODUCTION AND OPERATIONS MANAGEMENT
• ISSN/ISSBN: 1059-1478
• DOI: 10.1111/poms.13877
• 发表日期: 2023
• 页码: 397-415
• 关键词: random forests; Machine Learning; Data-Driven Optimization

摘要

In this paper, we examine a data-driven optimization approach to making optimal decisions as evaluated by a trained random forest, where these decisions can be constrained by an arbitrary polyhedral set. We model this optimization problem as a mixed-integer linear program. We show this model can be solved to optimality efficiently using pareto-optimal Benders cuts for ensembles containing a modest number of trees. We consider a random forest approximation that consists of sampling a subset of trees and establish that this gives rise to near-optimal solutions by proving analytical guarantees. In particular, for axis-aligned trees, we show that the number of trees we need to sample is sublinear in the size of the forest being approximated. Motivated by this result, we propose heuristics inspired by cross-validation that optimize over smaller forests rather than one large forest and assess their performance on synthetic datasets. We present two case studies on a property investment problem and a jury selection problem. We show this approach performs well against other benchmarks while providing insights into the sensitivity of the algorithm's performance for different parameters of the random forest.