GIFT: Guided Importance-Aware Fine-Tuning for Diffusion Language Models
On the map: Diffusion models → Discrete and text diffusion
Abstract (arXiv)
Diffusion models have recently shown strong potential in language modeling, offering faster generation compared to traditional autoregressive approaches. However, applying supervised fine-tuning (SFT) to diffusion models remains challenging, as they lack precise probability estimates at each denoising step. While the diffusion mechanism enables the model to reason over entire sequences, it also makes the generation process less predictable and often inconsistent. This highlights the importance of controlling key tokens that guide the direction of generation. To address this issue, we propose GIFT, an importance-aware finetuning method for diffusion language models, where tokens are assigned different importance weights based on their entropy. Derived from diffusion theory, GIFT delivers substantial gains: across diverse settings including different mainstream training datasets ranging from 1k to 10k in size, utilizing LoRA or full parameter fine-tuning, and training on base or instruct models, GIFT consistently achieves superior overall performance compared to standard SFT on four widely used reasoning benchmarks (Sudoku, Countdown, GSM8K, and MATH-500).
Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Placed in Discrete and text diffusion, one of 14 directions in Diffusion models. Builds on 18 of its 39 references that are in the map. First posted on arXiv 25 Sep 2025.
Builds on
18 of 39 references.
Closest ancestors · 4
Same direction, newest first
- 2025Dream 7B: Diffusion Large Language Models21 Aug
- 2025wd1: Weighted Policy Optimization for Reasoning in Diffusion Language Models7 Jul
- 2025Reinforcing the Diffusion Chain of Lateral Thought with Diffusion Language Models15 May
- 2023Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution25 Oct
Other references · 14
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,330
- 2024Tulu 3: Pushing Frontiers in Open Language Model Post-Trainingcited by 1,053
- 2025Beyond the 80/20 Rule: High-Entropy Minority Tokens Drive Effective Reinforcement Learning for LLM Reasoningcited by 559
- 2025The Entropy Mechanism of Reinforcement Learning for Reasoning Language Modelscited by 485
- 2025LLaDA 1.5: Variance-Reduced Preference Optimization for Large Language Diffusion Modelscited by 285
- 2021Curriculum Learning: A Surveycited by 259
- 2025MMaDA: Multimodal Large Diffusion Language Modelscited by 250
- 2025d1: Scaling Reasoning in Diffusion Large Language Models via Reinforcement Learningcited by 186
- 2024Rho-1: Not All Tokens Are What You Needcited by 133
Built on by
Citing papers per year: 2026: 4.
4 papers.
- 2026Learnability-Informed Fine-Tuning of Diffusion Language Modelscited by 3
- 2026Supervised Fine-Tuning Needs to Unlock the Potential of Token Prioritycited by 0
- 2026Towards Efficient Reasoning: Learning Causal Shortcuts for Diffusion Language Modelscited by 0
- 2026Fine-Tuning Diffusion Language Models with Context Selection and Target Weightingcited by 0
- 2026Fine-Tuning Diffusion Language Models with Context Selection and Target Weighting29 Sep 2026
- 2026Towards Efficient Reasoning: Learning Causal Shortcuts for Diffusion Language Models23 Sep 2026
- 2026Learnability-Informed Fine-Tuning of Diffusion Language Models21 May 2026
- 2026Supervised Fine-Tuning Needs to Unlock the Potential of Token Priority1 Feb 2026
Similar, not cited
Close in wording, no citation either way
- 2025Large Language Models to Diffusion Finetuning
- 2026Diffusion Language Models: An Experimental Analysis
- 2026Generative Frontiers: Why Evaluation Matters for Diffusion Language Models
- 2023Diffusion Language Models Can Perform Many Tasks with Scaling and Instruction-Finetuning
- 2026Scaling Beyond Masked Diffusion Language Models
- 2023Likelihood-Based Diffusion Language Models
- 2026Temporally-Resolved Token Attribution Reveals the Generation Dynamics of Diffusion Language Models
- 2023A Closer Look at Parameter-Efficient Tuning in Diffusion Models