Diffusion models for missing value imputation in tabular data

Shuhan Zheng, Nontawat Charoenphakdee

Keywords: Missing value imputation, Diffusion model, Tabular data

TL;DR: We propose a diffusion model for missing value imputation in tabular data that can handle both numerical and categorical features

Abstract: Missing value imputation in machine learning is the task of estimating the missing values in the dataset reasonably using available information. In this task, several deep generative modeling methods have been proposed and demonstrated their usefulness, e.g., generative adversarial imputation networks. Recently, diffusion models have gained popularity because of their effectiveness in the generative modeling task in images, texts, audio, etc. To our knowledge, less attention has been paid on the investigation of the effectiveness of diffusion models for missing value imputation in tabular data. Based on a recent development of diffusion models for time-series data imputation, we propose a diffusion model approach called ``Conditional Score-based Diffusion Models for Tabular data'' (CSDI_T). To effectively handle categorical variables and numerical variables simultaneously, we investigate three techniques: one-hot encoding, analog bit encoding, and feature tokenization. Experimental results on benchmark datasets demonstrated the effectiveness of CSDI_T compared with well-known existing methods, and also emphasized the importance of the categorical embedding techniques.

Community Implementations: [ 3 code implementations](https://www.catalyzex.com/paper/diffusion-models-for-missing-value-imputation/code)