最详细的LightGBM参数介绍与深入分析

前言

我使用LightGBM有一段时间了，它一直是我处理大多数表格数据问题的首选算法。它有很多强大的功能，如果你还没有看过的话，我建议你去了解一下。

但我一直对了解哪些参数对性能影响最大，以及如何调整LightGBM参数以发挥最大作用很感兴趣。

我决定进行一些研究，深入了解LightGBM的参数，并分享我的经验。

1.梯度增强方法

使用 LightGBM，您可以运行不同类型的梯度增强方法。您有：GBDT、DART 和 GOSS，可以使用 boosting 参数指定。

在接下来的部分中，我将解释并比较这些方法。

1.1 lgbm gbdt（梯度提升决策树）

该方法是本文首次提出的传统梯度提升决策树，也是 XGBoost 和 pGBRT 等一些优秀库背后的算法。

如今，gbdt 因其准确性、高效性和稳定性而被广泛使用。

它基于三个重要原则：

弱学习器（决策树）
梯度优化
升压技术

所以在gbdt方法中我们有很多决策树（弱学习器）。这些树是按顺序构建的：

第一棵树学习如何适应目标变量
第二棵树学习如何适应第一棵树的预测和地面实况之间的残差（差异）
第三棵树学习如何拟合第二棵树的残差，依此类推。

所有这些树都是通过在整个系统中传播误差梯度来训练的。

gbdt 的主要缺点是在每个树节点中找到最佳分割点是一项耗时且消耗内存的操作，其他 boosting 方法试图解决该问题。

1.2 dart 梯度提升

在这篇出色的论文论文论文中，您可以了解有关 DART 梯度提升的所有知识，这是一种使用神经网络标准 dropout 的方法，以改进模型正则化并处理其他一些不太明显的问题。

也就是说，gbdt 存在过度专业化的问题，这意味着在后续迭代中添加的树往往只会影响少数实例的预测，而对其余实例的贡献可以忽略不计。添加 dropout 使得后续迭代中的树更难专门处理那些少数样本，从而提高模型性能。

1.3 lgbm goss（基于梯度的一侧采样）

事实上，将该方法命名为lightgbm的最重要原因是使用了基于本文的Goss方法。 Goss 是更新且更轻的 GBDT 实现（因此是“轻”GBM）。

标准 GBDT 很可靠，但在大型数据集上速度不够快。因此，戈斯提出了一种基于梯度的采样方法，以避免搜索整个搜索空间。我们知道，对于每个数据实例，当梯度很小时，意味着不用担心数据经过良好训练，而当梯度很大时，应该再次重新训练。所以我们这里有两个方面，即具有大梯度和小梯度的数据实例。因此，goss 保留所有具有大梯度的数据，并对具有小梯度的数据进行随机采样（这就是为什么它被称为单侧采样）。这使得搜索空间更小，goss 可以更快地收敛。最后，为了更深入地了解 goss，

让我们将这些差异放在表格中：

2.正则化

我将介绍 lightgbm 的一些重要的正则化参数。显然，这些是您需要调整以对抗过度拟合的参数。

应该意识到，对于小型数据集（<10000 条记录），lightGBM 可能不是最佳选择。调整 lightgbm 参数可能对您没有帮助。

另外，lightgbm使用leaf-wise树生长算法，而XGBoost使用深度wise树生长算法。 Leaf-wise方法允许树更快地收敛，但过度拟合的机会增加。

根据 lightGBM 文档，当面临过度拟合时，您可能需要进行以下参数调整：

使用较小的 max_bin
使用小的 num_leaves
使用 min_data_in_leaf 和 min_sum_hessian_in_leaf
通过设置 bagging_fraction 和 bagging_freq 使用 bagging
通过设置feature_fraction来使用特征子采样
使用更大的训练数据
尝试 lambda_l1、lambda_l2 和 min_gain_to_split 进行正则化
尝试 max_depth 以避免树长得很深

在下面的部分中，我将更详细地解释每个参数。

2.1 lambda_l1

Lambda_l1（和 lambda_l2）对 l1/l2 的控制以及 min_gain_to_split 用于对抗过度拟合。我强烈建议您使用参数调整（在后面的部分中探讨）来找出这些参数的最佳值。

2.2 num_leaves

