本页目录 Uno

Uno

深入解析 Uno 如何以轻量扩散 LoRA 和 Ψ-Spec 验证采样,在保持自回归目标分布的同时并行生成多 token,并从训练目标、采样策略、并发吞吐、RL rollout、复现条件与跨模型比较边界审视其真实价值。

自动研究时间:2026-09-09 09:02(Asia/Shanghai)
来源日期:Hugging Face Daily Papers 页面实际显示 Sep 8,列表最顶部为 Unlocking Lossless Speedups in LLMs via Discrete Diffusion
研究路径:Daily Papers 列表 -> Hugging Face 详情页 -> arXiv 摘要页 -> arXiv v1 PDF 与完整 TeX 源文 -> 官方项目页、代码仓库及模型集合。

执行摘要

自回归大语言模型的质量来自 next-token prediction,但其推理也被“下一个 token 必须等上一个 token”锁成串行过程。离散扩散模型能并行更新多个位置,却通常需要在速度与质量之间交换;传统 speculative decoding 可以严格保留目标模型分布,却要额外维护一个足够快、又与目标模型足够一致的 draft model。

Uno 的核心不是把自回归模型改造成扩散模型,而是在同一模型中保留两条参数路径。 冻结的 AR 权重 θAR\theta_{\mathrm{AR}} 继续定义最终输出分布;附加的低秩扩散权重 θΔ\theta_\Delta 只负责一次并行提出一个 token block。推理时,扩散路径先 draft,AR 路径再用标准拒绝采样验证,接受最长合法前缀并在拒绝位置从残差分布补采样。因此,只要实现满足算法假设,最终样本仍来自原 AR 分布。“lossless”指的是目标分布不变,不是浮点结果逐位相同,也不是任何硬件与并发条件下都必然更快。

训练阶段称为 Diffusion Distillation。作者在每个 AR 权重矩阵旁加入 gated LoRA,把干净序列与完全噪声序列拼接,在一次 block-causal forward 中同时得到冻结 AR teacher 和扩散 student 的 logits。总变分距离(TV)直接优化 speculative verification 的连续接受长度;实验中 TV-only 反而优于原始 DCD KL 项,显示该方法真正优化的是“与 verifier 匹配的 drafter”,而不是独立扩散生成质量。

采样器 Ψ\Psi-Spec 有两种主要工作点:高并发时用单候选 Linear Sampler,避免额外验证计算;低并发、GPU 偏 memory-bound 时用 Tree Sampler,把闲置算力用于并行验证多个高概率前缀。自研约 8B Uno 在论文的 1K-input/8K-output、单 H200 测试中,最大系统吞吐为 5,255 tok/s,对应 AR 基线 3,577 tok/s;batch 1 为 383 tok/s,对应 176 tok/s。基于 Qwen3-8B 的 Uno-Qwen 则在最大批量达到 5,733 tok/s,并报告相对基线最高 2.5 倍的单请求加速。论文摘要汇总的最高加速为 3 倍。

这些结果证明了一条很有工程价值的路线:用轻量适配器学习并行 proposal,再以原模型校正质量。但证据边界也很清楚。自研 Uno 的 23T-token AR 训练数据不公开;Mercury 2 的参数量、量化和硬件口径不透明;DiffusionGemma、Nemotron-Labs-Diffusion 与 Uno 的规模和训练数据不同。RL 端到端最高 40% 加速仍标注为后续版本再给细节。公开仓库已经包含 Nano-vLLM 推理、训练与评测代码及三个模型配方,但主要依赖 H200/Hopper 专用栈,论文表格仍需独立复现。

1. 论文基本信息

项目内容
论文标题Unlocking Lossless Speedups in LLMs via Discrete Diffusion
方法与模型名Uno;Diffusion-augmented LLM;Ψ\Psi-Spec
论文编号arXiv:2609.04010
当前版本v1,2026-09-03 提交
主分类Machine Learning(cs.LG)
作者Subham Sekhar Sahoo、Lingjie Chen、Khiem Pham、Jonathan Geuter、Chaitanya Dwivedi、Varad Pimpalkhute、Yash Akhauri、Alexander Moreno、Mikhail Yurochkin、Zhenting Wang、Mostafa Elhoushi、Nolan Dey、Shane Bergsma、Joel Hestness、John Thickstun、Eric Xing、Zhengzhong Liu
机构Institute of Foundation Models、UIUC、Cornell Tech、Harvard University、Cerebras Systems
开源模型Uno 8B、Uno 1B、Uno-Qwen3 8B
开源状态训练、推理、评测代码与公开 checkpoint 已提供

