arXiv cs.AI· Desen Sun, Jason Hon, Jintao Zhang, Sihang Liu·· 3 天前AI 评分22
HybridStitch:面向高效扩散生成的 Token 自适应模型拼接
HybridStitch: Token-Adaptive Model Stitching for Efficient Diffusion Generation
AI 导读
HybridStitch 是一个免训练框架,通过 token 级反馈为每个更新选择提供模型的来源,在四个主要基准上实现图像生成提速 1.63–1.68 倍、扩散大语言模型(dLLM)吞吐量提升 1.15–1.58 倍。该方法让连续图像精修与离散语言承诺共享同一生成状态,并以模型专属 KV 缓冲为选定查询提供注意力上下文。dLLM 解码查询成本估计降低 50–66%,生成质量相当。
正文 · 原文
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