跳到正文
arXiv:cs.AI· Desen Sun, Jason Hon, Jintao Zhang, Sihang Liu·· 5 小时前

HybridStitch:面向高效扩散生成的 Token 自适应模型拼接

HybridStitch: Token-Adaptive Model Stitching for Efficient Diffusion Generation

AI 导读

HybridStitch 是一个无需训练的框架,通过 token 级反馈为每个更新选择由哪个模型提供,从而按 token 位置分配模型容量。它在四个主要基准上实现 1.63–1.68 倍图像生成加速和 1.15–1.58 倍 dLLM 吞吐提升,dLLM 解码查询开销估算下降 50–66%,生成质量相当。

正文

View PDF HTML (experimental)

Abstract:Iterative diffusion generation requires different amounts of model capacity at different token positions. Prior model combinations allocate capacity by denoising step or by drafting and verification role. We present HybridStitch, a training-free framework that uses token-level feedback to select the model supplying each adopted update. Continuous image refinement and discrete language commitment operate on a shared generation state, while model-specific KV buffers provide attention context for selected queries. HybridStitch achieves 1.63--1.68$\times$ faster image generation and 1.15--1.58$\times$ diffusion large language model (dLLM) throughput on four main benchmarks. Estimated dLLM decode query cost decreases by 50--66% with comparable generation quality.
Subjects: Computer Vision and Pattern Recognition (cs.CV); Artificial Intelligence (cs.AI)
Cite as: arXiv:2603.07815 [cs.CV]
  (or arXiv:2603.07815v3 [cs.CV] for this version)
  https://doi.org/10.48550/arXiv.2603.07815

arXiv-issued DOI via DataCite

Submission history

From: Desen Sun [view email]
[v1] Sun, 8 Mar 2026 21:34:50 UTC (2,851 KB)
[v2] Tue, 13 Oct 2026 17:09:25 UTC (263 KB)
[v3] Wed, 14 Oct 2026 04:33:47 UTC (263 KB)

来源:arXiv:cs.AI · arxiv.org