核心链接:

2. 背景与动机:为什么“多 token”仍是难题

2.1 自回归模型的质量优势伴随串行瓶颈

对长度为 LL 的序列,自回归模型按链式法则分解联合概率:

logpθ(x)==1Llogpθ(xx<).\log p_\theta(x)=\sum_{\ell=1}^{L}\log p_\theta(x^\ell\mid x^{<\ell}).

这种分解有稳定的 likelihood 训练目标,也天然适配 KV cache;代价是生成 LL 个 token 至少需要 LL 次因果决策。长 reasoning trace、代码生成和 Agent trajectory 会直接放大延迟,RL post-training 中 rollout 甚至可能成为主要成本。

并行预测对现代 GPU 尤其重要。低 batch 解码往往受模型权重与 KV 状态搬运限制,计算单元没有吃满;若一次 forward 能产出多个 token,就可以把同一次内存访问摊薄到更多有效输出上。但 batch 增大后,系统逐渐转为 compute-bound,多候选和多轮 denoising 的额外计算又可能吞掉收益。因此只测 batch 1 会高估部分并行生成方法的生产价值。

2.2 三条既有路线各有缺口

路线优点主要代价
Speculative decodingverifier 可严格保持目标 AR 分布需要独立 drafter、第二套参数与 KV cache,效果依赖 draft–target 对齐
Self-speculative / AR-diffusion 转换一个模型兼具两种生成模式若修改 base weights,通常不再保留原 AR 分布
原生 discrete diffusion LLM多位置并行生成,支持自纠错相对 AR 模型常有质量差距;高 batch 下加速可能消失
Multi-token prediction直接预测多个未来 token需要额外 heads 或架构修改,仍需验证与接受机制

Uno 想同时拿到四个属性:原 AR 模型的质量、严格的分布保持、多 token 并行、以及不依赖单独 draft model。它的设计原则是把“谁定义质量”和“谁优化速度”拆开。

3. 核心贡献

3.1 双权重路径:质量与速度解耦

Uno 的每一层包含两组逻辑权重:

  • AR 路径使用 θAR\theta_{\mathrm{AR}},按标准 pre-training、SFT 和 RL recipe 训练,定义目标分布;
  • 扩散路径使用 θAR+θΔ\theta_{\mathrm{AR}}+\theta_\Delta,其中 θΔ\theta_\Delta 是 LoRA adapter,只负责并行 draft;
  • 验证阶段关闭 adapter,只让 θAR\theta_{\mathrm{AR}} 计算目标概率并执行拒绝校正。

LoRA 不只是省参数。因为 draft 与 verifier 共享 base weights,它还把两个分布天然拉近,提升连续接受概率;同时两条路径共享一份主模型 KV cache,避免 EAGLE-3、DFlash 一类独立 drafter 的额外 cache。论文称框架也可用于 causal Transformer 之外的因果神经网络,例如 state-space model,不过实验只覆盖 Transformer。

3.2 Diffusion Distillation:一次 forward 训练 teacher 与 student

训练目标是让扩散路径一次并行给出的 block 分布逼近 AR 路径逐 token 产生的同一 block 分布。作者没有在 LLM 尺度保存完整 probability-flow ODE 的中间轨迹,而是把完全腐化序列 z1z_1 一步映射到干净序列 xx,并把长序列切成大小为 BB 的 block。

输入由 [x,z1][x,z_1] 拼接而成,block-causal mask 允许:

  1. 干净序列内部保持因果注意力;
  2. 每个噪声 block 内部使用因果注意力;
  3. 噪声 block 可以看到之前的干净 blocks;
  4. gated LoRA 在干净位置关闭、噪声位置打开。

这样一次 forward 便得到 θAR\theta_{\mathrm{AR}} 的 teacher logits 与 θAR+θΔ\theta_{\mathrm{AR}}+\theta_\Delta 的 student logits,base model 全程冻结。总训练目标为:

