一句话定位

Lumina-mGPT 2.0 是一个从零训练(train-from-scratch)的独立解码器式自回归图像生成模型(2B / 7B),不依赖任何预训练权重,用统一的 next-token 范式把 T2I、图像对生成、主体驱动、多轮编辑、可控生成、稠密预测全部塞进一个模型,在 GenEval 上以推理优化策略达到 0.80、DPG 达到 84.30,与 DALL·E 3 / SANA / Lumina-Image 2.0 等扩散 SOTA 持平甚至局部反超。

背景与定位

多模态自回归(AR)范式因其”用 next-token 统一所有模态与任务”的优雅性而复兴:2024 年 chameleon 重新点燃这条路线,lumina-mgpt(v1)作为首个开源工作证明 decoder-only AR 能生成高分辨率、灵活宽高比的图像,质量对标 SDXL。但 v1 有四个硬伤,本作正是逐一拆解:

  1. 受预训练模型束缚——v1 从 Chameleon 7B/34B 初始化,架构、图像/文本 tokenizer 都被锁死,无法换更好的组件,也继承了 Chameleon 的 license 限制,挡住商用。
  2. 多任务相互冲突——v1 对每个下游任务单独 SFT、各存一份 checkpoint,任务目标无法与主 T2I 对齐。
  3. 缺推理优化——AR 要数千步 next-token,推理极慢,v1 没做任何加速。
  4. 质量落后扩散 SOTA——仍不及 Lumina-Image 2.0 / SANA / DALL·E 3。

时间线(据 README News):模型与 T2I 推理代码 2025-04-03 首发;技术报告 2025-07-25 上 arXiv(2507.17801v1,2025-07-23 提交);2025-08-02 放出 image-to-image 推理 + all-in-one(Omni)checkpoint;2025-08-15 华为 MindSpeed MM 原生支持。frontmatter date: 2025-04 指首发月份。

同期路线分两支:一支做”建模混合”(show-o / Transfusion 把 AR 换成扩散做图像;Janus / Janus-Pro 给理解端接 SigLIP 连续编码器,但理解/生成不一致,难做统一生成);另一支做”纯 AR 堆资源”(emu3 保持纯 AR 但生成质量仍偏弱、统一生成潜力未挖)。Lumina-mGPT 2.0 选择纯 AR + 从零训练,用 64 张 A100 跑 4–5 周的”适度算力”复刻 GPT-4o 图像生成的部分能力。

模型架构

Backbone:decoder-only Transformer(纯自回归,next-token prediction),结构沿用 v1 的 decoder-only 形态,但全部参数随机初始化、从零训练,因此可以自由设计架构、自由选 tokenizer、无 license 约束。

架构配置(Table 2):

模型参数词表HiddenIntermediateHeadsKV HeadsLayers
Lumina-mGPT 2.02B171,3852,0488,192323232
Lumina-mGPT 2.07B171,3854,09611,008323232

缩放策略以增大 hidden dimension 为主(层数固定 32,head 数固定 32,full attention 即 KV Heads = Heads,未用 GQA)。

图像 tokenizer:SBER-MoVQGAN(270M)。作者对主流 AR 用 tokenizer 做了系统的重建质量对比(MS-COCO,Table 1):

Tokenizer年份下采样PSNR↑SSIM↑LPIPS↓
VQGAN202116×1618.700.480.17
ViT-VQGAN20228×818.880.600.16
MaskGIT-VQ202216×1618.550.470.19
SBER-MoVQGAN20238×822.770.630.08
LlamaGen-VQ20248×821.910.610.09
Chameleon-VQ(v1 用)202416×1618.630.470.18

SBER-MoVQGAN 重建质量最佳(PSNR/SSIM/LPIPS 全面领先),决定了生成质量上限,故被采用。代价:8×8 下采样导致 image token 序列更长,推理时间/成本上升(这也是后面要做加速的动机)。

文本侧:无预训练文本编码器。不同于 LlamaGen/Parti 那种”预训练 text encoder + MLP 投影”的做法,本作直接用 QwenTokenizer(来自 Qwen2-VL) 把文本编码成离散 token,与图像 token 一起喂进同一个 decoder,做纯粹的 next-token prediction,省掉了加载预训练 text encoder 的步骤。文本 + 图像共用 171,385 的词表。

