Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Why Do DiT Editors Drift? Plug-and-Play Low Frequency Alignment in VAE Latent Space

About

Recent advances in diffusion transformers (DiTs) have enabled promising single-turn image editing capabilities. However, multi-turn editing often leads to progressive semantic drift and quality degradation.In this work, we study this problem from a latent-space frequency perspective by decomposing the editing process into two functional components: VAE and DiT. Through systematic analysis in the VAE latent space, we uncover that the DiT introduces dominant low-frequency drift that accumulates as semantic misalignment across editing rounds, while the VAE contributes comparatively stable reconstruction bias.Based on this insight, we propose VAE-LFA (Low Frequency Alignment), a training-free, plug-and-play method that performs alignment in VAE latent space. VAE-LFA decomposes latent discrepancies across editing rounds via low-pass filtering, and aligns low-frequency statistics to an exponential moving average of previous rounds, effectively suppressing accumulated semantic drift while preserving high-frequency details.Our method requires no retraining, ground-truth priors, or access to diffusion parameters, making it applicable to both white-box and black-box DiT editors. For white-box models, VAE-LFA is seamlessly integrated into the editing pipeline by eliminating redundant VAE round trips; for black-box models, it operates via an off-the-shelf VAE to perform inter-round latent alignment.Extensive experiments demonstrate that VAE-LFA improves semantic consistency and visual fidelity across diverse multi-turn editing scenarios, including both controlled and in-the-wild images.

Xiaoce Wang, Sifan Zhou, Kaifei Wang, Leli Xu, Xuerui Qiu, Tao He, Ming Li• 2026

Related benchmarks

TaskDatasetResultRank
Cycle Image EditingCustom 120-image dataset Cycle Round 10 1.0 (test)
LPIPS (Photograph-Creature)0.3
8
Identity-Preserving Image EditingCustom 120-image dataset No-Op Round 10 1.0 (test)
LPIPS (Photograph vs Creature)0.17
8
Image EditingEdiVal-Agent 1.0 (Round 5)
DINO Clear Object Score (Photograph)0.61
8
Image EditingEdiVal-Agent 1.0 (Round 10)
Clear Object DINO (Photo)0.44
8
Showing 4 of 4 rows

Other info

Follow for update