L=αLDCD+βLTV.\mathcal L=\alpha\mathcal L_{\mathrm{DCD}}+\beta\mathcal L_{\mathrm{TV}}.

LDCD\mathcal L_{\mathrm{DCD}} 是 student 对 teacher 的 blockwise KL;LTV\mathcal L_{\mathrm{TV}} 最小化两个 token 分布的总变分距离。由于 speculative decoding 的期望接受率直接受 TV distance 控制,后者比“生成看起来像不像”更贴近最终吞吐目标。

消融结果也验证了这一点:在 Uno-Qwen 单 epoch 设置中,TV-only 的平均 TPF 为 2.39,DCD+TV 与 DCD-only 都是 2.23;把 DCD 权重降至 0.01 后仅小幅升至 2.40。论文实际发布的三 epoch Uno-Qwen 因此采用 α=0,β=1\alpha=0,\beta=1

3.3 Ψ\Psi-Spec:并行提案,AR 拒绝校正

给定已生成前缀,采样器追加 B1B-1 个来自 uniform-state prior 的随机 token,再通过扩散路径一次并行 denoise。block 的第一个位置只使用 AR 权重,因此它与 verifier 分布完全一致、必然被接受;其余位置由 diffusion weights 并行提出。

随后 AR 路径一次性计算整段候选的概率。对第 ii 个 draft token,标准接受概率为:

ai=min(1,pAR(xi)pdraft(xi)).a_i=\min\left(1,\frac{p_{\mathrm{AR}}(x_i)}{p_{\mathrm{draft}}(x_i)}\right).

算法保留第一个拒绝点之前的最长前缀。若发生拒绝,就从归一化残差 [pARpdraft]+[p_{\mathrm{AR}}-p_{\mathrm{draft}}]_+ 中补采样;若整个 block 都接受,verifier 再从末尾 logits 多采一个 token。这一 correction 才是“lossless”的来源,扩散 drafter 本身不需要完美复刻 AR 分布。

每轮包含一次 draft forward 和一次 verify forward。论文将 tokens per forward pass 定义为 TPF,其理论范围为:

1TPFB+12.1\le \mathrm{TPF}\le\frac{B+1}{2}.

3.4 Linear 与 Tree 两种 sampler 对应两种部署目标

  • Linear Sampler:每个位置只采一个候选序列。高 batch 已接近 compute-bound,额外候选会增加验证成本,因此它适合最大化系统总吞吐。
  • Tree Sampler:每个位置取 top-KK,再保留概率最高的 VV 个前缀,以 tree attention 并行验证。低 batch 时用空闲算力交换更长接受前缀,适合最大化单请求吞吐。

这比给一个“最佳 block size”更符合生产现实。同一模型在 batch 1 和 batch 64 的瓶颈不同,吞吐最优配置自然也不同。自研 Uno 的系统吞吐最优点是 Linear B=4B=4;单请求最优点是 Tree (B,K,V)=(16,32,32)(B,K,V)=(16,32,32)

3.5 不增加上下文长度的 inference-time scaling

Ψ\Psi-Spec 允许 denoising steps TT 大于 draft block size BB,用更多扩散修正换更高 sample quality,而不增长 context。论文提出一种可能性:如果多步 denoising 的质量超过原 AR 路径,就应关闭 AR verification,否则 verifier 反而会把输出限制回 AR 质量水平。

这还是研究方向,不是本文已验证的结果。论文没有系统给出 T>BT>B 时的质量—计算曲线,也没有证明超过 AR 后如何维持“lossless”主张;一旦关闭 verifier,输出当然不再保证来自原 AR 分布。

4. 模型与训练设置

4.1 从头训练的 Uno

自研模型是 dense decoder-only causal Transformer:36 层、hidden size 4,096、MLP width 12,288、32 个 query heads、8 个 KV heads、约 9B 总参数,其中 transformer body 6.95B,较大的非共享 embedding/output matrix 约 2.05B;最大配置上下文为 524,288。

