Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Continuous Diffusion Models Can Obey Formal Syntax

About

Diffusion language models offer a promising alternative to autoregressive models due to their global, non-causal generation process, but their continuous latent dynamics make discrete constraints -- e.g., the output should be a JSON file that matches a given schema -- difficult to impose. We introduce a training-free guidance method for steering continuous diffusion language models to satisfy formal syntactic constraints expressed using regular expressions. Our approach constructs an analytic score estimating the probability that a latent state decodes to a valid string accepted by a given regular expression, and uses its gradient to guide sampling, without training auxiliary classifiers. The denoising process targets the base model conditioned on syntactic validity. We implement our method in Diffinity on top of the PLAID diffusion model and evaluate it on 180 regular-expression constraints over JSON and natural-language benchmarks. Diffinity achieves 68-96\% constraint satisfaction while incurring only a small perplexity cost relative to unconstrained sampling, outperforming autoregressive constrained decoding in both constraint satisfaction and output quality.

Jinwoo Kim, Taylor Berg-Kirkpatrick, Loris D'Antoni• 2026

Related benchmarks

TaskDatasetResultRank
Constrained Text Generation70 JSON schema benchmarks
Saturation68.4
4
Constrained Text GenerationNatural Language Benchmarks Prefix constraint
Saturation95.7
3
Constrained Text GenerationNatural Language Benchmarks Suffix constraint
Saturation96.8
3
Constrained Text GenerationNatural Language Benchmarks Appearance constraint
Saturation92.5
3
Constrained Text GenerationNatural Language Benchmarks Between-n constraint
Saturation85.5
3
Constrained Text GenerationNatural Language Benchmarks Between (ubd.) constraint
Saturation93.8
3
Constrained Text GenerationNatural Language Benchmarks Word Length constraint
Saturation95
3
Showing 7 of 7 rows

Other info

Follow for update