Photo by Vilmos Heim on Unsplash6 Different Ways to Compensate for Missing Values In a Dataset (Data Imputation with examples)Popular strategies to statistically impute missing values in a dataset.
Will badrBlockedUnblockFollowFollowingJan 4Many real-world datasets may contain missing values for various reasons.
They are often encoded as NaNs, blanks or any other placeholders.
Training a model with a dataset that has a lot of missing values can drastically impact the machine learning model’s quality.
Some algorithms such as scikit-learn estimators assume that all values are numerical and have and hold meaningful value.
One way to handle this problem is to get rid of the observations that have missing data.
However, you will risk losing data points with valuable information.
A better strategy would be to impute the missing values.
In other words, we need to infer those missing values from the existing part of the data.
There are three main types of missing data:Missing completely at random (MCAR)Missing at random (MAR)Not missing at random (NMAR)However, in this article, I will focus on 6 popular ways for data imputation for cross-sectional datasets ( Time-series dataset is a different story ).
1- Do Nothing:That’s an easy one.
You just let the algorithm handle the missing data.
Some algorithms can factor in the missing values and learn the best imputation values for the missing data based on the training loss reduction (ie.
XGBoost).
Some others can just ignore them (ie.
LightGBM, Catboost).
However, other algorithms will just panic and throw an error complaining about the missing values (ie.
Scikit learn — LinearRegression).
In that case, you will need to handle the missing data and clean it before feeding it to the algorithm.
Let’s see some other ways to impute the missing values before training:Note: All the examples below use the California Housing Dataset from Scikit-learn.
2- Imputation Using (Mean/Median) Values:This works by calculating the mean/median of the non-missing values in a column and then replacing the missing values within each column separately and independently from the others.
It can only be used with numeric data.
Mean ImputationPros:Easy and fast.
Works well with small numerical datasets.
Cons:Doesn’t factor the correlations between features.
It only works on the column level.
Will give poor results on encoded categorical features (do NOT use it on categorical features).
Not very accurate.
Doesn’t account for the uncertainty in the imputations.
Mean/Median Imputation3- Imputation Using (Most Frequent) or (Zero/Constant) Values:Most Frequent is another statistical strategy to impute missing values and YES!!.It works with categorical features (strings or numerical representations) by replacing missing data with the most frequent values within each column.
Pros:Works well with categorical features.
Cons:It also doesn’t factor the correlations between features.
It can introduce bias in the data.
Most Frequent ImputationZero or Constant imputation — as the name suggests — it replaces the missing values with either zero or any constant value you specify4- Imputation Using k-NN:The k nearest neighbours is an algorithm that is used for simple classification.
The algorithm uses ‘feature similarity’ to predict the values of any new data points.
This means that the new point is assigned a value based on how closely it resembles the points in the training set.
This can be very useful in making predictions about the missing values by finding the k’s closest neighbours to the observation with missing data and then imputing them based on the non-missing values in the neighbourhood.
Let’s see some example code using Impyute library which provides a simple and easy way to use KNN for imputation:KNN Imputation for California Housing DatasetHow does it work?It creates a basic mean impute then uses the resulting complete list to construct a KDTree.
Then, it uses the resulting KDTree to compute nearest neighbours (NN).
After it finds the k-NNs, it takes the weighted average of them.
Pros:Can be much more accurate than the mean, median or most frequent imputation methods (It depends on the dataset).
Cons:Computationally expensive.
KNN works by storing the whole training dataset in memory.
K-NN is quite sensitive to outliers in the data (unlike SVM)5- Imputation Using Multivariate Imputation by Chained Equation (MICE)Main steps used in multiple imputations [1]This type of imputation works by filling the missing data multiple times.
Multiple imputations (MIs) are much better than a single imputation as it measures the uncertainty of the missing values in a better way.
The chained equations approach is also very flexible and can handle different variables of different data types (ie.
, continuous or binary) as well as complexities such as bounds or survey skip patterns.
MICE imputation using impyute6- Imputation Using Deep Learning (DataWig):This method works very well with categorical and non-numerical features.
It is a library that learns Machine Learning models to impute missing values in a dataframe.
It supports both CPU and GPU for training.
Imputation using DataWigPros:Quite accurate compared to other methods.
It has some functions that can handle categorical data.
Cons:Single Column imputation.
Can be quite slow with large datasets.
You have to specify the columns that contain information about the target column that will be imputed.
In conclusion, there is no perfect way to compensate for the missing values in a dataset.
Each strategy can perform better for certain datasets and missing data types but may perform much worse on other types of datasets.
There are some set rules to decide which strategy to use for particular types of missing values, but beyond that, you should experiment and check which model works best for your dataset.
References:Buuren, S.
V.
, & Groothuis-Oudshoorn, K.
(2011).
Mice: Multivariate Imputation by Chained Equations in R.
Journal of Statistical Softwarehttps://impyute.
readthedocs.
io/en/master/index.
html.. More details