AR 权重使用内部数据训练约 23T tokens,并从 8K 逐步扩展到 32K、128K 和 512K context。扩散权重是 rank-128、LoRA-α=256\alpha=256 的 adapters,在冻结 AR 权重后用 SFT 数据中的 7B tokens 训练;训练约 60 小时,使用 8 个节点、每节点 8 张 H200。其扩散阶段训练 token 仅约为 AR 阶段的 0.03%,但绝对计算成本仍是 64 张 H200 约两天半,不能称为低门槛。

4.2 在 Qwen3-8B 上追加扩散能力

Uno-Qwen 保持 Qwen3-8B base weights 冻结,添加 0.35B 个 rank-128 LoRA 参数,在 OpenThoughts3-1.2M 上训练三 epoch,共 14.7B tokens;block curriculum 为 B{2,4,6,8,12,16}B\in\{2,4,6,8,12,16\},每半个 epoch 增大一次。训练约 32 小时,使用 4 个 8×H200 节点。

值得注意的是,作者没有 Qwen3 原始训练分布,OpenThoughts fine-tuning 本身还会降低 Qwen 的部分 benchmark quality;但在 OpenThoughts 上学到的 adapters 仍能对原 Qwen 分布产生经 verifier 校正后的加速。这支持“diffusion weights 不必与 AR weights 同分布训练”,但速度能否泛化到语言、代码风格和领域差异更大的数据,仍需要更广泛测量。

5. 实验设计与指标

5.1 为什么固定 1K 输入与 8K 输出

如果直接在 benchmark 生成上计 tokens/s,输出更短的弱模型可能显得更快。作者先在所有任务上测平均 TPF,再用 1,024 个随机输入 token 和 8,192 个固定输出 token 执行 8192/TPF\lceil8192/\mathrm{TPF}\rceil 个 decoding steps,尽量让不同方法承受相同的有效序列长度。

论文分别报告:

  • batch 1 的 per-request throughput,代表单请求延迟;
  • 单张 H200 能容纳的最大 batch 下的 aggregate system throughput,代表服务容量与成本;
  • benchmark average pass@1,用于比较不同模型的能力;
  • TPF,用于解释 sampler 接受效率,但不能替代墙钟吞吐,因为不同 drafter 每次 forward 的成本不同。

5.2 评测范围

自研 Uno 覆盖 Agent tool use(τ2/τ3\tau^2/\tau^3-Bench)、Terminal-Bench v2.1、SWE-bench Verified、AA-LCR、HLE、GPQA-Diamond、GSM8K、MATH500、AIME 2024–2026、MBPP 与 HumanEval。最长配置为 262K context、131K generation。

Uno-Qwen 的 lossless 对照是 EAGLE-3 与 DFlash;lossy 对照还包括 SDAR、TiDAR、OPDLM、I-DLM、Jacobi Forcing、Fast-dLLM v2、FLARE 与 LLaDA2.1-Flash。所有方法的硬件、base model、训练数据和公开程度并不完全相同,因此本文最可信的是同一 base model、同一 runtime 下的 Qwen lossless 对照,而不是所有跨模型排行榜数字。

6. 核心实验结果

6.1 自研 Uno 对 AR 与 d-LLM

模型最大系统吞吐(tok/s)batch 1 吞吐(tok/s)说明
Uno5,255383bfloat16,H200;不同 sampler 分别优化两列
自研 AR 基线3,577176与 Uno 共享目标 AR 权重
DiffusionGemma 26B-A4B1,1368364B active MoE,质量与规模不同
Nemotron-Labs-Diffusion 14B2,794290masked diffusion,模型更大
Mercury 21,197769论文引用外部 tracker;硬件、量化、参数量不透明

同源 AR 对照最能说明方法本身:最大 batch 下 Uno 约快 1.47 倍,batch 1 约快 2.18 倍,而且理论上保持同一目标分布。DiffusionGemma 的 batch 1 吞吐更高,但系统吞吐较低;这正是论文强调不能只看单请求速度的原因。

跨模型能力表中,Uno 报告 SWE-bench Verified 68.4、Terminal-Bench v2.1 39.6、AA-LCR 68.0、GPQA-Diamond 77.1、AIME-25 90.7。它在 Agent、coding 和 long-context 项目上高于论文列出的 DiffusionGemma、Nemotron-Labs-Diffusion 与 Mercury 2,但这些差距混合了 AR base quality、训练数据、模型规模与生成方法,不能归因于 diffusion augmentation 单一因素。

