点击上方“Datawhale”,选择“星标”公众号
第一时间获取价值内容
CatBoost
CatBoost是俄罗斯的搜索巨头Y andex在2017年开源的机器学习库,也是Boosting族算法的一种,同前面介绍过的XGBoost和LightGBM类似,依然是在GBDT算法框架下的一种改进实现,是一种基于对称决策树(oblivious trees)算法的参数少、支持类别型变量和高准确性的GBDT框架,主要说解决的痛点是高效合理地处理类别型特征,这个从它的名字就可以看得出来,CatBoost是由catgorical和boost组成,另外是处理梯度偏差(Gradient bias)以及预测偏移(Prediction shift)问题,提高算法的准确性和泛化能力。 集成学习 CatBoost主要有以下五个特性:Categorical features
所谓类别型变量(Categorical features)是指其值是离散的集合且相互比较并无意义的变量,比如用户的ID、产品ID、颜色等。 因此,这些变量无法在二叉决策树当中直接使用。 常规的做法是将这些类别变量通过预处理的方式转化成数值型变量再喂给模型,比如用一个或者若干个数值来代表一个类别型特征。 目前广泛用于 低势(一个有限集的元素个数是一个自然数)类别特征的处理方法是One-hot encoding
: 将原来的特征删除,然后对于每一个类别加一个0/1的用来指示是否含有该类别的数值型特征。
One-hot encoding
可以在数据预处理时完成,也可以在模型训练的时候完成,从训练时间的角度,后一种方法的实现更为高效,CatBoost对于低势类别特征也是采用后一种实现。 显然,在
高势特征当中,比如
user ID
,这种编码方式会产生大量新的特征,造成维度灾难。 一种折中的办法是可以将类别分组成有限个的群体再进行
One-hot encoding
。 一种常被使用的方法是根据目标变量统计(Target Statistics,以下简称TS)进行分组,目标变量统计用于估算每个类别的目标变量期望值。 甚至有人直接用TS作为一个新的数值型变量来代替原来的类别型变量。 重要的是,可以通过对TS数值型特征的阈值设置,基于对数损失、基尼系数或者均方差,得到一个对于训练集而言将类别一分为二的所有可能划分当中最优的那个。 在LightGBM当中,类别型特征用每一步梯度提升时的梯度统计(Gradient Statistics,以下简称GS)来表示。 虽然为建树提供了重要的信息,但是这种方法有以下两个缺点:
One-hot encoding
还是好不少。 不过如果采用TS特征,那么对于每个类别只需要计算和存储一个数字。 如此看到,采用TS作为一个新的数值型特征是最有效、信息损失最小的处理类别型特征的方法。 TS也被广泛采用,在点击预测任务当中,这个场景当中的类别特征有用户、地区、广告、广告发布者等。 接下来我们着重讨论TS,暂时将
One-hot encoding
和GS放一边。
Target statistics 一个有效和高效的处理类别型特征 的方式是用一个与某些TS相等的数值型变量 来代替第 个训练样本的类别 。 通常用基于类别的目标变量 的期望来进行估算: 。
Greedy TS 估算 最直接的方式就是用训练样本当中相同类别 的目标变量 的平均值。 显然,这样的处理方式很容易引起过拟合。 举个例子,假如在整个训练集当中所有样本的类别 都互不相同,即 个样本有 个类别,那么新产生的数值型特征的值将与目标变量的值相同。 某种程度上,这是一种目标穿越(target leakage),非常容易引起过拟合。 比较好的一种做法是采用一个先验概率 进行平滑处理: 其中 是先验概率 的权重,而对于先验概率,通常的做法是设置为数据集当中目标变量的平均值。 不过这样的平滑处理依然无法完全避免目标穿越: 特征 是通过自变量 的目标 计算所得。 这将会导致条件偏移: 对于训练集和测试集, 的分布会有所不同。 再举个例子,假设第 个特征为类别型特征,并且特征所有取值为无重复的集合,然后对于每一个类别 ,对于一个分类任务,我们有 。 然后在训练集当中, ,于是用阈值 就可以仅用一次分裂就训练集完美分开。 但是,对于测试集,因为还无法判断此时目标变量的类别,所以这一项 ,最后得到的TS值为 ,并且得到的模型在$p 其中, 是第 个训练样本。 在我们的例子当中, , ,显然无法满足上述条件。
Holdout TS 留出TS,就是将训练集一分为二: ,然后根据下式用 来计算TS,并将 作为训练样本。 这样处理能够满足同分布的问题,但是却大大减少了训练样本的数量。
Leave-one-out TS 初看起来,留一TS(Leave-one-out TS)能够非常好地工作:
Gradient bias
CatBoost,和所有标准梯度提升算法一样,都是通过构建新树来拟合当前模型的梯度。 然而,所有经典的提升算法都存在由有偏的点态梯度估计引起的过拟合问题。 在每个步骤中使用的梯度都使用当前模型中的相同的数据点来估计,这导致估计梯度在特征空间的任何域中的分布与该域中梯度的真实分布相比发生了偏移,从而导致过拟合。 为了解决这个问题,CatBoost对经典的梯度提升算法进行了一些改进,简要介绍如下: 在许多利用GBDT框架的算法(例如,XGBoost、LightGBM)中,构建下一棵树分为两个阶段: 选择树结构和在树结构固定后计算叶子节点的值。 为了选择最佳的树结构,算法通过枚举不同的分割,用这些分割构建树,对得到的叶子节点中计算值,然后对得到的树计算评分,最后选择最佳的分割。 两个阶段叶子节点的值都是被当做梯度或牛顿步长的近似值来计算。 在CatBoost中,第二阶段使用传统的GBDT框架执行,第一阶段使用修改后的版本。 既然原来的梯度估计是有偏的,那么能不能改成无偏估计呢? 设 为构建 棵树后的模型, 为构建 棵树后第 个训练样本上面的梯度值。 为了使得 无偏于模型 ,我们需要在没有 参与的情况下对模型 进行训练。 由于我们需要对所有训练样本计算无偏的梯度估计,乍看起来对于 的训练不能使用任何样本,貌似无法实现的样子。 我们运用下面这个技巧来处理这个问题: 对于每一个样本 ,我们训练一个单独的模型 ,且该模型从不使用基于该样本的梯度估计进行更新。 我们使用 来估计 上的梯度,并使用这个估计对结果树进行评分。 用伪码描述如下,其中 是需要优化的损失函数, 是标签值, 是公式计算值。 Gradient bias 值得注意的是 模型的建立并没有样本 的参与,并且CatBoost中所有的树 的共享同样的结构。 在CatBoost中,我们生成训练数据集的 个随机排列。 采用多个随机排列是为了增强算法的鲁棒性,这在前面的Odered TS当中对于类别型特征的处理有介绍到: 针对每一个随机排列,计算得到其梯度,为了与Ordered TS保持一致,这里的排列与用于计算Ordered TS时的排列相同。 我们使用不同的排列来训练不同的模型,因此不会导致过拟合。 对于每个排列 ,我们训练 个不同的模型 ,如上所示。 这意味着为了构建一棵树,需要对每个排列存储并重新计算,其时间复杂度近似于 : 对于每个模型 ,我们必须更新 。 因此,时间复杂度变成 。 当然,在具体实现当中,CatBoost使用了其它的技巧,可以将构建一个树的时间复杂度降低到 。Prediction shift
预测偏移(Prediction shift)是由上一节所讨论的梯度偏差造成的。 本节希望用数学语言严格地对预测偏差进行描述和分析。 首先来看下梯度提升的整体迭代过程:Ordered boosting
为了克服上一节所描述的预测偏移问题,我们提出了一种新的叫做Ordered boosting的算法。 假设用 棵树来学习一个模型,为了确保 无偏,需要确保模型 的训练没有用到样本 。 由于我们需要对所有训练样本计算无偏的梯度估计,乍看起来对于 的训练不能使用任何样本,貌似无法实现的样子,但是事实上可以通过一些技巧来进行克服,具体的算法在前面已经有所描述,而且是作者较新的论文当中的描述,这里不再赘述。 本节主要讲讲Ordered boosting的具体实现。 Ordered boosting算法好是好,但是在大部分的实际任务当中都不具备使用价值,因为需要训练 个不同的模型,大大增加的内存消耗和时间复杂度。 在CatBoost当中,我们实现了一个基于GBDT框架的修改版本。 前面提到过,在传统的GBDT框架当中,构建下一棵树分为两个阶段: 选择树结构和在树结构固定后计算叶子节点的值。 CatBoost主要在第一阶段进行优化。 First phase 在建树的阶段,CatBoost有两种提升模式,Ordered和Plain。 Plain模式是采用内建的ordered TS对类别型特征进行转化后的标准GBDT算法。 Ordered则是对Ordered boosting算法的优化。 Ordered boosting mode 一开始,CatBoost对训练集产生 个独立的随机序列。 序列 用来评估定义树结构的分裂, 用来计算所得到的树的叶子节点的值。 因为,在一个给定的序列当中,对于较短的序列,无论是TS的计算还是基于Ordered boosting的预测都会有较大的方差,所以仅仅用一个序列可能引起最终模型的方差,这里我们会有多个序列进行学习。 CatBoost采用对称树作为基学习器,对称意味着在树的同一层,其分裂标准都是相同的。 对称树具有平衡、不易过拟合并能够大大减少测试时间的特点。 建树的具体算法如下伪码描述。 Building a tree in CatBoost 在Ordered boosting模式的学习过程当中,我们维持一个模型 ,其中 表示基于在序列 当中的前 个样本学习得到的模型对于第 个样本的预测。 在算法的每一次迭代 ,我们从 当中抽样一个随机序列 ,并基于此构建第 步的学习树 。 然后,基于 ,计算相应的梯度 。 接下来,我们会用余弦相似度来近似梯度 ,其中对于每一个样本 ,我们取梯度 。 在候选分裂评估过程当中,第 个样本的叶子节点的值 由与 同属一个叶子的 的所有样本的前 个样本的梯度值 求平均得到。 需要注意的是, 取决于选定的序列 ,因为 会影响第 个样本的Ordered TS。 当树 的结构确定以后,我们用它来提升所有的模型 ,我们需要强调下,一个相同的树结构 会被用于所有的模型,但是会根据 和 的不同设置不同的叶子节点的值以后应用于不同的模型。 Plain boosting mode Plain boosting模式的算法与标准GBDT流程类似,但是如果出现了类别型特征,它会基于 得到的TS维持 个支持模型 。 Second phase 当所有的树结构确定以后,最终模型的叶子节点值的计算与标准梯度提升过程类似。 第 个样本与叶子 进行匹配,我们用 来计算这里的TS。 当最终模型 在测试期间应用于新的样本,我们采用整个训练集来计算TS。GPU加速
就GPU内存使用而言,CatBoost至少与LightGBM一样有效,CatBoost的GPU实现可支持多个GPU,分布式树学习可以通过样本或特征进行并行化。sklearn参数
sklearn
本身的文档当中并没有CatBoost的描述, CatBoost python-reference_parameters-list 上面看到主要参数如下:
iterations
: 迭代次数, 解决机器学习问题能够构建的最大树的数目,default=1000 learning_rate
: 学习率,default=0.03 depth
: 树的深度,default=6 l2_leaf_reg
: 正则化数,default=3.0 model_size_reg
:模型大小正则化系数,数值越到,模型越小,仅在有类别型变量的时候起作用,取值范围从0到 ,GPU计算时不可用, default=None rsm
: =None, loss_function
: 损失函数,字符串 (分类任务,default= Logloss
,回归任务,default= RMSE
) border_count
: 数值型变量的分箱个数 feature_border_type
: 数值型变量分箱个数的初始量化模式,default=GreedyLogSum per_float_feature_quantization
: 指定特定特征的分箱个数,default=None, input_borders
=None, output_borders
=None, fold_permutation_block
: 对数据集进行随机排列之前分组的block大小,default=1 od_pval
: 过拟合检测阈值,数值越大,越早检测到过拟合,default=0 od_wait
: 达成优化目标以后继续迭代的次数,default=20 od_type
: 过拟合检测类型,default=IncToDec nan_mode
: 缺失值的预处理方法,字符串类型,default=Min Forbidden
: 不支持缺失值 Min
: 缺失值赋值为最小值 Max
: 缺失值赋值为最大值 counter_calc_method
: 计算Counter CTR类型的方法,default=None leaf_estimation_iterations
: 计算叶子节点值时候的迭代次数,default=None, leaf_estimation_method
: 计算叶子节点值的方法,default=Gradient thread_count
: 训练期间的进程数,default=-1,进程数与部件的核心数相同 random_seed
: 随机数种子,default=0 use_best_model
: 如果有设置 eval_set
设置了验证集的话可以设为True,否则为False verbose
: 是否显示详细信息,default=1 logging_level
: 打印的日志级别,default=None metric_period
: 计算优化评估值的频率,default=1 ctr_leaf_count_limit
: 类别型特征最大叶子数,default=None store_all_simple_ctr
: 是否忽略类别型特征,default=False max_ctr_complexity
: 最大特征组合数,default=4 has_time
: 是否采用输入数据的顺序,default=False allow_const_label
: 使用它为所有对象用具有相同标签值的数据集训练模型,default=None classes_count
: 多分类当中类别数目上限,defalut=None class_weights
: 类别权重,default=None one_hot_max_size
: one-hot编码最大规模,默认值根据数据和训练环境的不同而不同 random_strength
: 树结构确定以后为分裂点进行打分的时候的随机强度,default=1 name
: 在可视化工具当中需要显示的实验名字 ignored_features
: 在训练当中需要排除的特征名称或者索引,default=None train_dir
: 训练过程当中文件保存的目录 custom_loss
: 用户自定义的损失函数 custom_metric
: 自定义训练过程当中输出的评估指标,default=None eval_metric
: 过拟合检测或者最优模型选择的评估指标 bagging_temperature
: 贝叶斯bootstrap强度设置,default=1 save_snapshot
: 训练中断情况下保存快照文件 snapshot_file
: 训练过程信息保存的文件名字 snapshot_interval
: 快照保存间隔时间,单位秒 fold_len_multiplier
: 改变fold长度的系数,default=2 used_ram_limit
: 类别型特征使用内存限制,default=None gpu_ram_part
: GPU内存使用率,default=0.95 allow_writing_files
: 训练过程当中允许写入分析和快照文件,default=True final_ctr_computation_mode
: Final CTR计算模式 approx_on_full_history
: 计算近似值的原则,default=False boosting_type
: 提升模式 simple_ctr
: 单一类别型特征的量化设置 combinations_ctr
: 组合类别型特征的量化设置 per_feature_ctr
: 以上几个参数的设置具体可以细看下面的文档 task_type
: 任务类型,CPU或者GPU,default=CPU device_config
: =None devices
: 用来训练的GPU设备号,default=NULL bootstrap_type
: 自采样类型,default=Bayesian subsample
: bagging的采样率,default=0.66 sampling_unit
: 采样模式,default=Object dev_score_calc_obj_block_size
: =None, max_depth
: 树的最大深度 n_estimators
: 迭代次数 num_boost_round
: 迭代轮数 num_trees
: 树的数目 colsample_bylevel
: 按层抽样比例,default=None random_state
: 随机数状态 reg_lambda
: 损失函数 范数,default=3.0 objective
: =同损失函数 eta
: 学习率 max_bin
: =同 border_coucnt
scale_pos_weight
: 二分类任务当中1类的权重,default=1.0 gpu_cat_features_storage
: GPU训练时类别型特征的存储方式,default=GpuRam data_partition
: 分布式训练时数据划分方法 metadata
: =None early_stopping_rounds
: 早停轮次,default=False cat_features
: =指定类别型特征的名称或者索引 grow_policy
: 树的生长策略 min_data_in_leaf
: 叶子节点最小样本数,default=1 min_child_samples
: 叶子节点最小样本数,default=1 max_leaves
: 最大叶子数,default=31 num_leaves
: 叶子数 score_function
: 建树过程当中的打分函数 leaf_estimation_backtracking
: 梯度下降时回溯类型 ctr_history_unit
: =None monotone_constraints
: =None
class CatBoostClassifier(iterations=None,
learning_rate=None,
depth=None,
l2_leaf_reg=None,
model_size_reg=None,
rsm=None,
loss_function=None,
border_count=None,
feature_border_type=None,
per_float_feature_quantization=None,
input_borders=None,
output_borders=None,
fold_permutation_block=None,
od_pval=None,
od_wait=None,
od_type=None,
nan_mode=None,
counter_calc_method=None,
leaf_estimation_iterations=None,
leaf_estimation_method=None,
thread_count=None,
random_seed=None,
use_best_model=None,
verbose=None,
logging_level=None,
metric_period=None,
ctr_leaf_count_limit=None,
store_all_simple_ctr=None,
max_ctr_complexity=None,
has_time=None,
allow_const_label=None,
classes_count=None,
class_weights=None,
one_hot_max_size=None,
random_strength=None,
name=None,
ignored_features=None,
train_dir=None,
custom_loss=None,
custom_metric=None,
eval_metric=None,
bagging_temperature=None,
save_snapshot=None,
snapshot_file=None,
snapshot_interval=None,
fold_len_multiplier=None,
used_ram_limit=None,
gpu_ram_part=None,
allow_writing_files=None,
final_ctr_computation_mode=None,
approx_on_full_history=None,
boosting_type=None,
simple_ctr=None,
combinations_ctr=None,
per_feature_ctr=None,
task_type=None,
device_config=None,
devices=None,
bootstrap_type=None,
subsample=None,
sampling_unit=None,
dev_score_calc_obj_block_size=None,
max_depth=None,
n_estimators=None,
num_boost_round=None,
num_trees=None,
colsample_bylevel=None,
random_state=None,
reg_lambda=None,
objective=None,
eta=None,
max_bin=None,
scale_pos_weight=None,
gpu_cat_features_storage=None,
data_partition=None
metadata=None,
early_stopping_rounds=None,
cat_features=None,
grow_policy=None,
min_data_in_leaf=None,
min_child_samples=None,
max_leaves=None,
num_leaves=None,
score_function=None,
leaf_estimation_backtracking=None,
ctr_history_unit=None,
monotone_constraints=None)
br
CatBoostRegressor CatBoostRegressor
class CatBoostRegressor(iterations=None,
learning_rate=None,
depth=None,
l2_leaf_reg=None,
model_size_reg=None,
rsm=None,
loss_function='RMSE',
border_count=None,
feature_border_type=None,
per_float_feature_quantization=None,
input_borders=None,
output_borders=None,
fold_permutation_block=None,
od_pval=None,
od_wait=None,
od_type=None,
nan_mode=None,
counter_calc_method=None,
leaf_estimation_iterations=None,
leaf_estimation_method=None,
thread_count=None,
random_seed=None,
use_best_model=None,
best_model_min_trees=None,
verbose=None,
silent=None,
logging_level=None,
metric_period=None,
ctr_leaf_count_limit=None,
store_all_simple_ctr=None,
max_ctr_complexity=None,
has_time=None,
allow_const_label=None,
one_hot_max_size=None,
random_strength=None,
name=None,
ignored_features=None,
train_dir=None,
custom_metric=None,
eval_metric=None,
bagging_temperature=None,
save_snapshot=None,
snapshot_file=None,
snapshot_interval=None,
fold_len_multiplier=None,
used_ram_limit=None,
gpu_ram_part=None,
pinned_memory_size=None,
allow_writing_files=None,
final_ctr_computation_mode=None,
approx_on_full_history=None,
boosting_type=None,
simple_ctr=None,
combinations_ctr=None,
per_feature_ctr=None,
ctr_target_border_count=None,
task_type=None,
device_config=None,
devices=None,
bootstrap_type=None,
subsample=None,
sampling_unit=None,
dev_score_calc_obj_block_size=None,
max_depth=None,
n_estimators=None,
num_boost_round=None,
num_trees=None,
colsample_bylevel=None,
random_state=None,
reg_lambda=None,
objective=None,
eta=None,
max_bin=None,
gpu_cat_features_storage=None,
data_partition=None,
metadata=None,
early_stopping_rounds=None,
cat_features=None,
grow_policy=None,
min_data_in_leaf=None,
min_child_samples=None,
max_leaves=None,
num_leaves=None,
score_function=None,
leaf_estimation_backtracking=None,
ctr_history_unit=None,
monotone_constraints=None)
br
应用场景
作为GBDT框架内的算法,GBDT、XGBoost、LightGBM能够应用的场景CatBoost也都适用,并且在处理类别型特征具备独有的优势,比如广告推荐领域。优缺点
优点参考