arXiv · 2411.16720
Importance-Based Token Merging for Efficient Image and Video Generation
Abstract
Token merging can effectively accelerate various vision systems by processing groups of similar tokens only once and sharing the results across them. However, existing token grouping methods are often ad hoc and random, disregarding the actual content of the samples. We show that preserving high-information tokens during merging - those essential for semantic fidelity and structural details - significantly improves sample quality, producing finer details and more coherent, realistic generations. Despite being simple and intuitive, this approach remains underexplored. To do so, we propose an importance-based token merging method that prioritizes the most critical tokens in computational resource allocation, leveraging readily available importance scores, such as those from classifier-free guidance in diffusion models. Experiments show that our approach significantly outperforms baseline methods across multiple applications, including text-to-image synthesis, multi-view image generation, and video generation with various model architectures such as Stable Diffusion, Zero123++, AnimateDiff, or PixArt-$\alpha$.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Haoyu Wu, Jingyi Xu, Hieu Le, Dimitris Samaras. 2024-11-23. Importance-Based Token Merging for Efficient Image and Video Generation. https://arxiv.org/abs/2411.16720
Cite the original work for its findings. Save a collection to share your selection of sources.