觉醒空间智能:面向统一多模态理解与生成的 JoyAI-Image Awaking Spatial Intelligence in Unified Multimodal Understanding and Generation
京东提出统一多模态模型 JoyAI-Image,以空间智能为枢轴打通理解、生成与编辑。
前置知识
多模态大语言模型(MLLM)
在 LLM 基础上接入视觉编码器(通常为 ViT),将图像特征投影到语言 token 空间,从而同时理解图像与文本并能进行多轮对话的模型。本文以 Qwen3-VL-8B-Instruct 为底座,既作为理解引擎也作为生成条件源。
JoyAI-Image 主张用一个增强后的 MLLM 充当'认知大脑',把语义、空间先验统一注入到下游扩散过程,是它和'理解/生成松耦合'设计的本质区别。
Multimodal Diffusion Transformer(MMDiT)
基于 DiT 的双流扩散 Transformer,对文本条件与图像潜变量分别建模后做跨模态融合。JoyAI-Image 用 16B 参数、40 层、4096 隐藏维、32 头 MMDiT,Wan-2.1 VAE 供 16 维潜码,MRoPE 替换 MSRoPE。
MMDiT 是把 MLLM 输出的语义-空间条件变成像素的'渲染器',其容量与位置编码直接决定空间可控生成/编辑的上限。
Flow Matching 与 Flow-GRPO
Flow Matching 把扩散视为噪声到数据的常速流场估计;Flow-GRPO 把 GRPO 群体相对优势推广到流匹配设定,对同 prompt 的一组采样按奖励归一化后再做裁剪优化。
JoyAI-Image 的 T2I 与编辑都基于 flow matching,RL 阶段靠 Flow-GRPO 与 DiffusionNFT 在偏好上继续打磨,是它在长文本与多视角生成上表现突出的关键。
空间智能(Spatial Intelligence)
在三维物理世界中关于物体大小、距离、相对位置、相机视角、跨视图一致性等的几何-语义理解与推理能力。论文把它分解为 SM(空间测量)、SR(空间关系)、CP(相机感知)、MC(多视图一致性)和 SAR(场景感知推理)五类。
这是本文的核心叙事主线:理解、生成、编辑不再被视作三个独立能力,而通过空间智能这一'公共语言'实现双向耦合。
研究动机
当前'统一多模态'路线(如 Qwen-Image、BAGEL、OmniGen2 等)虽然把图像理解、文生图、指令编辑塞进同一个模型,但两个痛点始终没有解决。第一,模块之间是松耦合的:MLLM 的语义/空间理解并没有真正被作为可控信号注入扩散过程,扩散过程也鲜少把视角变化、几何重构等反馈回流给 MLLM,形成'理解→生成'的单向管道。第二,这些模型对真实三维世界缺乏几何级把握:在精细空间理解基准上,主流 8B 级 MLLM 在 VSI、AllAngles、3DSR、BLINK 等指标上落后 Gemini-2.5-Pro 5–15 分,对多视角、相机运动、物体前后关系的判断常常出现'语义看着对、几何不对'的错误;编辑侧则在相机控制、视角变换、目标对象局部操作等任务上,要么把不该动的背景一起改掉,要么无法遵循 yaw/pitch/zoom 这种显式几何指令。
本文的目标是构建一个真正'以空间智能为枢轴'的统一基础模型 JoyAI-Image,在三大类任务上同时做到 SOTA 或具有竞争力:在 9 项空间/通用基准上让 8B 级 MLLM 平均分从 59.1 提到 64.4、追平 Gemini-2.5-Pro;在文本到图像上把 LongText-Bench EN/ZH 双双推到 0.963(超过 Qwen-Image 0.943/0.946 和 GPT Image 1 的 0.956/0.619);在编辑上把 SpatialEdit-Bench 的对象级 Overall 从 LongCat 的 0.439 提到 0.649、相机级 Overall Error 从 0.743 降到 0.429;并把'空间编辑+视觉推理'做成可外溢的'Thinking with Novel Views'范式,让空间推理在 4 个不同体量的 Reasoner 上普遍提升 1.7–4.4 个百分点。
与已有工作不同的是,已有工作要么把空间能力当作 MLLM 的孤立增强(VST、SpaceR),要么只在生成侧引入 3D 一致性(ReconFusion、SV3D)。本文抓住的空白是:把'空间数据引擎—空间增强 MLLM—空间条件化 MMDiT—空间编辑数据引擎—空间推理回路'做成一条端到端闭环,并通过显式 Blender 3D 数据和双分支空间编辑引擎,绕开'真实数据中难以配对的视角变化'这一根本瓶颈。
核心方法
直觉上,可以把 JoyAI-Image 看作一个'左脑-右脑'架构:左脑是经过空间数据专门 SFT 的 Qwen3-VL-8B MLLM,负责读懂图像、解析指令、规划相机运动;右脑是 16B 的双流 MMDiT 渲染器,根据左脑给出的语义-空间条件(最后一层 hidden state)和 VAE 潜码,去噪出像素。两者之间通过一个共享的多模态接口串接,使得理解、生成、编辑共用同一套'感知→意图→动作'的编码。具体到实现,它包含 5 个相互咬合的环节:(1) OpenSpatial 3D 框中心的数据引擎把网络视频和室内扫描变成 3M 条带 3D OBB 的空间 QA;(2) 在 Qwen3-VL-8B 上做空间专项 SFT,配合 KL 蒸馏 + FlashAttention 变长打包,保留通用能力同时新增 5.3 分空间平均分;(3) 16B MMDiT 配合 Wan-2.1 VAE、MRoPE,按 208P→512P→1024P 渐进式预训练,再经过 Continue Training + SFT + Flow-GRPO 三段式微调;(4) 空间编辑数据引擎用 Blender 双分支(Static-Camera 物体变换 + Dynamic-Camera 视角变换)合成具有显式 yaw/pitch/distance 标签的成对样本,把视角控制变成可学习的语言模板;(5) TwNV 范式把 JoyAI-Image-Edit 当作 Synthesizer,配合 MLLM Planner 预测 6-DOF 相机动作、MLLM Reasoner 联合原图与合成图推理,把'生成'反哺'理解'。
与 BAGEL/OmniGen2/Seedream 这类'通用统一模型'最本质的区别在于:本文不是把空间能力当作 SFT 数据里的一类普通指令,而是把它做成贯穿数据、训练、推理三层的'耦合变量'。在数据层,用 3D OBB 作为几何锚点把 2D 视频升格成 3D 空间 QA,并通过 Blender 合成带显式相机标签的视角对;在模型层,把空间增强后的 MLLM 隐藏态作为 MMDiT 的核心条件信号,并在 MMDiT 中把 MSRoPE 替换为 MRoPE 以更好对齐结构化空间条件;在推理层,让生成结果(新视角图像)回流进 MLLM 做空间推理,构造出'理解 ⇄ 生成'的双向回路。这套设计的本质是把'空间一致性'从后处理约束提升为一等公民,因此能实现'用图像编辑的精度去增强空间推理,再用空间推理的反馈去调优编辑'。
方法步骤详情
训练流程分 5 步。第一步,OpenSpatial 数据合成:从 ScanNet、Matterport3D、ARKitScenes、ScanNet++、Hypersim 等高精度室内扫描及网络视频出发,将 3D 定向边界框(OBB)作为统一几何锚点,通过 2D mask + 深度图把 OBB 反投影到像素平面,再用多视图循环一致性约束剔除不一致的 3D 框;然后产出两类 QA——单视图 QA(物体大小、距离、深度、姿态等)和多视图 QA(跨帧同一物体的位置/朝向推理),覆盖 SM/SR/CP/MC/SAR 五大类共 19 个子任务,最终得到 3M 条 OpenSpatial 数据。第二步,MLLM 空间专项 SFT:以 Qwen3-VL-8B-Instruct 为底座,配合 1.7M 通用 VQA、1.4M 数学、1.0M 文档图表、907.8K 语言、610.1K 多任务、463.7K OCR、3.3M OpenSpatial、49.4K VST 多视图子集等共 11.3M 样本,用 FSDP2 + 动态序列打包 + BF16 训练,ViT 用 5e-6、LLM 用 5e-5 的解耦学习率,并只在通用数据上加 $\lambda=10$ 的 KL 蒸馏以防灾难性遗忘。第三步,MMDiT 预训练:用 WanVAE 把图像压到潜码 $z_1$,采样噪声 $z_0$ 与 $z_t=(1-t)z_0+tz_1$,训练网络用 flow matching 目标预测 $z_1-z_0$;分辨率从 208P→512P→1024P 渐进训练,并在 1024P 阶段混入多视图数据;之后是 Continue Training(缩窄数据分布、强化高美感样本)和 SFT(聚焦双语长文本渲染 + 多视角一致性)。第四步,强化学习与空间编辑对齐:T2I 端用 Flow-GRPO,以美学奖励 + 文图对齐奖励做群体相对优势优化;编辑端用 DiffusionNFT(结合 HPSv3 自然度奖励与 Gemini-3-Flash 指令跟随奖励,引入指令跟随优先级混合策略防 reward hacking),并配合 Blender 双分支引擎生成的 Static-Camera 与 Dynamic-Camera 成对样本,让模型学会在自然语言模板('Camera rotation: Yaw X°, Pitch Y°')下控制相机。第五步,TwNV 推理范式:测试时用 MLLM Planner 预测 6-DOF 相机动作,JoyAI-Image-Edit 作为 Synthesizer 生成新视角,再交给 MLLM Reasoner 在原图+新视图上联合推理。
技术新颖性
技术新颖性体现在四个交叉点:(a) OpenSpatial 用 3D OBB 作为统一几何锚点的数据范式,把'网络视频→3D QA'的合成成本压到工业可用规模,并把多视图循环一致性写进几何校验;(b) 把 MLLM 中间层 hidden state 直接喂给 MMDiT 作为条件信号,相比以往'文本重写后再编码'的方案保留了更丰富的空间语义;(c) MR→M RoPE 的替换和 Qwen-Image 的位置编码对齐,让 MMDiT 显式感知相机 yaw/pitch/zoom 而非靠 prompt 隐式学习;(d) '空间编辑'被建模成由 Blender 双分支驱动的'显式几何指令'任务,并首次系统地把 DiffusionNFT + HPSv3 引入编辑 RL,使自然度与指令遵循在 PPO 类目标下共同优化。
实验结果
空间理解方面(Table 3),JoyAI-Image-Und 在 9 项空间基准上平均 64.4 分,比基座 Qwen3-VL-8B-Instruct(59.1)提升 5.3 分、与闭源 Gemini-2.5-Pro(64.4)持平、显著领先 GPT-4o(57.7)、InternVL3-8B(56.2)、MiMo-VL-7B(58.2)和 Qwen3-VL-4B-Instruct(58.8);其中 AllAnglesBench 61.0(+11.5)、3DSR_C 60.5(+7.7)、MMSI 35.8(+7.7)、ERQA 45.0(+4.9)、VSI 60.1(+4.5)涨幅最大,同时在通用基准 MMB_CN 83.7、MathVista 74.4、MMStar 71.3、OCRB 87.9 上保持甚至略超基线(OCRB -2.4 是唯一明显回落点)。文本到图像方面(Table 7/8/9),LongText-Bench-EN 与 LongText-Bench-ZH 双双 0.963,刷新 SOTA,超过 Qwen-Image(0.943/0.946)、Z-Image(0.935/0.936)、GPT Image 1(0.956/0.619);CVTG-2K 上 Word Accuracy 0.8739(第一)、NED 0.9369;OneIG-EN 0.542、OneIG-ZH 0.521、DPG 88.05;T2I-CoReBench 上 Composition Mean 94.2 拿下第一(超过 Qwen-Image 83.7、GPT Image 1 79.8),Reasoning Mean 55.9 排名第二,Overall 68.7。图像编辑方面(Table 10/11/13),GEdit-Bench-EN w/PE 取得 G_SC 8.806、G_PQ 8.273、G_O 8.290,超过 FireRed-Image-Edit(8.363/8.245/7.943)和 Qwen-Image-Edit-2511(8.297/8.202/7.877);GEdit-Bench-CN w/PE 取得 G_SC 8.861、G_PQ 8.119、G_O 8.208;ImgEdit-Bench w/PE Overall 4.57 与 FireRed(4.56)并列第一,Extract 子项 4.32 表现突出;SpatialEdit-Bench 上 Moving 0.652、Rotation 0.646、Viewpoint Error 0.290、Framing Error 0.568,Object Overall 0.649(vs LongCat 0.439、Qwen-Image-Edit 0.421、Nano-Banana-Pro 0.260)、Camera Overall Error 0.429(vs LongCat 0.743、视频世界模型 ReCamMaster 0.738)。人类评估(Figure 12)显示对 Qwen-Image-Edit-2511,JoyAI-Image-Edit 在 Semantic Following(29.5% vs 19.0%)、Consistency(35.9% vs 31.7%)、Overall(45.3% vs 36.1%)上均胜出,对 Flux.2[DEV] 在 Consistency(56.3% vs 18.5%)、Overall(60.8% vs 23.2%)上大幅领先。TwNV 应用(Table 14)以 JoyAI-Image-Edit 为 Synthesizer 时 GPT-5 Overall 从 68.8% 提升到 71.7%(+2.9),多对象关系子任务 60.5%→66.2%(+5.7pp),并在 4 个不同 Reasoner 上稳定带来 +1.7 到 +4.4pp 的提升,其中 Qwen3-VL-32B 相对增益 7.8% 最大,证明该范式对算力受限的 MLLM 帮助更显著。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| 空间理解平均(9 项) | Accuracy | 64.4 | 59.1 (Qwen3-VL-8B) | +5.3 |
| VSI-Bench | Accuracy | 60.1 | 55.6 (Qwen3-VL-8B) | +4.5 |
| AllAnglesBench | Accuracy | 61.0 | 49.5 (Qwen3-VL-8B) | +11.5 |
| 3DSR_C | Accuracy | 60.5 | 52.8 (Qwen3-VL-8B) | +7.7 |
| MMSI | Accuracy | 35.8 | 28.1 (Qwen3-VL-8B) | +7.7 |
| RealWorldQA | Accuracy | 74.0 | 70.7 (Qwen3-VL-8B) | +3.3 |
| LongText-Bench EN | Accuracy | 0.963 | 0.943 (Qwen-Image) | +0.020 |
| LongText-Bench ZH | Accuracy | 0.963 | 0.946 (Qwen-Image) | +0.017 |
| CVTG-2K | Word Accuracy | 0.8739 | 0.8671 (Z-Image) | +0.0068 |
| T2I-CoReBench Composition | Mean | 94.2 | 83.7 (Qwen-Image) | +10.5 |
| GEdit-Bench-EN Overall w/PE | G_O | 8.290 | 7.943 (FireRed-Image-Edit) | +0.347 |
| ImgEdit-Bench w/PE | Overall | 4.57 | 4.56 (FireRed-Image-Edit) | +0.01(并列 SOTA) |
| SpatialEdit-Bench Object Overall | Score↑ | 0.649 | 0.439 (LongCat-Image-Edit) | +0.210 |
| SpatialEdit-Bench Camera Overall Error | Error↓ | 0.429 | 0.743 (LongCat-Image-Edit) | -0.314 |
| TwNV (GPT-5 Reasoner) | Overall Accuracy | 71.7% | 68.8% (无 NV) | +2.9pp(Multi-Obj +5.7pp) |
局限与改进
作者承认的局限:(1) OCRB 从 90.3 掉到 87.9,说明空间 SFT 损伤了少量通用 OCR 能力,可能因为长几何推理占据了 LLM 表征容量;(2) T2I-CoReBench 的 Reasoning Mean 55.9 虽排名第二但绝对值不算高,BR(binding reasoning 18.7)和 RR(reasoning routing 17.8)仍是明显短板;(3) 人类评估中 JoyAI-Image-Edit 在 Naturalness 上对 Nano-Banana-2 是 24.7% vs 48.1%,表明生成纹理、肤色、材质真实感方面仍有差距;(4) 空间编辑的 SpatialEdit-Bench 虽然大幅领先,但 Framing Error 仍有 0.568 表明对象裁切与画框一致性仍不够稳。作者未明确指出的隐含局限:(5) TwNV 推理需要 3 次 MLLM 调用 + 1 次图像合成,延迟和成本都显著高于单次 CoT 推理,实际落地需要工程优化;(6) OpenSpatial 数据虽然规模 3M,但来源以室内扫描为主,对室外/街景/自然场景的覆盖相对有限,从 Table 3 的 RealWorldQA 提升幅度 (+3.3) 明显小于室内/合成基准可以印证。
独立分析的弱点
第一,空间编辑虽然 SpatialEdit-Bench 总体领先,但 Framing Error 仍有 0.568,说明在'局部对象前后裁切 + 画幅保持'上易过/欠裁;改进方向是在 Spatial Edit Data Engine 中加入'对象级裁切监督'与'画框掩码一致损失',并在 RL 阶段把 framing 一致性作为单独奖励项。第二,OCRB 87.9 vs 基座 90.3 掉 2.4 分,说明 KL 蒸馏项 $\mathcal{L}=\mathcal{L}_{SFT}+\lambda\mathcal{L}_{KL}$($\lambda=10$)的'通用数据子集'可能未覆盖到 OCR-heavy 样本;改进方向是分层设定 $\lambda$——对 OCR/文档类数据用 $\lambda=20$,对空间数据继续用 $\lambda=0$。第三,TwNV 的 Synthesizer 在 Orientation 子任务上仅 65.3(baseline 63.6、Qwen-Image-Edit 61.8),相对位置/方向类合成仍不够稳,原因是 Blender 训练数据以单对象居中为主,对多对象前后遮挡的特殊视角覆盖不足;改进方向是扩充 Dynamic-Camera 分支中的多对象场景,并引入 Explicit Occlusion 数据。第四,T2I-CoReBench 的 BR 18.7 和 RR 17.8 表明绑定与路由推理弱;改进方向是把空间 SFT 阶段增加计数-绑定-路径规划类合成 QA。第五,RealWorldQA 仅 +3.3 而 3DSR_C +7.7,说明室外街景空间理解是数据-能力 gap;改进方向是 OpenSpatial 引入 nuScenes、Waymo 这类自动驾驶数据。
未来方向
作者明确指出的方向:把 JoyAI-Image 进一步接入 VLA(vision-language-action)与 world model,作为可被操控的视觉基础模块——本工作已通过 TwNV 与'单图→多视角→VGGT 重建'实验给出验证雏形。我们认为还可延伸:(1) 把 Synthesizer 升级为可微分的几何-渲染一体化模块,让 Reasoner 拿到的不只是像素还有深度/法线,从而支持更精细的 6-DOF 推理;(2) 把 TwNV 范式从图像编辑扩展到视频生成/编辑,让时间维度上的一致性也通过'视角推理'来强化;(3) OpenSpatial 的 3D 框中心范式可与神经辐射场、3D Gaussian Splatting 协同,训练一个可查询的'空间记忆';(4) 当前 MMDiT 16B 仍需要约百卡级算力训练,可探索'小 MLLM + 中等 MMDiT'的稀疏化方案以降低部署成本。
复现评估
代码已开源(https://github.com/jd-opensource/JoyAI-Image),论文给出了 Table 1-2 的超参数(MMDiT 40 层/4096 隐藏维/32 头/MRoPE、LR 5e-5、ViT LR 5e-6、FSDP2、Batch 128、BF16、$\lambda=10$)以及 Table 4 的多阶段过滤阈值、Table 6 的多视图字段定义、Table 2 的 SFT 详细设置,便于复现。但完整复现仍存在门槛:(1) 11.3M 训练语料中 OpenSpatial-3M 是内部数据,需自行按 3D OBB + 多视图循环一致性流程重建;多视角 1M 数据需要 Blender 4.5 渲染管线;(2) MLLM 8B 全参 SFT + 16B MMDiT 从头预训练在 1024P 上的算力需求,按类似 Qwen-Image 量级估算需要 256-512 张 H100/A100 训练数周;(3) 长文本渲染的 HTML/CSS 渲染管线、多视角标注的 Gemini-3-Flash 调用都需要付费 API;(4) RL 阶段 Flow-GRPO 与 DiffusionNFT 的工程实现细节在文中描述较粗。整体而言:能复现整体管线、但要达到论文指标需投入显著算力与数据建设,更适合工业团队从其'数据引擎+训练策略'借鉴,而非直接重训。
论文图表