6.2 Uno-Qwen 对 EAGLE-3 与 DFlash

方法附加参数最大系统吞吐batch 1 吞吐峰值显存(系统/单请求配置)
Uno-Qwen0.35B5,733 tok/s445 tok/s122.2 / 118.0 GiB
EAGLE-30.40B4,944 tok/s284 tok/s130.0 / 129.4 GiB
DFlash1.05B5,351 tok/s370 tok/s130.1 / 129.8 GiB

系统吞吐最优时三者都使用 B=4B=4;单请求最优时 Uno-Qwen 用 Tree (16,32,32)(16,32,32),EAGLE-3 用 Tree (8,32,60)(8,32,60),DFlash 用 B=16B=16。Uno-Qwen 在论文测试的所有 batch size 上形成更好的 throughput Pareto frontier,并以较少附加参数和共享 KV cache 降低峰值显存。

这些峰值显存可以装入 141 GB 规格的单张 H200,但已经接近只适用于高显存数据中心卡的区间。换成 H100 80 GB、消费级 GPU、量化权重或 tensor parallel 后,最大可用 batch 与最佳 sampler 配置都会变化;复现时必须逐表核对硬件拓扑,不能直接外推吞吐比例。

6.3 sampler 消融

自研 Uno 配置平均 TPF最高系统吞吐最高单请求吞吐
Linear B=4B=41.85,190275
Linear B=8B=82.25,034319
Linear B=16B=162.33,445345
Tree (16,32,32)(16,32,32)2.62,665379

更大的 block 和更多候选确实提高 TPF,却不保证提高 tok/s。Linear B=16B=16B=4B=4 接受更多 token,但验证和计算成本使系统吞吐下降约三分之一。这是论文最值得保留的工程结论:接受长度是中间指标,墙钟吞吐才是目标函数。

6.4 RL post-training

作者从 SFT checkpoint 训练数学、代码、工具使用与网页搜索四个 DAPO experts。扩散 adapter 保持冻结,只加速 rollout;数学与代码专家的端到端训练最高加速 40%,工具与搜索的收益较小,因为 tool call 时间占主导。RL 后平均 TPF 从 2.25 降至 2.10,仅下降约 6%,说明 adapter 对 AR 参数更新有一定鲁棒性。

但论文明确表示详细 RL timing 将在下一版提供。目前没有逐任务 wall-clock、集群利用率、rollout 占比与成本分解,40% 应视为初步报告,而不是已充分复现的核心结论。

7. 如何理解“无损”

7.1 它保证什么

在精确算法假设下,拒绝采样 correction 使 Ψ\Psi-Spec 的最终 token 序列服从 verifier 的 AR 分布。draft 不一致只会降低接受率和速度,不会系统性改变目标概率。这与普通 speculative decoding 的分布保持含义相同。

7.2 它不保证什么

  • 不保证同一个随机种子逐 token 复现普通 AR 解码;执行顺序和随机数消费会变化。
  • 不消除有限精度、kernel nondeterminism、top-kk/top-pp 实现差异或数值溢出的工程影响。
  • 不保证每种 prompt、batch、context 和硬件都加速;draft forward、verify forward 与 tree expansion 都有成本。
  • 不意味着多步、关闭 verifier 的 inference-time scaling 仍然保持 AR 分布。
  • 不意味着 Uno 的 benchmark 质量来自扩散路径;最终质量上限首先由 θAR\theta_{\mathrm{AR}} 决定。

所以更准确的产品表述是:Uno 是由扩散 proposal 驱动、由 AR verifier 保真的推测解码系统。

8. 局限与批判性分析

8.1 关键比较没有完全控制变量

Uno、DiffusionGemma、Nemotron-Labs-Diffusion 与 Mercury 2 的参数规模、训练数据、稀疏结构、量化、上下文和硬件不同。Uno 在 benchmark 上领先,最多说明这套完整模型系统在作者口径下更强,不能证明 diffusion augmentation 比所有原生 d-LLM 架构本质上更有能力。方法归因应优先看 Uno-Qwen、EAGLE-3、DFlash 的同 base 对照。

8.2 内部训练数据削弱自研模型的可复现性

