HybridStitch: Token-Adaptive Model Stitching for Efficient Diffusion Generation
cs.CV, cs.AI
Submitted: 2026-03-08
Updated: 2026-10-14
Code: https://github.com/vipshop/cache-dit
License: http://creativecommons.org/licenses/by/4.0/
The gist: Iterative diffusion generation requires different amounts of model capacity at different token positions.
Terminology
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.
Sources
- SRDiffusion: Accelerate Video Diffusion Inference via Sketching-Rendering Cooperation
- Scaling Rectified Flow Transformers for High-Resolution Image Synthesis
- xDiT: an Inference Engine for Diffusion Transformers (DiTs) with Massive Parallelism
- Distribution Matching Distillation Meets Reinforcement Learning
- FLUX.1 Kontext: Flow Matching for In-Context Image Generation and Editing in Latent Space
- Microsoft COCO: Common Objects in Context
- Decoupled DMD: CFG Augmentation as the Spear, Distribution Matching as the Shield
- SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesis
- FlexCache: Flexible Approximate Cache System for Video Diffusion
- Z-Image: An Efficient Image Generation Foundation Model with Single-Stream Diffusion Transformer
- Qwen-Image Technical Report
- SpargeAttention2: Trainable Sparse Attention via Hybrid Top-k+Top-p Masking and Distillation Fine-Tuning
- SLA2: Sparse-Linear Attention with Learnable Routing and Quantization-aware Training
- SageAttention3: Microscaling FP4 Attention for Inference and An Exploration of 8-Bit Training
Related papers
- Loss Knows Best: Detecting Annotation Errors in Videos via Loss Trajectories
- AnchorWeave: World-Consistent Video Generation with Retrieved Local Spatial Memories
- Benchmarking the Robustness of Foundation Models for Mammography under Domain Shift
- MambaX-Net: Dual-Input Mamba-Enhanced Cross-Attention Network for Longitudinal MRI Segmentation
- TeleOCR: Navigating Document Parsing Across Digital and Camera-Captured Documents
- A Survey on Efficient Vision-Language-Action Models