Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Impugan: Learning Conditional Generative Models for Robust Data Imputation

About

Incomplete data are common in real-world applications. Sensors fail, records are inconsistent, and datasets collected from different sources often differ in scale, sampling rate, and quality. These differences create missing values that make it difficult to combine data and build reliable models. Standard imputation methods such as regression models, expectation-maximization, and multiple imputation rely on strong assumptions about linearity and independence. These assumptions rarely hold for complex or heterogeneous data, which can lead to biased or over-smoothed estimates. We propose Impugan, a conditional Generative Adversarial Network (cGAN) for imputing missing values and integrating heterogeneous datasets. The model is trained on complete samples to learn how missing variables depend on observed ones. During inference, the generator reconstructs missing entries from available features, and the discriminator enforces realism by distinguishing true from imputed data. This adversarial process allows Impugan to capture nonlinear and multimodal relationships that conventional methods cannot represent. In experiments on benchmark datasets and a multi-source integration task, Impugan achieves up to 82\% lower Earth Mover's Distance (EMD) and 70\% lower mutual-information deviation (MI) compared to leading baselines. These results show that adversarially trained generative models provide a scalable and principled approach for imputing and merging incomplete, heterogeneous data. Our model is available at: github.com/zalishmahmud/impuganBigData2025

Zalish Mahmud, Anantaa Kotal, Aritran Piplai• 2025

Related benchmarks

TaskDatasetResultRank
ClassificationDiabetes 25% held-out (test)
Accuracy76
12
ClassificationHeart Disease 25% held-out (test)
Accuracy0.6
12
ClassificationAdult (test)
Accuracy85
12
ImputationAdult
KS Statistic0.54
4
ImputationHEART DISEASE
KS0.54
4
ImputationDiabetes
KS0.36
4
Data ImputationAdult
RMSE0.24
4
Data ImputationDiabetes
RMSE0.32
4
Data ImputationHEART DISEASE
RMSE0.3
4
Data ImputationHEART DISEASE
MI (dev)0.15
4
Showing 10 of 12 rows

Other info

Follow for update