Understand your dataset with XGBoost
The purpose of this vignette is to show you how to use Xgboost to discover and understand your own dataset better.
This vignette is not about predicting anything (see Xgboost presentation). We will explain how to use Xgboost to highlight the link between the features of your data and the outcome.
require(xgboost) require(Matrix) require(data.table) if (!require('vcd')) install.packages('vcd')
VCD package is used for one of its embedded dataset only.
Preparation of the dataset
Numeric v.s. categorical variables
Xgboost manages only
What to do when you have categorical data?
A categorical variable has a fixed number of different values. For instance, if a variable called Colour can have only one of these three values, red, blue or green, then Colour is a categorical variable.
In R, a categorical variable is called
?factorin the console for more information.
To answer the question above we will convert categorical variables to
Conversion from categorical to numeric variables
Looking at the raw data
In this Vignette we will see how to transform a dense
data.frame (dense = few zeroes in the matrix) with categorical variables to a very sparse matrix (sparse = lots of zero in the matrix) of
The method we are going to see is usually called one-hot encoding.
The first step is to load
Arthritis dataset in memory and wrap it with
data(Arthritis) df <- data.table(Arthritis, keep.rownames = F)
data.tableis 100% compliant with R
data.framebut its syntax is more consistent and its performance for large dataset is best in class (
dplyrfrom R and
Pandasfrom Python included). Some parts of Xgboost R package use
The first thing we want to do is to have a look to the first few lines of the
Now we will check the format of each column.
2 columns have
factor type, one has
- can take a limited number of values (like
- these values are ordered (unlike
factor). Here these ordered values are:
Marked > Some > None
Creation of new features based on old ones
We will add some new categorical features to see if it helps.
Grouping per 10 years
For the first feature we create groups of age by rounding the real age.
Note that we transform it to
factor so the algorithm treat these age groups as independent values.
Therefore, 20 is not closer to 30 than 60. To make it short, the distance between ages is lost in this transformation.
head(df[,AgeDiscret := as.factor(round(Age/10,0))])
Random split into two groups
Following is an even stronger simplification of the real age with an arbitrary split at 30 years old. We choose this value based on nothing. We will see later if simplifying the information based on arbitrary values is a good strategy (you may already have an idea of how well it will work...).
head(df[,AgeCat:= as.factor(ifelse(Age > 30, "Old", "Young"))])
Risks in adding correlated features
These new features are highly correlated to the
Age feature because they are simple transformations of this feature.
For many machine learning algorithms, using correlated features is not a good idea. It may sometimes make prediction less accurate, and most of the time make interpretation of the model almost impossible. GLM, for instance, assumes that the features are uncorrelated.
Fortunately, decision tree algorithms (including boosted trees) are very robust to these features. Therefore we have nothing to do to manage this situation.
We remove ID as there is nothing to learn from this feature (it would just add some noise).
We will list the different values for the column
Next step, we will transform the categorical data to dummy variables. This is the one-hot encoding step.
The purpose is to transform each value of each categorical feature into a binary feature
For example, the column
Treatment will be replaced by two columns,
Treated. Each of them will be binary. Therefore, an observation which has the value
Placebo in column
Treatment before the transformation will have after the transformation the value
1 in the new column
Placebo and the value
0 in the new column
Treated. The column
Treatment will disappear during the one-hot encoding.
Improved is excluded because it will be our
label column, the one we want to predict.
sparse_matrix <- sparse.model.matrix(Improved~.-1, data = df) head(sparse_matrix)
Improved~.-1used above means transform all categorical features but column
Improvedto binary values. The
-1is here to remove the first column which is full of
1(this column is generated by the conversion). For more information, you can type
?sparse.model.matrixin the console.
Create the output
numeric vector (not as a sparse
output_vector = df[,Improved] == "Marked"
1for rows where
Improved == Markedis
Build the model
The code below is very usual. For more information, you can look at the documentation of
xgboost function (or at the vignette Xgboost presentation).
bst <- xgboost(data = sparse_matrix, label = output_vector, max_depth = 4, eta = 1, nthread = 2, nrounds = 10,objective = "binary:logistic")
You can see some
train-error: 0.XXXXX lines followed by a number. It decreases. Each line shows how well the model explains your data. Lower is better.
A model which fits too well may overfit (meaning it copy/paste too much the past, and won't be that good to predict the future).
Here you can see the numbers decrease until line 7 and then increase.
It probably means we are overfitting. To fix that I should reduce the number of rounds to
nrounds = 4. I will let things like that because I don't really care for the purpose of this example :-)
Measure feature importance
Build the feature importance data.table
Remember, each binary column corresponds to a single value of one of categorical features.
importance <- xgb.importance(feature_names = colnames(sparse_matrix), model = bst) head(importance)
Gainprovide the information we are looking for.
As you can see, features are classified by
Gain is the improvement in accuracy brought by a feature to the branches it is on. The idea is that before adding a new split on a feature X to the branch there was some wrongly classified elements, after adding the split on this feature, there are two new branches, and each of these branch is more accurate (one branch saying if your observation is on this branch then it should be classified as
1, and the other branch saying the exact opposite).
Cover measures the relative quantity of observations concerned by a feature.
Frequency is a simpler way to measure the
Gain. It just counts the number of times a feature is used in all generated trees. You should not use it (unless you know why you want to use it).
Improvement in the interpretability of feature importance data.table
We can go deeper in the analysis of the model. In the
data.table above, we have discovered which features counts to predict if the illness will go or not. But we don't yet know the role of these features. For instance, one of the question we may want to answer would be: does receiving a placebo treatment helps to recover from the illness?
One simple solution is to count the co-occurrences of a feature and a class of the classification.
For that purpose we will execute the same function as above but using two more parameters,
importanceRaw <- xgb.importance(feature_names = colnames(sparse_matrix), model = bst, data = sparse_matrix, label = output_vector) # Cleaning for better display importanceClean <- importanceRaw[,`:=`(Cover=NULL, Frequency=NULL)] head(importanceClean)
In the table above we have removed two not needed columns and select only the first lines.
First thing you notice is the new column
Split. It is the split applied to the feature on a branch of one of the tree. Each split is present, therefore a feature can appear several times in this table. Here we can see the feature
Age is used several times with different splits.
How the split is applied to count the co-occurrences? It is always
<. For instance, in the second line, we measure the number of persons under 61.5 years with the illness gone after the treatment.
The two other new columns are
RealCover %. In the first column it measures the number of observations in the dataset where the split is respected and the label marked as
1. The second column is the percentage of the whole population that
Therefore, according to our findings, getting a placebo doesn't seem to help but being younger than 61 years may help (seems logic).
You may wonder how to interpret the
< 1.00001on the first line. Basically, in a sparse
Matrix, there is no
0, therefore, looking for one hot-encoded categorical observations validating the rule
< 1.00001is like just looking for
1for this feature.
Plotting the feature importance
All these things are nice, but it would be even better to plot the results.
xgb.plot.importance(importance_matrix = importance)
Feature have automatically been divided in 2 clusters: the interesting features... and the others.
Depending of the dataset and the learning parameters you may have more than two clusters. Default value is to limit them to
10, but you can increase this limit. Look at the function documentation for more information.
According to the plot above, the most important features in this dataset to predict if the treatment will work are :
- the Age ;
- having received a placebo or not ;
- the sex is third but already included in the not interesting features group ;
- then we see our generated features (AgeDiscret). We can see that their contribution is very low.
Do these results make sense?
Let's check some Chi2 between each of these features and the label.
Higher Chi2 means better correlation.
c2 <- chisq.test(df$Age, output_vector) print(c2)
Pearson correlation between Age and illness disapearing is
r round(c2$statistic, 2 ).
c2 <- chisq.test(df$AgeDiscret, output_vector) print(c2)
Our first simplification of Age gives a Pearson correlation is
r round(c2$statistic, 2).
c2 <- chisq.test(df$AgeCat, output_vector) print(c2)
The perfectly random split I did between young and old at 30 years old have a low correlation of
r round(c2$statistic, 2). It's a result we may expect as may be in my mind > 30 years is being old (I am 32 and starting feeling old, this may explain that), but for the illness we are studying, the age to be vulnerable is not the same.
Morality: don't let your gut lower the quality of your model.
In data science expression, there is the word science :-)
As you can see, in general destroying information by simplifying it won't improve your model. Chi2 just demonstrates that.
But in more complex cases, creating a new feature based on existing one which makes link with the outcome more obvious may help the algorithm and improve the model.
The case studied here is not enough complex to show that. Check Kaggle website for some challenging datasets. However it's almost always worse when you add some arbitrary rules.
Moreover, you can notice that even if we have added some not useful new features highly correlated with other features, the boosting tree algorithm have been able to choose the best one, which in this case is the Age.
Linear model may not be that smart in this scenario.
Special Note: What about Random Forests™?
Both trains several decision trees for one dataset. The main difference is that in Random Forests™, trees are independent and in boosting, the tree
N+1 focus its learning on the loss (<=> what has not been well modeled by the tree
This difference have an impact on a corner case in feature importance analysis: the correlated features.
Imagine two features perfectly correlated, feature
A and feature
B. For one specific tree, if the algorithm needs one of them, it will choose randomly (true in both boosting and Random Forests™).
However, in Random Forests™ this random choice will be done for each tree, because each tree is independent from the others. Therefore, approximatively, depending of your parameters, 50% of the trees will choose feature
A and the other 50% will choose feature
B. So the importance of the information contained in
B (which is the same, because they are perfectly correlated) is diluted in
B. So you won't easily know this information is important to predict what you want to predict! It is even worse when you have 10 correlated features...
In boosting, when a specific link between feature and outcome have been learned by the algorithm, it will try to not refocus on it (in theory it is what happens, reality is not always that simple). Therefore, all the importance will be on feature
A or on feature
B (but not both). You will know that one feature have an important role in the link between the observations and the label. It is still up to you to search for the correlated features to the one detected as important if you need to know all of them.
If you want to try Random Forests™ algorithm, you can tweak Xgboost parameters!
For instance, to compute a model with 1000 trees, with a 0.5 factor on sampling rows and columns:
data(agaricus.train, package='xgboost') data(agaricus.test, package='xgboost') train <- agaricus.train test <- agaricus.test #Random Forest™ - 1000 trees bst <- xgboost(data = train$data, label = train$label, max_depth = 4, num_parallel_tree = 1000, subsample = 0.5, colsample_bytree =0.5, nrounds = 1, objective = "binary:logistic") #Boosting - 3 rounds bst <- xgboost(data = train$data, label = train$label, max_depth = 4, nrounds = 3, objective = "binary:logistic")
Note that the parameter
roundis set to
Random Forests™ is a trademark of Leo Breiman and Adele Cutler and is licensed exclusively to Salford Systems for the commercial release of the software.