Abstract
We study supervised classification problems with missing feature values. Existing approaches often decouple imputation from classification, producing imputations that may be plausible but uninformative for prediction. Instead, we propose the Integrated Imputation and Classification Network (IICN), which jointly trains an imputer and an ${(n{+}1)}$-classdiscriminator adversarially with a single class supervised classification objective, where the discriminator learns to distinguish among the $n$ true classes and an additional ``imputed" class. We prove that at the global optimum, the imputer and discriminator together implement marginalization over missing coordinates and yield a Bayes-optimal classifier. We evaluate IICN on FashionMNIST, CIFAR-10, and tabular datasets with naturally occurring missingness. IICN outperforms classical impute-then-classify pipelines and recent generative baselines, showing strong robustness and accuracy in challenging settings.
Keywords
Subject
Publication details
- Journal
- Not available
- Open access
- Green open access
Cite this article
APA 7
Liu, Y., Liang, B., Tizghadam, A., & Albanese, I. (2026). Integrated Imputation-Classification for Supervised Learning with Missing Data. https://omanscience.com/en/articles/integrated-imputation-classification-for-supervised-learning-with-missing-data
MLA 9
Liu, Yue, et al. "Integrated Imputation-Classification for Supervised Learning with Missing Data." https://omanscience.com/en/articles/integrated-imputation-classification-for-supervised-learning-with-missing-data.
Chicago (author–date)
Liu, Yue, Ben Liang, Ali Tizghadam, and Ilijc Albanese. 2026. "Integrated Imputation-Classification for Supervised Learning with Missing Data." https://omanscience.com/en/articles/integrated-imputation-classification-for-supervised-learning-with-missing-data.
Harvard
Liu, Y., Liang, B., Tizghadam, A. and Albanese, I. (2026) 'Integrated Imputation-Classification for Supervised Learning with Missing Data', Available at: https://omanscience.com/en/articles/integrated-imputation-classification-for-supervised-learning-with-missing-data.
Vancouver
Liu Y, Liang B, Tizghadam A, Albanese I. Integrated Imputation-Classification for Supervised Learning with Missing Data. https://omanscience.com/en/articles/integrated-imputation-classification-for-supervised-learning-with-missing-data
IEEE
Y. Liu, B. Liang, A. Tizghadam, and I. Albanese, "Integrated Imputation-Classification for Supervised Learning with Missing Data," https://omanscience.com/en/articles/integrated-imputation-classification-for-supervised-learning-with-missing-data.