arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2606.03347cs.LGcs.AIstat.ML

AugMask: 通过随机增强和掩码在不完整表格数据上训练扩散模型

AugMask: Training Diffusion Models on Incomplete Tabular Data via Stochastic Augmentation and Masking

KAIST(韩国科学技术院)

Jungkyu Kim, Taeyoung Park, Kibok Lee

AI总结:

提出AugMask训练框架,通过条件随机增强和仅对观测坐标去噪,使标准扩散模型适应缺失表格数据,并连接Rao-Blackwellized目标实现方差加权惩罚,优于专门处理缺失的基线。

摘要:

基于分数的扩散模型已成为突出的深度生成模型;然而,它们在表格数据上的应用仍然具有挑战性,因为其主干网络假设输入完全指定,而现实世界的表格数据通常包含缺失值。我们提出了AugMask,一个即插即用的训练框架,通过将条件与监督分离,使对缺失不敏感的主干网络适应不完整数据。AugMask 1) 使用轻量级辅助模型通过条件随机增强构建数值输入,2) 仅对观测坐标应用去噪监督。实际上,增强的缺失条目作为不确定的条件上下文,而不是训练目标。我们将此训练规则与Rao-Blackwellized目标联系起来,并表明对缺失条目进行边缘化会产生方差加权的敏感性惩罚,从而阻止对不确定补全的过度依赖。在多种数据集和缺失机制下,AugMask使基于扩散的标准表格生成器优于专门处理缺失的基线方法。

英文摘要:

Score-based diffusion models have emerged as prominent deep generative models; however, their application to tabular data remains challenging because their backbones assume fully specified inputs, whereas real-world tabular data often contain missing values. We propose AugMask, a plug-and-play training framework that adapts missing-unaware backbones to incomplete data by separating conditioning from supervision. AugMask 1) constructs numeric inputs via conditional stochastic augmentation using lightweight auxiliary models, and 2) applies denoising supervision only to observed coordinates. In effect, augmented missing entries serve as uncertain conditioning context rather than training targets. We connect this training rule to a Rao--Blackwellized objective and show that marginalizing missing entries yields a variance-weighted sensitivity penalty, discouraging over-reliance on uncertain completions. Across diverse datasets and missingness regimes, AugMask enables standard diffusion-based tabular generators to outperform specialized missing-aware baselines.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