cs.AIMar 19, 2026

D5P4: Partition Determinantal Point Process for Diversity in Parallel Discrete Diffusion Decoding

Authors: Jonathan Lys, Vincent Gripon, Axel Marmoret, Lukas Mauch, Fabien Cardinaux, Ghouthi Boukli Hacene, Bastien Pasdeloup

Organizations: IMT Atlantique, Lab-STICC, UMR CNRS 6285, F-29238 Brest, France. · Sony Europe Ltd. Stuttgart Technology Center, EUREC, Germany.

Abstract

Discrete diffusion models are promising alternatives to autoregressive approaches for text generation, yet their decoding methods remain under-studied. Standard autoregressive search procedures, such as beam search, do not directly apply to iterative denoising, where hypotheses are complete intermediate sequences rather than left-to-right prefixes. Furthermore, existing diffusion decoding procedures only provide limited control over the diversity and coverage of retained hypotheses. In this work, we introduce D5P4, a beam-style decoding method tailored to discrete diffusion models, which casts intermediate beam selection as MAP inference under a partitioned Determinantal Point Process. This yields a model-internal batch objective that balances quality and diversity without external verifiers. Experiments on open-ended generation, question answering, and mathematical reasoning show that D5P4 improves diversity and pass@kk coverage while matching or surpassing baseline quality and fidelity

Explore similar work

CardsList
  1. Mean-Field Parallel Decoding for Discrete Diffusion Language Models

    Jun 14, 2026Tamim Zoabi, Ameen Ali, Liran Ringel +1Parallel DecodingDiffusion Language Model Decoding

  2. Generalized Discrete Diffusion from Snapshots

    Mar 22, 2026Oussama Zekri, Théo Uscidda, Nicolas Boullé +1Diffusion ModelsDiffusion Language Models

  3. Beyond Single Tokens: Distilling Discrete Diffusion Models via Discrete MMD

    Mar 20, 2026Emiel Hoogeboom, David Ruhe, Jonathan Heek +2Diffusion Model DistillationDiffusion Models