当然，"num_leaves" 绝对是控制模型复杂度的最重要参数之一。通过这个参数，你设定了每个弱学习器的最大叶子节点数。较大的 "num_leaves" 可以提高在训练集上的准确性，但同时也增加了过拟合的风险。根据文档的描述，一种简单的设置方式是 num_leaves = 2^(max_depth)，然而，考虑到在LightGBM中，基于叶子节点的树比基于层级的树更深，因此你需要谨慎处理过拟合问题！因此，有必要将 "num_leaves" 与 "max_depth" 一起进行调整。

2.3 subsample

使用 subsample （或 bagging_fraction），您可以指定每次树构建迭代使用的行的百分比。这意味着将随机选择一些行来适合每个学习器（树）。这提高了泛化能力，也提高了训练速度。

我建议对基线模型使用较小的子样本值，然后在完成其他实验（不同的特征选择、不同的树结构）时增加该值。

2.4 feature_fraction

特征分数或 sub_feature 处理列采样，LightGBM 将在每次迭代（树）上随机选择特征子集。例如，如果将其设置为 0.6，LightGBM 将在训练每棵树之前选择 60% 的特征。

该功能有两种用途：

可用于加速训练
可以用来处理过拟合

2.5 max_depth

该参数控制每个训练树的最大深度，并将影响：

num_leaves 参数的最佳值
模型性能
训练时间

注意如果您使用较大的 max_depth 值，您的模型可能会过度拟合训练集。

2.6 max_bin

分箱是一种以离散视图（直方图）表示数据的技术。 Lightgbm 使用基于直方图的算法来找到最佳分割点，同时创建弱学习器。因此，每个连续的数字特征（例如视频的观看次数）应分为离散的容器。

max_bin 对 CPU 和 GPU 的影响如下图：

如果将 max_bin 定义为 255，则意味着每个特征最多可以有 255 个唯一值。那么较小的 max_bin 会导致更快的速度，而较大的值会提高准确性。

3.训练参数

当您想使用 lightgbm 训练模型时，训练 lightgbm 模型时可能会出现的一些典型问题是：

训练是一个耗时的过程
处理计算复杂性（CPU/GPU RAM 限制）
处理分类特征
数据集不平衡
自定义指标的需求
需要针对分类或回归问题进行的调整

3.1 num_iterations

Num_iterations 指定提升迭代（要构建的树）的数量。您构建的树越多，模型就越准确，但代价是：

训练时间更长
过拟合的可能性更高

建议使用较小的learning_rate和较大的num_iterations。另外，如果您选择更高的 num_iterations 来在没有学习到任何有用的东西时停止训练，则应该使用 Early_stopping_rounds .

3.2 early_stopping_rounds

如果验证指标在最后一轮提前停止后没有改善，则该参数将停止训练。这应该与多次迭代一起定义。如果将其设置得太大，则会增加过度拟合的机会（但您的模型可以更好）。

经验法则是让它占 num_iterations 的 10%。

3.3 lightgbm categorical_feature

使用lightgbm的优点之一是它可以很好地处理分类特征。是的，这个算法非常强大，但是你必须小心如何使用它的参数。 lightgbm 使用特殊的整数编码方法（由 Fisher 提出）来处理分类特征

实验表明，该方法比常用的 one-hot 编码具有更好的性能。

它的默认值是“auto”，这意味着：让 lightgbm 决定，这意味着 lightgbm 将推断哪些特征是分类的。

它并不总是工作得很好（一些实验说明了为什么在这里和这里），我强烈建议您使用此代码手动设置分类特征

 cat_col = dataset_name.select_dtypes(‘object’).columns.tolist()

但是幕后发生了什么以及 lightgbm 如何处理分类特征？

根据lightgbm的文档，我们知道树学习器不能很好地使用一种热门的编码方法，因为它们在树中生长得很深。在所提出的替代方法中，树学习器是最优构造的。例如，对于具有 k 个不同类别的一个特征，有 2^(k-1) – 1 个可能的分区，并且使用 Fisher 方法，可以通过在值的排序直方图上找到最佳分割方式来改进为 k * log(k) 在分类特征中。

3.4 lightgbm is_unbalance vs scale_pos_weight

在二元分类问题中您可能面临的问题之一是如何处理不平衡的数据集。显然，您需要平衡正/负样本，但是在 lightgbm 中到底如何做到这一点呢？

lightgbm中有两个参数可以让你处理这个问题is_unbalance和scale_pos_weight，但是它们之间有什么区别以及如何使用它们呢？

当您设置 Is_unbalace: True 时，算法将尝试自动平衡主导标签的权重（与训练集中的 pos/neg 分数）
如果您想在数据集不平衡的情况下更改scale_pos_weight（默认为1，这意味着假设正标签和负标签相等），您可以使用以下公式（基于lightgbm存储库上的此问题）来正确设置它

