Pith. sign in

REVIEW 8 cited by

MissDiff: Training Diffusion Models on Tabular Data with Missing Values

Not yet reviewed by Pith; the record is open.

This paper has not been read by Pith yet. Machine review is queued; the pith claim, tier, and objections will appear here once it completes.

SPECIMEN: schema-true, not a live event

T0 review · schema-true

One-sentence machine reading of the paper's core claim.

pith:XXXXXXXX · record.json · timestamp

arxiv 2307.00467 v1 pith:WWBUUCSB submitted 2023-07-02 cs.LG stat.ML

classification cs.LGstat.ML
keywords datadiffusiontabulartraininglearningmissingmodelproposed
verification ladder T0 review T1 audit T2 compute T3 formal
0 comments
read the original abstract

The diffusion model has shown remarkable performance in modeling data distributions and synthesizing data. However, the vanilla diffusion model requires complete or fully observed data for training. Incomplete data is a common issue in various real-world applications, including healthcare and finance, particularly when dealing with tabular datasets. This work presents a unified and principled diffusion-based framework for learning from data with missing values under various missing mechanisms. We first observe that the widely adopted "impute-then-generate" pipeline may lead to a biased learning objective. Then we propose to mask the regression loss of Denoising Score Matching in the training phase. We prove the proposed method is consistent in learning the score of data distributions, and the proposed training objective serves as an upper bound for the negative likelihood in certain cases. The proposed framework is evaluated on multiple tabular datasets using realistic and efficacious metrics and is demonstrated to outperform state-of-the-art diffusion model on tabular data with "impute-then-generate" pipeline by a large margin.

Discussion (0). Sign in to comment.

Forward citations

Cited by 8 Pith papers

Reviewed papers in the Pith corpus that reference this work. Sorted by Pith novelty score.

  1. Learning What Not to Impute: An Uncertainty-Aware Diffusion Framework for Meaningful Missingness

    cs.LG 2026-06 unverdicted novelty 7.0 of 10

    Diff-Joint is a diffusion framework that alternates conditional sampling and uncertainty-aware aggregation to jointly infer imputation values and missingness labels for selective imputation in tabular data.

  2. CoreFlow: Low-Rank Matrix Generative Models

    cs.LG 2026-04 unverdicted novelty 6.0 of 10

    CoreFlow is a low-rank matrix generative model that trains normalizing flows on shared subspaces to improve efficiency and quality for high-dimensional limited-sample data, including incomplete matrices.

  3. Harpoon: Generalised Manifold Guidance for Conditional Tabular Diffusion

    cs.LG 2026-02 conditional novelty 6.0 of 10

    Gradients of any differentiable tabular constraint lie in the tangent space of the data manifold, enabling a single trained diffusion model to satisfy unseen conditions via inference-time guidance.

  4. Flow Matching with Missing Data

    cs.LG 2026-07 conditional novelty 5.0 of 10

    Resampling missing coordinates and averaging the flow-matching loss reproduces the complete-data objective exactly under MCAR with oracle completions; one completion per example is optimal for a fixed budget.

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

    cs.LG 2026-06 unverdicted novelty 5.0 of 10

    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...

  6. Latent Diffusion for Missing Data

    cs.LG 2026-05 unverdicted novelty 5.0 of 10

    A VAE-based latent diffusion model trained on incomplete data maintains sample quality and imputation performance up to 50% missingness while pixel-space diffusion degrades.

  7. Incomplete Data, Complete Dynamics: A Diffusion Approach

    cs.LG 2025-09 unverdicted novelty 5.0 of 10

    A conditional diffusion model trained on partitioned incomplete samples for physical dynamics achieves asymptotic convergence to the true generative process under mild conditions and outperforms baselines in imputation.

  8. How Far Are We from Generating Missing Modalities with Foundation Models?

    cs.MM 2025-06 unverdicted novelty 5.0 of 10

    Evaluates 42 variants of foundation models across three formalized paradigms for missing modality reconstruction, identifies shortfalls in semantic extraction and validation, and introduces an agentic framework that r...

Pith tools