← 研踪 PaperTrailarXiv:2610.04273 · v2
Integrated Imputation-Classification for Supervised Learning with Missing Data
首次提交 2026-10-03 11:59
版本更新 2026-11-09 05:48
雷达观测 2026-11-10 13:19
摘要
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)}$-class discriminator adversarially with a single class-supervised objective, where the discriminator learns to distinguish among the $n$ true classes and an additional ``imputed'' class. We prove that it admits a global optimum where 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.
作者备注：camera ready
观测到的版本
相邻记录可能跳过中间版本；首次发现 v2 及以上也会标为版本更新。
v2 · 2026-11-09 05:48 · 最新观测版本
Integrated Imputation-Classification for Supervised Learning with Missing Data
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)}$-class discriminator adversarially with a single class-supervised objective, where the discriminator learns to distinguish among the $n$ true classes and an additional ``imputed'' class. We prove that it admits a global optimum where 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.
查看此版本解读 →查看此版本原文 ↗