Missing Value Handle
LightGBM enables the missing value handle by default. Disable it by setting
LightGBM uses NA (NaN) to represent missing values by default. Change it to use zero by setting
zero_as_missing=false(default), the unrecorded values in sparse matrices (and LightSVM) are treated as zeros.
zero_as_missing=true, NA and zeros (including unrecorded values in sparse matrices (and LightSVM)) are treated as missing.
Categorical Feature Support
LightGBM offers good accuracy with integer-encoded categorical features. LightGBM applies Fisher (1958) to find the optimal split over categories as described here. This often performs better than one-hot encoding.
categorical_featureto specify the categorical features. Refer to the parameter
Categorical features will be cast to
int32(integer codes will be extracted from pandas categoricals in the Python-package) so they must be encoded as non-negative integers (negative values will be treated as missing) less than
Int32.MaxValue(2147483647). It is best to use a contiguous range of integers started from zero. Floating point numbers in categorical features will be rounded towards 0.
cat_smoothto deal with over-fitting (when
#datais small or
For a categorical feature with high cardinality (
#categoryis large), it often works best to treat the feature as numeric, either by simply ignoring the categorical interpretation of the integers or by embedding the categories in a low-dimensional numeric space.
The label should be of type
int, such that larger numbers correspond to higher relevance (e.g. 0:bad, 1:fair, 2:good, 3:perfect).
label_gainto set the gain(weight) of
lambdarank_truncation_levelto truncate the max DCG.
Cost Efficient Gradient Boosting
Cost Efficient Gradient Boosting (CEGB) makes it possible to penalise boosting based on the cost of obtaining feature values. CEGB penalises learning in the following ways:
Each time a tree is split, a penalty of
When a feature is used for the first time,
cegb_penalty_feature_coupledis applied. This penalty can be different for each feature and should be specified as one
When a feature is used for the first time for a data row,
cegb_penalty_feature_lazyis applied. Like
cegb_penalty_feature_coupled, this penalty is specified as one
Each of the penalties above is scaled by
Using this parameter, it is possible to change the overall strength of the CEGB penalties by changing only one parameter.
Refer to Parameters Tuning.
Refer to Distributed Learning Guide.
Recommendations for gcc Users (MinGW, *nix)
Refer to gcc Tips.
Support for Position Bias Treatment
Often the relevance labels provided in Learning-to-Rank tasks might be derived from implicit user feedback (e.g., clicks) and therefore might be biased due to their position/location on the screen when having been presented to a user. LightGBM can make use of positional data.
For example, consider the case where you expect that the first 3 results from a search engine will be visible in users’ browsers without scrolling, and all other results for a query would require scrolling.
LightGBM could be told to account for the position bias from results being “above the fold” by providing a
positions array encoded as follows:
0 = "above the fold" and
1 = "requires scrolling".
The specific values are not important, as long as they are consistent across all observations in the training data.
An encoding like
100 = "above the fold" and
17 = "requires scrolling" would result in exactly the same trained model.
In that way,
positions in LightGBM’s API are similar to a categorical feature.
Just as with non-ordinal categorical features, an integer representation is just used for memory and computational efficiency… LightGBM does not care about the absolute or relative magnitude of the values.
Unlike a categorical feature, however,
positions are used to adjust the target to reduce the bias in predictions made by the trained model.
The position file corresponds with training data file line by line, and has one position per line. And if the name of training data file is
train.txt, the position file should be named as
train.txt.position and placed in the same folder as the data file.
In this case, LightGBM will load the position file automatically if it exists. The positions can also be specified through the
Dataset constructor when using Python API. If the positions are specified in both approaches, the
.position file will be ignored.
Currently, implemented is an approach to model position bias by using an idea of Generalized Additive Models (GAM) to linearly decompose the document score
s into the sum of a relevance component
f and a positional component
s(x, pos) = f(x) + g(pos) where the former component depends on the original query-document features and the latter depends on the position of an item.
During the training, the compound scoring function
s(x, pos) is fit with a standard ranking algorithm (e.g., LambdaMART) which boils down to jointly learning the relevance component
f(x) (it is later returned as an unbiased model) and the position factors
g(pos) that help better explain the observed (biased) labels.
Similar score decomposition ideas have previously been applied for classification & pointwise ranking tasks with assumptions of binary labels and binary relevance (a.k.a. “two-tower” models, refer to the papers: Towards Disentangling Relevance and Bias in Unbiased Learning to Rank, PAL: a position-bias aware learning framework for CTR prediction in live recommender systems, A General Framework for Debiasing in CTR Prediction).
In LightGBM, we adapt this idea to general pairwise Lerarning-to-Rank with arbitrary ordinal relevance labels.
Besides, GAMs have been used in the context of explainable ML (Accurate Intelligible Models with Pairwise Interactions) to linearly decompose the contribution of each feature (and possibly their pairwise interactions) to the overall score, for subsequent analysis and interpretation of their effects in the trained models.