Search papers, labs, and topics across Lattice.
The paper introduces SEGA, a training-free method for improving resolution extrapolation in diffusion transformers (DiTs) by dynamically scaling attention across Rotary Position Embedding (RoPE) components based on the latent's spatial-frequency structure. SEGA addresses the trade-off between global structure and fine detail preservation inherent in uniform RoPE scaling approaches. Experiments demonstrate that SEGA outperforms existing training-free methods in high-resolution image synthesis across various target resolutions.
Stop sacrificing either global structure or fine details when extrapolating DiTs to higher resolutions – SEGA dynamically adjusts attention based on spatial frequency, giving you both.
Diffusion transformers (DiTs) have emerged as a dominant architecture for text-to-image generation, yet their performance drops when generating at resolutions beyond their training range. Existing training-free approaches mitigate this by modifying inference-time attention behavior, often through Rotary Position Embeddings (RoPE) extrapolation combined with attention scaling. However, these strategies apply a uniform and content-agnostic scaling across RoPE components with distinct frequency characteristics, inducing a trade-off between preserving global structure and recovering fine detail. We introduce SEGA, a training-free method that dynamically scales attention across RoPE components according to the latent's spatial-frequency structure at each denoising step. This adaptive scaling improves both structural coherence and fine-detail fidelity. Experiments show that SEGA consistently improves high-resolution synthesis across multiple target resolutions, outperforming state-of-the-art training-free baselines.