sample_pos_weight = number of negative samples / number of positive samples

3.5 lgbm feval

有时您想要定义自定义评估函数来衡量模型的性能，您需要创建 feval 函数。

Feval 函数应接受两个参数：

preds
train_data

并返回

eval_name
eval_result
is_higher_better

让我们逐步创建一个自定义指标函数。定义一个单独的Python函数

def feval_func(preds, train_data):
   # Define a formula that evaluates the results
    return ('feval_func_name', eval_result, False)

使用此函数作为参数：

print('Start training...')
lgb_train = lgb.train(...,
                      metric=None,
                      feval=feval_func)

3.6 classification 参数 vs regression 参数

我之前提到的大多数内容对于分类和回归都是正确的，但有些事情需要调整。

具体来说，您应该：

名称	分类注意事项	回归注意事项
objective	Set it binary or multiclass	Set it regression
metric	Binary_logloss or AUC or etc.	RMSE or mean_absolute_error and or etc.
is_unbalance	True or false	–
scale_pos_weight	used only in binary and multi class applications	–
num_class	used only in multi-class classification application	–
reg_sqrt	–	Used to fit sqrt(label) instead of original values for large range label

4. 最重要的lightgbm参数

我们在前面的部分中回顾并了解了一些有关 lightgbm 参数的知识，但是如果不提及 Laurae 令人难以置信的基准，任何增强树文章都是不完整的。

您可以了解 lightGBM 和 XGBoost 的许多问题的最佳默认参数。

您可以在这里查看，但一些最重要的要点是：

参数名称	默认值	范围	参数类型r	别名	约束或注释	用途
objective	regression	regression, binary	enum	objective_type, app	when you change it affects other parameters	specify the type of ML model
metric	null	+20 different metrics	multi-enum	metrics, metric_types	null means that metric corresponding to specified objective will be used	specify metric, support multiple metrics
boosting	gbdt	gbdt, rf, dart, goss	enum	boosting_type	if you set it RF, that would be a bagging approach	method of boosting
lambda_l1	0.0	[0, ∞]	double	reg_alpha	lambda_l1 >= 0.0	regularization
bagging_fraction	1.0	[0, 1]	double	subsample	0.0 < bagging_fraction <= 1.0	randomly select part of data without resampling
bagging_freq	0.0	[0, ∞]	int	subsample_freq	to enable bagging, bagging_fraction should be set to value smaller than 1.0 as well	0 means disable bagging; k means perform bagging at every k iteration
num_leaves	31	[1, ∞]	int	num_leaf	1 < num_leaves <= 131072	max number of leaves in one tree
feature_fraction	1.0	[0,1]	double	sub_feature	0.0 < feature_fraction <= 1.0	if you set it to 0.8, LightGBM will select 80% of features
max_depth	-1	[-1, ∞]	int	max_depth	larger is usually better, but overfitting speed increases	limit the max depth for tree model
max_bin	255	[2, ∞]	int	histogram binning	max_bin > 1	eal with over-fitting
num_iterations	100	[1, ∞]	int	num_boost_round, n_iter	num_iterations >= 0	number of boosting iterations
learning_rate	0.1	[0, 1]	double	eta	learning_rate > 0.0Typical: 0.05	in dart, it also affects on normalization weights of dropped trees
early_stopping _round	0	[0, ∞]	double	early_stopping _round	will stop training if validation doesn’t improve in last early_stopping _round	model performance, number of iterations, training time
categorical_feature	empty string	specify a number for a column index	multi-int or string	cat_feature	–	handle categorical features
bagging_freq	0.0	[0, ∞]	int	subsample_freq	0 means disable bagging; k means perform bagging at every k iteration	to enable bagging, bagging_fraction should be set to value smaller than 1.0 as well
verbosity	0	[-∞, ∞]	int	verbose	< 0: Fatal, = 0: Error (Warning), = 1: Info, > 1: Debug	useful for debging
min_data_in_leaf	20	min_data	int	min_data	min_data_in_leaf >= 0	can be used to deal with over-fitting

5.python中的Lightgbm参数调整示例

最后，在解释了所有重要参数之后，是时候进行一些实验了！

我将使用流行的 Kaggle 竞赛之一：桑坦德客户交易预测（Santander Customer Transaction Prediction. ）。