自研 AR 权重使用约 23T internal quality tokens,SFT 数据也未完整公开。这不仅影响能力分数,还会影响 diffusion adapter 的 draft acceptance,因为语言模式可预测性与训练分布相关。公开 Uno-Qwen 降低了复现门槛,却不能替代自研 Uno 的完整训练证据。

8.3 训练成本“相对小”不等于便宜

7B 或 14.7B adapter tokens 相对 23T pre-training 确实很少,但仍需要 32–64 张 H200 持续 32–60 小时。若组织只有少量 GPU,adapter 训练成本、Hopper 专用 kernel 和 tree sampler 的 FlashAttention-3 依赖都可能成为采用障碍。

8.4 DCD 的理论动机与最终最佳目标存在张力

附录指出 DCD 基于 deterministic probability-flow trajectory,而采样走 stochastic denoising trajectory,存在 train–test mismatch。更直接的是,消融中 TV-only 胜过 DCD。论文名强调 discrete diffusion,但实用版本越来越像“uniform-noise 初始化的并行 LoRA drafter + speculative correction”。这不是缺点,却提示后续工作应分离哪些收益来自扩散理论,哪些来自目标函数与共享参数结构。

8.5 高并发收益仍需更多真实服务证据

固定 1K/8K 测试避免了输出长度偏差,方法上比混合 benchmark tok/s 更干净;但生产请求的长度分布、prefix cache、continuous batching、tool wait、请求抢占、量化与 tensor parallel 都会改变最优点。论文强调 agentic concurrency,却主要报告合成固定长度吞吐,而不是端到端服务 SLO、P50/P99 latency 或每美元 token。

8.6 相关性争议尚未在论文正文中充分处理

Hugging Face 讨论区有人指出 Uno 与同期更早的 Orthrus 都采用“冻结 AR backbone、增加 diffusion 参数、并行 draft、AR verify”的框架。作者回应称 Uno 保持原 attention 架构并使用 causal diffusion path,而 Orthrus 添加 diffusion attention heads 且使用 bidirectional attention。当前 v1 Related Work 未讨论 Orthrus;在“首个”或框架原创性表述上,后续版本需要更精确定位。方法差异真实存在,但不应把架构实现差异自动等同于框架层面的完全独立。

8.7 论文与公开仓库仍在快速演进

抓取时 arXiv v1 列出 17 位作者,而项目页和当前代码仓库列出 19 位;训练示例中的部分 LoRA 超参数也与 v1 正文表述不同。本文以 arXiv v1 作为论文事实基准,并把仓库视为最新工程实现。复现者应固定 commit、checkpoint 和论文版本,避免把滚动更新的 README 配置与 v1 实验表混用。

9. 应用与工程影响

9.1 在线 LLM serving

Uno 最适合 base model 已经达到质量要求、又不能接受输出分布变化的场景。平台可为低并发交互请求选择 Tree Sampler,为高并发批处理选择 Linear Sampler,并按 context length 与实时 batch 动态切换配置。真正的 scheduler 目标应是 SLO 下的成本,而不是固定 TPF。

9.2 Agent 与长轨迹任务

代码 Agent、搜索 Agent 和工具调用 Agent 会产生长 reasoning trace,并可能为一个用户并行展开多个 trajectory。若模型生成仍是主瓶颈,Uno 的高 batch 加速比只优化 batch 1 更有吸引力;若 API、浏览器或测试运行占据大部分时间,论文中的 tool/search RL 结果说明模型加速会被外部等待稀释。

9.3 RL rollout 基础设施

在 SFT 后训练 diffusion adapter,再冻结它去加速不断更新的 policy rollout,是论文最有潜力的系统贡献。它不要求改写 PPO、GRPO 或 DAPO 目标;但 adapter 与 policy drift 的容忍范围尚无理论保证。生产训练应持续监控 acceptance length、TPF 和 KL drift,并在速度跌破阈值时重训 adapter。

9.4 开放权重模型的后装加速层

Uno-Qwen 表明无需原始 pre-training corpus,也可以给现有模型追加并行 draft 能力。这使 adapter 有机会成为类似量化配置的可分发性能资产。不过 adapter 对 tokenizer、base revision、chat template、sampling 参数和 runtime kernel 高度绑定,发布时需要完整声明兼容矩阵。

