---
格式版本: 2
标题: "LeanGRPO: Eliminating Redundant Recomputation in Diffusion RL"
原文链接: "https://arxiv.org/abs/2609.03528"
发布日期: "2026-09-03"
发布时间校准状态: "found"
发布时间需复核: "否"
发布时间来源: "rule:local:strict_original_body"
发布时间证据: "**\\[v1\\]** Thu, 3 Sep 2026 08:24:56 UTC (1,466 KB)"
发布时间校准原因: "规则确认唯一严格发布时间，来源 local:strict_original_body"
发布时间校准置信度: "high"
发布时间候选数量: 18
发布时间严格候选数量: 6
发布时间原页读取状态: "source template page reused from URL open"
发布时间未找到原因: ""
发布时间校准时间: "2026-09-06T21:26:28+08:00"
发布时间仲裁状态: "skipped"
发布时间仲裁尝试次数: 0
发布时间仲裁耗时毫秒: 0
发现时间: "2026-09-06T21:24:50+08:00"
入库时间: "2026-09-06T13:26:28.311Z"
来源平台: "arXiv 学术论文搜索"
搜索渠道: "source_template"
搜索词: "https://arxiv.org/search/?query=Scale-up&searchtype=all"
匹配关键词:
  - "Scale-up"
  - "AI"
相关厂家:
  []
相关专家:
  []
内容类型: "网页"
抓取工具: "Free Fetch + Defuddle"
清洗工具: "Defuddle Markdown + Defuddle/Readability 正文提取"
原始附件:
  []
AI优质: "否"
AI打分: 15
AI分档: "非优质"
AI质检状态: "不通过"
AI打分理由: "论文主题为扩散强化学习训练优化，与超节点/AI Rack/机柜级AI基础设施、供电散热互连等主题完全无关。来源权威但内容无关，仅学科分类提及硬件架构，无技术或商业信号。"
AI质检模型: "zj-deepseek-v4-flash"
AI质检时间: "2026-09-06T21:26:34+08:00"
AI主题相关性: 0
AI来源权威性: 10
AI新颖性: 2
AI技术细节: 0
AI商业部署信号: 0
AI完整性: 3
AI摘要: "LeanGRPO 提出两种免重计算调度，用于扩散强化学习的轨迹对数概率训练，通过重构数据并行布局并让 rollout 阶段直接保留计算图或立即反向传播，消除了 DanceGRPO、FlowGRPO 在策略更新时对选定时间步的冗余重计算；"
AI摘要模型: "ali-deepseek-v4-flash"
AI摘要时间: "2026-09-06T23:44:07.986Z"
采集批次: "2026年9月6日19点27分09秒"
采集批次ID: "20260906-192709-237"
去重键: "https://arxiv.org/abs/2609.03528"
---

## Computer Science > Machine Learning

## Title:LeanGRPO: Eliminating Redundant Recomputation in Diffusion RL

Authors:[Sijie Wang](https://arxiv.org/search/cs?searchtype=author&query=Wang,+S), [Zhiqiang Tan](https://arxiv.org/search/cs?searchtype=author&query=Tan,+Z), [Xinrui Yang](https://arxiv.org/search/cs?searchtype=author&query=Yang,+X), [Shaohuai Shi](https://arxiv.org/search/cs?searchtype=author&query=Shi,+S)

[View PDF](https://arxiv.org/pdf/2609.03528) [HTML (experimental)](https://arxiv.org/html/2609.03528v1)

> Abstract:Diffusion reinforcement learning (RL) has recently achieved significant success in post-training image and video generative models. However, most diffusion RL methods, including DanceGRPO and FlowGRPO, recompute selected timesteps with gradient tracking after rollout. Under on-policy training with the same backend for rollout and update, this recomputation is mathematically redundant. Intuitively, the rollout and policy update steps can reuse the same feed-forward backbone to avoid redundant computation, but doing so can incur a large memory overhead during rollout. To address the issue, we present LeanGRPO by restructuring the data-parallel layout and introducing two recompute-free training schedules for trajectory-logprob diffusion RL: (1) LeanGRPO-Retain enables gradient tracking during rollout and directly reuses the resulting computation graphs and saved activations for backward during update, requiring no recomputation; and (2) LeanGRPO-Reweight also enables gradients during rollout, but immediately backpropagates each selected step using a provisional advantage and delays gradient synchronization, then corrects the provisional gradients with the true advantage after the trajectory is completed. These schedules target different model scales and input sizes. Across FlowGRPO/DanceGRPO with FLUX.1-dev and Wan, LeanGRPO achieves up to 1.83x end-to-end speedup while preserving the original optimization objective.

| Subjects: | Machine Learning (cs.LG); Artificial Intelligence (cs.AI); Hardware Architecture (cs.AR) |
| --- | --- |
| Cite as: | [arXiv:2609.03528](https://arxiv.org/abs/2609.03528) \[cs.LG\] |
|  | (or [arXiv:2609.03528v1](https://arxiv.org/abs/2609.03528v1) \[cs.LG\] for this version) |
|  | [https://doi.org/10.48550/arXiv.2609.03528](https://doi.org/10.48550/arXiv.2609.03528) |

## Submission history

From: Sijie Wang \[[view email](https://arxiv.org/show-email/d98d1850/2609.03528)\]  
**\[v1\]** Thu, 3 Sep 2026 08:24:56 UTC (1,466 KB)

[Which authors of this paper are endorsers?](https://arxiv.org/auth/show-endorsers/2609.03528) | Disable MathJax ([What is MathJax?](https://info.arxiv.org/help/mathjax.html))