Multimodal | | arXiv:2610.39633

Intrinsic Distribution Matching

内在分布匹配

🧭 少步文生图蒸馏 · 矩匹配

Inclusion AI · Westlake University · Tsinghua University · Zhejiang University · Shanghai Innovation Institute

Deyuan Liu, Yansen Han, Kaiwen Zheng, Jun Xie, Jiacheng Liu, Zhenglin Cheng 等

提出 IDM,在 DMD 中显式匹配生成器 EMA 特征的均值与逐坐标标准差,提升少步文生图长文本渲染

📌 工作参考价值

  • 少步生成中的细节保真:在商品图/广告创意生成中,可显式匹配生成器自身 EMA 特征的均值与标准差作为辅助损失,提升提示文本细节(如品牌文字、卖点、活动文案)的还原度。
  • 不依赖外部特征编码器,使用生成器 EMA 特征做分布匹配,减少额外训练依赖和线上资源;适合业务侧快速蒸馏或微调文生图模型。
  • LoRA 参数高效微调即可获得收益,适合在已有大模型基础上低成本迭代 AIGC 素材生成。
  • 可借鉴其分解与消融思路:将分布差拆成高斯矩和非高斯残差,先约束可解释的低阶统计量,比盲目堆损失更可控,也可迁移到其他生成式任务。

动机
少步文生图必须用极少采样步准确还原提示细节,长文本渲染是典型压力测试。DMD 通过分数差逼近期望分布,但学到的监督可能缺少对关键统计量的显式控制。

方法关键点
对 reverse-KL 场做分解,识别出可由高斯矩表达的分量与非高斯残差。IDM 保留完整 DMD 校正,同时用生成器自身 EMA 副本的特征,显式匹配特征均值和逐坐标标准差,不依赖外部特征编码器。支持全参数与 LoRA 两种微调方式,在 Z-Image-Turbo 6B 和 Qwen-Image 20B 上验证。

关键结果数字
在两次网络评估预算下,中文/英文 LongText-Bench 分别达到 94.25% 和 90.20%,为同预算最优;消融显示新增矩损失带来增益,生成器 EMA 特征与外部 DINOv2 编码器在五项指标上表现相当。