MissHDD: Hybrid Deterministic Diffusion for Hetrogeneous Incomplete Data Imputation
By: Youran Zhou, Mohamed Reda Bouadjenek, Sunil Aryal
Potential Business Impact:
Fixes missing info in computer charts.
Incomplete data are common in real-world tabular applications, where numerical, categorical, and discrete attributes coexist within a single dataset. This heterogeneous structure presents significant challenges for existing diffusion-based imputation models, which typically assume a homogeneous feature space and rely on stochastic denoising trajectories. Such assumptions make it difficult to maintain conditional consistency, and they often lead to information collapse for categorical variables or instability when numerical variables require deterministic updates. These limitations indicate that a single diffusion process is insufficient for mixed-type tabular imputation. We propose a hybrid deterministic diffusion framework that separates heterogeneous features into two complementary generative channels. A continuous DDIM-based channel provides efficient and stable deterministic denoising for numerical variables, while a discrete latent-path diffusion channel, inspired by loopholing-based discrete diffusion, models categorical and discrete features without leaving their valid sample manifolds. The two channels are trained under a unified conditional imputation objective, enabling coherent reconstruction of mixed-type incomplete data. Extensive experiments on multiple real-world datasets show that the proposed framework achieves higher imputation accuracy, more stable sampling trajectories, and improved robustness across MCAR, MAR, and MNAR settings compared with existing diffusion-based and classical methods. These results demonstrate the importance of structure-aware diffusion processes for advancing deep learning approaches to incomplete tabular data.
Similar Papers
MissDDIM: Deterministic and Efficient Conditional Diffusion for Tabular Data Imputation
Artificial Intelligence
Fills in missing table data quickly and reliably.
Diffusion Transformers for Imputation: Statistical Efficiency and Uncertainty Quantification
Machine Learning (CS)
Fixes missing data in charts and graphs.
Latent Discrete Diffusion Models
Machine Learning (CS)
Makes computers write better stories and sentences.