Missing data imputation is a fundamental problem in data analysis, and many studies have been conducted to improve its performance by exploring model structures and learning procedures. However, data augmentation, as a simple yet effective method, has not received enough attention in this area. In this paper, we propose a novel data augmentation method called Missingness Augmentation (MisA) for generative imputation models. Our approach dynamically produces incomplete samples at each epoch by utilizing the generator's output, constraining the augmented samples using a simple reconstruction loss, and combining this loss with the original loss to form the final optimization objective. As a general augmentation technique, MisA can be easily integrated into generative imputation frameworks, providing a simple yet effective way to enhance their performance. Experimental results demonstrate that MisA significantly improves the performance of many recently proposed generative imputation models on a variety of tabular and image datasets. The code is available at \url{https://github.com/WYu-Feng/Missingness-Augmentation}.
翻译:缺失数据插补是数据分析中的基本问题,许多研究通过探索模型结构和学习流程来提升其性能。然而,数据增强作为一种简单而有效的方法,在该领域尚未得到充分关注。本文提出一种名为缺失增强(MisA)的新型数据增强方法,专门用于生成式插补模型。该方法在每个训练周期利用生成器的输出动态生成不完整样本,通过简单的重构损失约束增强后的样本,并将该损失与原始损失结合构成最终优化目标。作为一种通用增强技术,MisA可轻松集成到生成式插补框架中,提供一种简单有效的性能提升方式。实验结果表明,MisA在多种表格和图像数据集上显著提升了近期提出的多个生成式插补模型的性能。代码开源在\url{https://github.com/WYu-Feng/Missingness-Augmentation}。