Aug, 2023

基于 GBM 的 Bregman Proximal 约束学习算法

TL;DR现代机器学习面临了新的约束学习范式,其中包括最重要的应用领域如 Neyman-Pearson 分类和公平性分类,本文针对这些约束学习任务在 Bregman 近端算法框架下适配了梯度提升机算法,并提出了具有全局最优性保证的 Bregman 原始 — 对偶方法。我们的算法与现有的约束学习算法不同之处在于能够与 XGBoost 和 LightGBM 这样的公开实现无缝集成,并已通过实验证明了算法框架的有效性。虽然主要关注于 Neyman-Pearson 分类和公平性机器学习,但我们的框架在更广泛的约束学习应用领域也具有重要潜力。