LightGBM回归器输出负值，如何像XGBoost那样避免？

问题背景 #

LightGBM回归器模型会输出负值，而XGBoost中可通过设置objective='count:poisson'超参数避免输出负值，询问LightGBM是否有类似实现方式。

解决方案 #

LightGBM没有直接对应XGBoost count:poisson的目标函数，但可以通过以下方式实现非负输出：

• 使用泊松回归目标：设置objective='poisson'，LightGBM的泊松回归目标会约束模型输出为非负值，因为泊松分布的预测参数λ本身是非负的，模型训练时会自动适配这种约束。
• 后处理预测结果：对模型的原始预测值做截断处理，例如用max(pred, 0)将所有负值替换为0，这种方法简单直接，但可能不如使用对应目标函数的模型拟合效果好。
• 自定义目标函数：如果需要更灵活的逻辑，可以自行实现带非负约束的目标函数，同时定义对应的梯度和海森矩阵计算逻辑，让模型在训练阶段就考虑非负输出的要求。

另外，在LightGBM的官方讨论issue #5629中，也针对类似需求给出了上述核心方案的建议。

内容的提问来源于stack exchange，提问作者hersekai