我将使用这篇文章解释如何在 Python 中对任何脚本运行超参数调整。
在我们开始之前，有一个重要问题！我们应该调整哪些参数？

        注意你想要解决的问题，例如桑坦德数据集是高度不平衡的，在你的调整中应该考虑到这一点！ Laurae2，lightgbm 的贡献者之一，在这里对此做了很好的解释。
        有些参数是相互依赖的，必须一起调整或一项一项调整。例如，min_data_in_leaf 取决于训练样本的数量和 num_leaves。
        注意：最好为超参数创建两个字典，一个包含您不想调整的参数和值，另一个包含您想要调整的参数和值范围。

SEARCH_PARAMS = {'learning_rate': 0.4,
                 'max_depth': 15,
                 'num_leaves': 20,
                 'feature_fraction': 0.8,
                 'subsample': 0.2}

FIXED_PARAMS={'objective': 'binary',
              'metric': 'auc',
              'is_unbalance':True,
              'boosting':'gbdt',
              'num_boost_round':300,
              'early_stopping_rounds':30}

5.1 结果分析

这里我一步一步解释如何调整超参数的值。创建基线训练代码：

from sklearn.metrics import roc_auc_score, roc_curve
from sklearn.model_selection import train_test_split
import neptunecontrib.monitoring.skopt as sk_utils
import lightgbm as lgb
import pandas as pd
import neptune
import skopt
import sys
import os

SEARCH_PARAMS = {'learning_rate': 0.4,
                'max_depth': 15,
                'num_leaves': 32,
                'feature_fraction': 0.8,
                'subsample': 0.2}

FIXED_PARAMS={'objective': 'binary',
             'metric': 'auc',
             'is_unbalance':True,
             'bagging_freq':5,
             'boosting':'dart',
             'num_boost_round':300,
             'early_stopping_rounds':30}

def train_evaluate(search_params):
   # you can download the dataset from this link(https://www.kaggle.com/c/santander-customer-transaction-prediction/data)
   # import Dataset to play with it
   data= pd.read_csv("sample_train.csv")
   X = data.drop(['ID_code', 'target'], axis=1)
   y = data['target']
   X_train, X_valid, y_train, y_valid = train_test_split(X, y, test_size=0.2, random_state=1234)
   train_data = lgb.Dataset(X_train, label=y_train)
   valid_data = lgb.Dataset(X_valid, label=y_valid, reference=train_data)

   params = {'metric':FIXED_PARAMS['metric'],
             'objective':FIXED_PARAMS['objective'],
             **search_params}

   model = lgb.train(params, train_data,
                     valid_sets=[valid_data],
                     num_boost_round=FIXED_PARAMS['num_boost_round'],
                     early_stopping_rounds=FIXED_PARAMS['early_stopping_rounds'],
                     valid_names=['valid'])
   score = model.best_score['valid']['auc']
   return score
Use the hyperparameter optimization library of your choice (for example scikit-optimize):

neptune.init('mjbahmani/LightGBM-hyperparameters')
neptune.create_experiment('lgb-tuning_final', upload_source_files=['*.*'],
                              tags=['lgb-tuning', 'dart'],params=SEARCH_PARAMS)

SPACE = [
   skopt.space.Real(0.01, 0.5, name='learning_rate', prior='log-uniform'),
   skopt.space.Integer(1, 30, name='max_depth'),
   skopt.space.Integer(10, 200, name='num_leaves'),
   skopt.space.Real(0.1, 1.0, name='feature_fraction', prior='uniform'),
   skopt.space.Real(0.1, 1.0, name='subsample', prior='uniform')
]
@skopt.utils.use_named_args(SPACE)
def objective(**params):
   return -1.0 * train_evaluate(params)

monitor = sk_utils.NeptuneMonitor()
results = skopt.forest_minimize(objective, SPACE,
                                n_calls=100, n_random_starts=10,
                                callback=[monitor])
sk_utils.log_results(results)

neptune.stop()

尝试不同类型的配置并在 Neptune 应用程序中跟踪您的结果。

最后，在下表中，您可以看到参数发生了哪些变化。

超参数	微调前	微调后
learning_rate	0.4	0.094
max_depth	15	10
num_leaves	32	12
feature_fraction	0.8	0.1
subsample	0.2	0.75
boosting	gbdt	dart
Score(auc)	0.8256	0.8605

总结

长话短说，本博客您了解到：

lightgbm的主要参数是什么
如何使用 feval 函数创建自定义指标
主要参数的默认值是多少
看到了如何调整 lightgbm 参数以提高模型性能的示例