条件注入与多任务统一(核心创新):利用 AR 的”raster-scan 顺序天然把先生成的区域当上下文”这一性质,把所有 text-image-to-image 任务统一为双面板(dual-panel)图像生成——把参考图/条件图放在上半部分(<upper half>,待生成图放在下半部分(<lower half>,竖直拼成图像网格联合建模。<upper half> / <lower half> 是特殊 token。任务类型靠 system prompt 区分(Table 3 给出 T2I、subject-driven、image-editing、controllable、dense-prediction 五类模板),其中 <Control Task> 指代 canny/depth/pose 等。损失只对 image token 计算,text token 不参与 loss——这是 AR 图像生成的标准做法。这样所有任务都被统一成”text-to-image”,无需额外架构改动或单独 SFT 阶段。

数据

T2I 数据:从 lumina-image-2-0(Lumina-Image 2.0)的数据集中抽取的子集,含真实 + 合成数据,已用 OmniCaptioner 做过精细过滤与 re-captioning。

多任务数据(各任务用不同来源):

  • 主体驱动生成 → Subject200K
  • 图像编辑 → OmniEdit
  • 可控生成 + 稠密预测 → 从 T2I 数据集随机采样 200K 条自建

分阶段数据规模(层级化 hierarchical data,质量随分辨率递增)

  • 256px 阶段:50M 数据
  • 512px 阶段:19M 数据
  • 768px 阶段:8M 数据

随数据集规模递减,数据质量逐级提升(与 Lumina-Image 2.0 的层级化训练理念一致)。美学/安全过滤的具体阈值未单独披露(继承自 Lumina-Image 2.0 的过滤流程 + OmniCaptioner 重标注)。

训练方法

训练目标:纯 next-token prediction(离散 token 的交叉熵)——非 diffusion / 非 flow matching / 非 masked-token,是标准自回归语言建模式的图像生成,loss 只算 image token。

层级化三阶段训练(hierarchical three-phase)

  1. 256px 预训练:learning rate = 2e-4,50M 数据
  2. 512px 微调:lr 降到 2e-5,19M 数据
  3. 768px 微调:lr 2e-5,8M 数据

全局 batch size 通过梯度累积在 512–1024 间动态调整。多任务(subject/edit/controllable/dense)以”原生多任务”方式与 T2I 一起训(所有任务都是 text-to-image),无需额外 SFT 阶段,这是相对 v1”每任务单独 SFT”的关键改进。

模型成长(Model Growth):从 2B 扩到 7B,主要靠增大 hidden dim。观察到模型越大,训练 loss 收敛越快、生成质量(连贯性、细节、对精细 prompt 的保真度)越好(Fig 10 三阶段 loss 曲线显示 7B 全程 loss 低于 2B)。

偏好对齐 / RLHF / DPO / 蒸馏未使用。本作不做 RL 或偏好对齐训练,质量提升完全靠从零训练 + 好 tokenizer + 推理期策略(见下)。

Infra(训练 / 推理工程)

训练算力:64 张 A100,约 4–5 周(论文反复强调这是”适度算力”复刻 GPT-4o 图像能力的一部分)。分布式训练跨 64 A100,batch 用梯度累积调控。更细的并行策略(TP/PP/ZeRO 等)、精度、吞吐未在报告中披露。

推理加速(两条正交策略,可叠加)

  1. 模型量化(TorchAo 后训练量化):权重量化到 4-bit 整数 + 128 元素分组,激活保持 bfloat16;配合 PyTorch 2.0 的 torch.compile(reduce-overhead 模式)做 kernel auto-tuning 与静态图优化,不改模型架构。实测采样时间降 48%、显存降 47%
  2. Speculative Jacobi Decoding(SJD):把确定性 Jacobi 迭代与随机采样结合,引入”基于 draft/target 分布似然比的概率收敛准则”来接受/拒绝 token,实现并行解码且保留采样多样性。实测采样时间再降 72%
    • 工程难点:SJD 需要动态 KV cache(token 可被接受或回滚,序列长度可变),与 torch.compile 要求的静态预分配 KV cache 冲突。作者提出静态 KV cache + 静态因果 attention mask + 指针机制:预分配固定大小缓冲,用指针管理有效序列长度,避免动态扩容;mask 预计算、推理时用指针调整,从而让 SJD 兼容静态编译框架。

实测推理开销(A100,README 数据,768×768)

方法推理时间显存
Lumina-mGPT 2.0(baseline)694s80 GB
+ speculative_jacobi324s79.2 GB
+ speculative_jacobi & quant304s33.8 GB

可见 SJD 砍掉一半多时间、量化把显存从 80GB 压到 33.8GB(可在单卡 40GB 级显卡跑)。局限:即便优化后仍需数分钟级采样,这是所有 AR 生成模型的通病。

评测 benchmark(把效果讲清楚)

T2I —— GenEval / DPG(Table 4) 表示用了采样优化策略(thinking + inference-time scaling):

模型类型参数GenEval OverallDPG Overall
SANA-1.6B扩散1.6B0.6684.80
DALL·E 3扩散0.6783.50
Lumina-Image 2.0扩散2.6B0.7387.20
Emu3AR8.0B0.54
Janus-Pro-7BAR7B0.8084.19
Lumina-mGPT(v1)AR7B0.5679.70
Lumina-mGPT 2.0†AR2B0.6882.05
Lumina-mGPT 2.0†AR7B0.8084.30

7B 版 GenEval 0.80(与 Janus-Pro-7B 持平,居 AR 顶级;在 “Two Obj.” 0.92 / “Color Attri.” 0.72 子项尤其突出),DPG 84.30(论文称”刷新 AR 上限”,且高于 DALL·E 3 的 83.50;但仍略低于 SANA-1.6B 的 84.80 与 Lumina-Image 2.0 的 87.20——并非全面超过扩散 SOTA)。作者指出 GenEval 主要靠 VLM 评文图对齐、不评画质美学;在定性对比(Fig 8)里 Lumina-mGPT 2.0 的真实感/细节/构图明显优于 Janus-Pro 和 v1。

高质量采样消融(Table 7,7B,GenEval)——两个推理期策略逐项加成:

配置Two Obj.CountingPositionColor Attri.Overall
Lumina-mGPT 2.0(朴素单图)0.870.440.440.540.69
+ Thinking Before Generation0.870.490.520.620.73(+4%)
+ Inference-Time Scaling0.920.570.700.720.80(+11%)
  • Thinking Before Generation:先用外部 LLM(GPT-4o)对用户 prompt 做 step-by-step CoT 分析并改写成更清晰、信息更丰富的 prompt,再喂给模型。平均 +4%,position 和 color-attribution 各 +8%。
  • Inference-time Scaling(best-of-N):随机生成一批候选图,用多个 verifier(VQAScore + LAION-AestheticScore + PickScore)打分选最优。从 16 张候选里选,相比单图 +11%。本作是”纯 AR 推理期 scaling”的首次尝试。

可控生成(Table 5,VisualCloze benchmark)——Canny / Depth 条件,与 ControlNet / OmniControl / OneDiffusion / OmniGen 比(专家模型灰底)。Lumina-mGPT 2.0 作为单一通才:Canny 条件 F1 = 0.49(非专家里最高)、FID = 30.89;Depth 条件 RMSE = 17.42、FID = 36.52,在可控性与质量上达到非专家方法的第一/第二,且文图一致性(CLIP-Score Canny 34.44 / Depth 34.03)保持高位。相比 v1 的 Canny FID 85.03、Depth FID 61.44 是巨大跃升。

主体驱动生成(Table 6,1000 样本):DINOv2 = 76.60(最高,主体身份保持最好)、CLIP-I = 87.37、CLIP-T = 33.90,超过 OminiControl / OneDiffusion / OmniGen 等。

加速消融(Fig 11):量化降时 48% / 降显存 47%;叠加 SJD 再降时 72%,画质保持。

模型成长消融:7B 在 GenEval/DPG 全面优于 2B;7B 训练 loss 收敛更快、视觉更稳定细节更丰富。

创新点与影响

核心贡献

  1. 首个从零训练、不依赖任何预训练权重的独立 decoder-only AR 图像模型——摆脱 Chameleon 等预训练模型的架构 + tokenizer + license 三重束缚,可自由选 SBER-MoVQGAN、自由缩放到 2B/7B、HF 上以 apache-2.0 开源。
  2. 用 dual-panel raster-scan + system prompt 把 6 类生成任务原生统一进一个纯 AR 模型,所有任务即”text-to-image”,无需额外模块或单独 SFT,证明纯 AR 架构在统一生成上的天然优势(这是相对 Janus-Pro 解耦编码、Emu3 仅做 T2I 的关键差异化)。
  3. 首次在纯 AR 上做推理期 scaling(best-of-N + 多 verifier)与 thinking-before-generation(GPT-4o 改写 prompt),把 GenEval 从 0.69 推到 0.80,证明”牺牲推理效率可显著换质量”。
  4. 工程上让 SJD 与量化/torch.compile 共存(静态 KV cache + 静态 mask + 指针),把 AR 的慢推理从 694s 压到 304s、显存压到 33.8GB。

影响:作为上海 AI Lab Alpha-VLLM 系列(lumina-mgpt → Lumina-Image 2.0 → 本作)的纯 AR 旗舰,给”纯自回归能否追平扩散 SOTA”提供了肯定的开源答案,并被华为 MindSpeed MM 原生支持(2025-08)。Omni 版(all-in-one checkpoint)补齐 image-to-image 全任务推理。

已知局限

  • 即使优化后采样仍需数分钟级,用户体验受限(所有 AR 生成的通病)。
  • “thinking” 依赖外部 LLM(GPT-4o),模型自身不能自主思考;未来计划内化。
  • 当前只做多模态生成,不含多模态理解;未来计划补理解能力。
  • 8×8 tokenizer 带来长序列,是推理成本的根因。
  • 未做任何 RLHF/DPO 偏好对齐,质量靠推理期策略兜底。

原始链接

一手源存档(sources/)

  • arxiv-2507.17801.pdf (arXiv 2507.17801v1,2025-07-23 提交,技术报告) (arXiv 原文 PDF,不入 git)
  • readme.md (GitHub README 清洗版,含 A100 推理耗时表 + News 时间线)
  • github-readme.md (GitHub 页面原始抓取,含导航噪声,内容为上者超集)
  • hf-modelcard.md (HF model card,license: apache-2.0)