JUMP Boosts Membership Inference on Diffusion Language Models.

Yeachan Jun, Albert No· July 21, 2026 View original

Summary

This paper introduces JUMP (Joint Uncertainty-Guided Mask Probing), a single-pass membership inference attack (MIA) designed for fine-tuned discrete diffusion language models (dLLMs). JUMP significantly improves the detection of whether an example was part of a model's fine-tuning data by exploiting dLLMs' unique parallel and any-order decoding capabilities.

Membership inference attacks (MIAs) are crucial for assessing data privacy in machine learning models, determining if specific data points were used in training. This research focuses on MIAs against fine-tuned discrete diffusion language models (dLLMs), which differ from autoregressive models by allowing attackers to mask arbitrary token sets and obtain parallel token distributions. Previous dLLM attacks, like SAMA, relied on multiple queries and random masking. The new attack, JUMP (Joint Uncertainty-Guided Mask Probing), leverages dLLMs' distinctive properties: any-order decodability to select low-reference-confidence positions, and parallel decodability to score these positions in a single joint masked query. JUMP computes a clipped target/reference reconstruction-gap statistic over these jointly masked positions. Evaluations on LLaDA-8B-Base across six domains show JUMP significantly outperforms SAMA, improving ROC-AUC from 0.82 to 0.90 and enhancing low-FPR detection. Crucially, JUMP achieves this with only one selector pass and one scoring pass per model, making it far more efficient.

Why it matters

Professionals involved in AI security, privacy, and model deployment need to understand and defend against advanced membership inference attacks, especially as dLLMs become more prevalent.

How to implement this in your domain

  1. 1Assess the privacy risks of fine-tuned diffusion language models within your organization.
  2. 2Implement robust data anonymization and privacy-preserving techniques for training data.
  3. 3Develop or adopt monitoring tools to detect potential membership inference attacks on deployed dLLMs.
  4. 4Conduct internal red-teaming exercises using JUMP-like attacks to evaluate model vulnerabilities.
  5. 5Stay updated on the latest research in membership inference and privacy-enhancing technologies for LLMs.

Who benefits

CybersecurityAI/ML DevelopmentData PrivacyLegal & ComplianceCloud Services

Key takeaways

  • JUMP is a new, efficient membership inference attack targeting fine-tuned discrete diffusion language models (dLLMs).
  • It exploits dLLMs' unique parallel and any-order decoding for improved attack efficacy.
  • JUMP significantly outperforms previous attacks like SAMA, achieving higher ROC-AUC with fewer queries.
  • This highlights increased privacy risks for dLLMs and the need for stronger defenses.

Original post by Yeachan Jun, Albert No

"arXiv:2607.16207v1 Announce Type: new Abstract: Membership inference attacks (MIAs) test whether a candidate example appeared in a model's training data. We study MIAs for fine-tuned discrete diffusion language models (dLLMs), where membership means inclusion in the target model'…"

View on X

Originally posted by Yeachan Jun, Albert No on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses