mlsanityData to 14 Oct 2026

Discrete Diffusion Language Models Are Training-Free Multi-Label Classifiers

arXiv 2608.14649 ↗First posted 30 Jul 2026cs.LGcited by 0

On the map: Diffusion models → Discrete and text diffusion

Abstract (arXiv)

We present dLLM-SetScore, a training-free method that uses discrete masked-diffusion language models for multi-label text classification. For each candidate label, it asks a short yes/no question and compares the probabilities of the two answer tokens at one masked position. The method uses no task-specific fine-tuning or training on textual-entailment datasets; a 200-example labelled validation slice selects thresholds, temperature, and prompt wording. We first show that placing all labels in one prompt creates a strong slot-position asymmetry: the first answer slot is predicted positive on 99.4% of GoEmotions examples and 100% of Reuters examples. Per-label scoring places every label in the same syntactic position, making predictions invariant to label order and avoiding this artifact. We evaluate LLaDA-8B and Dream-7B on six datasets against NLI models, an autoregressive LLM, SetFit, and supervised classifiers. On the five datasets shared by both diffusion families, Instruct checkpoints improve macro-F1 in 9 of 10 comparisons and micro-F1 in 8 of 10, although these comparisons do not identify the cause. Within our protocol, LLaDA-Instruct records the highest training-free values for both Reuters and ECtHR metrics. We prove permutation invariance, characterize thresholded decisions under weighted Hamming loss, and derive shortlist ceilings for recall and F1. An exploratory local Joint Set Refinement step lowers F1 from biased and unbiased initializations and is retained as a negative result.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Discrete and text diffusion, one of 14 directions in Diffusion models. Builds on 13 of its 20 references that are in the map. First posted on arXiv 30 Jul 2026.

Builds on

13 of 20 references.

Closest ancestors · 1

Same direction, newest first

  1. 2025Dream 7B: Diffusion Large Language Models21 Aug

Other references · 12

  1. 2019BART: Denoising Sequence-to-Sequence Pre-training for Natural Language Generation, Translation, and Comprehensioncited by 5,893
  2. 2020DeBERTa: Decoding-enhanced BERT with Disentangled Attentioncited by 2,181
  3. 2021Structured Denoising Diffusion Models in Discrete State-Spacescited by 1,719
  4. 2020GoEmotions: A Dataset of Fine-Grained Emotionscited by 486
  5. 2022A Continuous Time Framework for Discrete Denoising Modelscited by 451
  6. 2019Benchmarking Zero-shot Text Classification: Datasets, Evaluation and Entailment Approachcited by 308
  7. 2023Your Diffusion Model is Secretly a Zero-Shot Classifiercited by 306
  8. 2023Text-to-Image Diffusion Models are Zero-Shot Classifierscited by 143
  9. 2019Large-Scale Multi-Label Text Classification on EU Legislationcited by 104
  10. 2018SGM: Sequence Generation Model for Multi-label Classificationcited by 89
  11. 2021Paragraph-level Rationale Extraction through Regularization: A case study on European Court of Human Rights Casescited by 77
  12. 2023An Exploration of Encoder-Decoder Approaches to Multi-Label Classification for Legal and Biomedical Textcited by 23

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025Large Language Models Do Multi-Label Classification Differently
  2. 2024Scaling up Masked Diffusion Models on Text
  3. 2025Calibrating Pre-trained Language Classifiers on LLM-generated Noisy Labels via Iterative Refinement
  4. 2025Diffusion-Based Cross-Modal Feature Extraction for Multi-Label Classification
  5. 2026Improved Large Language Diffusion Models
  6. 2024Comparing Specialised Small and General Large Language Models on Text Classification: 100 Labelled Samples to Achieve Break-Even Performance
  7. 2026Diffusion Language Models for Speech Recognition
  8. 2026Diffusion Language Models: An Experimental Analysis