10. 相关工作定位

  • Speculative decoding:Leviathan 等人与 Chen 等人奠定了 draft–verify 与残差 correction;Uno 沿用其无损保证,创新点在 diffusion proposal 与共享 base weights。
  • EAGLE-3、DFlash:前者用轻量 AR drafter,后者用条件化 diffusion drafter;两者都维护独立 draft 参数与 cache,是 Uno 最直接的 lossless 系统对照。
  • TiDAR 与 self-speculative decoding:同一模型在 diffusion 与 AR 模式间切换,但修改 base weights 后不能保留原模型分布;Uno 用冻结 AR verifier 避免这一点。
  • DiffusionGemma、Nemotron-Labs-Diffusion、LLaDA 系列:以 diffusion 作为主要生成分布,提供原生并行性,但需要面对质量—速度权衡和高 batch 效率。
  • Medusa 与 multi-token prediction:通过多个 heads 预测未来 token;其并行预测与 Uno 可互补,论文将组合更强 MTP head 与 diffusion adapter 留作未来工作。
  • Quadratic sampling:TiDAR 等方法把 draft 与 verify 合并到一次包含 k2k^2 mask positions 的 forward;可能减少 Uno 当前每轮两次 forward,但需要专用高效 kernel。
  • Orthrus:社区指出其框架目标与 Uno 相近;Uno 作者强调 causal attention、LoRA augmentation 和不改主架构的区别。当前应将两者视为需要直接、同口径比较的邻近路线。

11. 复现建议与结论

复现应分三层推进:

  1. 先在公开 Uno-Qwen3-8B checkpoint 上验证输出分布:对多个前缀比较 AR 与 Ψ\Psi-Spec 的统计量,而不只看单次文本相同与否;
  2. 在固定同一 GPU 拓扑、precision、attention backend、1K/8K 长度和 sampling 参数下,扫描 batch size、B/K/VB/K/V,报告 tok/s、P50/P99、显存与功耗;
  3. 再按公开 OpenThoughts curriculum 重训 adapter,确认 TV loss、block curriculum 和 LoRA rank 消融,并记录完整 commit 与依赖版本。

综合来看,Uno 最重要的贡献不是“扩散模型终于替代 AR”,而是提出了一种更务实的分工:AR 分布负责可信质量,扩散 adapter 负责并行提案,拒绝采样负责把速度优化与质量风险隔离。 论文在同 base 的 Qwen 对照上已经给出有说服力的吞吐与显存证据,公开代码也覆盖核心链路;但 3 倍 headline、跨 d-LLM 能力领先和 40% RL 加速仍需要独立复现与更严格的硬件、数据控制。

如果后续结果能证明 adapter 在多语言、多领域、量化和 continuous batching 下仍稳定接受,并能根据实时负载自动选择 Linear/Tree 工作点,那么 diffusion-augmented LLM 更可能成为 serving stack 的通用加速层,而不是另一类必须替换现有模型的生成范式。

参考资料

  1. Hugging Face, “Daily Papers,” 2026-09-08:https://huggingface.co/papers
  2. Hugging Face, “Unlocking Lossless Speedups in LLMs via Discrete Diffusion”:https://huggingface.co/papers/2609.04010
  3. Sahoo et al., “Unlocking Lossless Speedups in LLMs via Discrete Diffusion,” arXiv:2609.04010:https://arxiv.org/abs/2609.04010
  4. Uno 官方项目页:https://s-sahoo.com/uno/
  5. Uno 官方代码仓库:https://github.com/ifm-ai/uno
  6. Uno Hugging Face 模型集合:https://huggingface.co/collections/s-sahoo/uno
  7. Leviathan et al., “Fast Inference from Transformers via Speculative Decoding”:https://arxiv.org/abs/2211.17192
  8. Chen et al., “Accelerating Large Language Model Decoding with Speculative Sampling”:https://arxiv.org/abs/2302.01318
  9. Hu et al., “LoRA: Low-Rank Adaptation of Large Language Models”:https://arxiv.org/abs/2106.09685
  10. Orthrus, arXiv:2605.12825:https://arxiv.org/abs/2605.12825