AugMask is a plug-and-play training framework that lets diffusion models on incomplete tabular data use stochastic augmentation for conditioning and observed-only supervision, outperforming missing-aware baselines via a Rao-Blackwellized objective.
Impute-then-scale,
1 Pith paper cite this work. Polarity classification is still indexing.
1
Pith paper citing it
fields
cs.LG 1years
2026 1verdicts
UNVERDICTED 1representative citing papers
citing papers explorer
-
AugMask: Training Diffusion Models on Incomplete Tabular Data via Stochastic Augmentation and Masking
AugMask is a plug-and-play training framework that lets diffusion models on incomplete tabular data use stochastic augmentation for conditioning and observed-only supervision, outperforming missing-aware baselines via a Rao-Blackwellized objective.