A comparison of imputation methods for categorical data
Objectives: Missing data is commonplace in clinical databases, which are being increasingly used for research. Without giving any regard to missing data, results from analysis may become biased and unrepresentative. Clinical databases contain mainly categorical variables. This study aims to assess t...
Main Authors: | , , |
---|---|
Format: | Article |
Language: | English |
Published: |
Elsevier
2023-01-01
|
Series: | Informatics in Medicine Unlocked |
Subjects: | |
Online Access: | http://www.sciencedirect.com/science/article/pii/S2352914823002289 |
_version_ | 1797646620461891584 |
---|---|
author | Shaheen MZ. Memon Robert Wamala Ignace H. Kabano |
author_facet | Shaheen MZ. Memon Robert Wamala Ignace H. Kabano |
author_sort | Shaheen MZ. Memon |
collection | DOAJ |
description | Objectives: Missing data is commonplace in clinical databases, which are being increasingly used for research. Without giving any regard to missing data, results from analysis may become biased and unrepresentative. Clinical databases contain mainly categorical variables. This study aims to assess the methods used for imputation in categorical variables. Materials and methods: We utilized data extracted from paper-based maternal health records from Kawempe National Referral Hospital, Uganda. We compared the following imputation methods for categorical data in an empirical analysis: Mode, K-Nearest Neighbors (KNN), Random Forest (RF), Sequential Hot-Deck (SHD), and Multiple Imputation by Chained Equations (MICE). The five imputation methods were first compared by accuracy of predicting the missing values. Next, the imputation methods were compared by predictive accuracy of the outcome variable in four classifiers. The consistency of performance of imputation methods across different levels of missing data (5%–50 %) was assessed by Kendall's W test. Results: KNN imputation had the highest precision score at levels (5%–50 %) of MCAR missing data. At lower proportions of missing data (5 %, 10 %, 15 %, 20 %), RF imputation had the second-highest precision score. SHD imputation had the worst precision at all levels of missing data. In the prediction of the outcome, the methods performed differently at all proportions of missing data in the four classifiers. Even though KNN imputation was the best method in predicting the missing values, it did not consistently enhance the predictive accuracy of the classifiers at all levels of missing data. Our findings show that a high precision score of an imputation method does not translate into higher predictive accuracy in classifiers. Conclusions: KNN imputation is the best method in predicting missing values in categorical variables. There is no universal best imputation method that yields the highest predictive accuracy at all proportions of missing data. |
first_indexed | 2024-03-11T15:05:14Z |
format | Article |
id | doaj.art-e09ede66a247432596c4b20e295f0850 |
institution | Directory Open Access Journal |
issn | 2352-9148 |
language | English |
last_indexed | 2024-03-11T15:05:14Z |
publishDate | 2023-01-01 |
publisher | Elsevier |
record_format | Article |
series | Informatics in Medicine Unlocked |
spelling | doaj.art-e09ede66a247432596c4b20e295f08502023-10-30T06:05:17ZengElsevierInformatics in Medicine Unlocked2352-91482023-01-0142101382A comparison of imputation methods for categorical dataShaheen MZ. Memon0Robert Wamala1Ignace H. Kabano2African Centre of Excellence in Data Science, University of Rwanda, P.O. Box 4285 Kigali-Rwanda, Kigali, Rwanda; Corresponding author.Makerere University, P.O. Box 7062, Kampala, UgandaAfrican Centre of Excellence in Data Science, University of Rwanda, P.O. Box 4285 Kigali-Rwanda, Kigali, RwandaObjectives: Missing data is commonplace in clinical databases, which are being increasingly used for research. Without giving any regard to missing data, results from analysis may become biased and unrepresentative. Clinical databases contain mainly categorical variables. This study aims to assess the methods used for imputation in categorical variables. Materials and methods: We utilized data extracted from paper-based maternal health records from Kawempe National Referral Hospital, Uganda. We compared the following imputation methods for categorical data in an empirical analysis: Mode, K-Nearest Neighbors (KNN), Random Forest (RF), Sequential Hot-Deck (SHD), and Multiple Imputation by Chained Equations (MICE). The five imputation methods were first compared by accuracy of predicting the missing values. Next, the imputation methods were compared by predictive accuracy of the outcome variable in four classifiers. The consistency of performance of imputation methods across different levels of missing data (5%–50 %) was assessed by Kendall's W test. Results: KNN imputation had the highest precision score at levels (5%–50 %) of MCAR missing data. At lower proportions of missing data (5 %, 10 %, 15 %, 20 %), RF imputation had the second-highest precision score. SHD imputation had the worst precision at all levels of missing data. In the prediction of the outcome, the methods performed differently at all proportions of missing data in the four classifiers. Even though KNN imputation was the best method in predicting the missing values, it did not consistently enhance the predictive accuracy of the classifiers at all levels of missing data. Our findings show that a high precision score of an imputation method does not translate into higher predictive accuracy in classifiers. Conclusions: KNN imputation is the best method in predicting missing values in categorical variables. There is no universal best imputation method that yields the highest predictive accuracy at all proportions of missing data.http://www.sciencedirect.com/science/article/pii/S2352914823002289ImputationCategorical variablesPrecision scoreSingle imputationMultiple imputation |
spellingShingle | Shaheen MZ. Memon Robert Wamala Ignace H. Kabano A comparison of imputation methods for categorical data Informatics in Medicine Unlocked Imputation Categorical variables Precision score Single imputation Multiple imputation |
title | A comparison of imputation methods for categorical data |
title_full | A comparison of imputation methods for categorical data |
title_fullStr | A comparison of imputation methods for categorical data |
title_full_unstemmed | A comparison of imputation methods for categorical data |
title_short | A comparison of imputation methods for categorical data |
title_sort | comparison of imputation methods for categorical data |
topic | Imputation Categorical variables Precision score Single imputation Multiple imputation |
url | http://www.sciencedirect.com/science/article/pii/S2352914823002289 |
work_keys_str_mv | AT shaheenmzmemon acomparisonofimputationmethodsforcategoricaldata AT robertwamala acomparisonofimputationmethodsforcategoricaldata AT ignacehkabano acomparisonofimputationmethodsforcategoricaldata AT shaheenmzmemon comparisonofimputationmethodsforcategoricaldata AT robertwamala comparisonofimputationmethodsforcategoricaldata AT ignacehkabano comparisonofimputationmethodsforcategoricaldata |