MAMBO-G: Magnitude-Aware Mitigation for Boosted Guidance

TL;DR

MAMBO-G is a training-free, guidance magnitude optimization framework that accelerates high-fidelity text-to-image and video generation by stabilizing guidance scales, achieving up to 4x speedup without sacrificing quality.

Contribution

It introduces a novel guidance modulation method that dynamically adjusts guidance scales, significantly reducing computational costs in diffusion-based generation tasks.

Findings

Up to 4x speedup on diffusion models.

Effective acceleration of large-scale video synthesis.

Maintains visual fidelity despite speed improvements.

Abstract

High-fidelity text-to-image and text-to-video generation typically relies on Classifier-Free Guidance (CFG), but achieving optimal results often demands computationally expensive sampling schedules. In this work, we propose MAMBO-G, a training-free acceleration framework that significantly reduces computational cost by dynamically optimizing guidance magnitudes. We observe that standard CFG schedules are inefficient, applying disproportionately large updates in early steps that hinder convergence speed. MAMBO-G mitigates this by modulating the guidance scale based on the update-to-prediction magnitude ratio, effectively stabilizing the trajectory and enabling rapid convergence. This efficiency is particularly vital for resource-intensive tasks like video generation. Our method serves as a universal plug-and-play accelerator, achieving up to 3x speedup on Stable Diffusion v3.5 (SD3.5) and 4x on Lumina. Most notably, MAMBO-G accelerates the 14B-parameter Wan2.1 video model by 2x while preserving visual fidelity, offering a practical solution for efficient large-scale video synthesis. Our implementation follows a mainstream open-source diffusion framework and is plug-and-play with existing pipelines.