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

Early Estimation of Language to Latent Alignment in Diffusion Models

About

Conditional diffusion models frequently suffer from language-image misalignments. Due to the ambiguity of intermediate noise corrupted latents, assessing prompt adherence currently requires completing the entire sampling trajectory. This late-stage evaluation incurs even higher computational costs during test-time scaling strategies, such as Best-of-N (BoN) sampling, as all misaligned trajectories must finish generation before being discarded. To tackle this, we propose NoisyCLIP, a noise-aware twin-tower model that enables early language-to-latent alignment estimation. By learning a vision encoder on noise-corrupted latents, we allow the model to "see" through the ambiguity of intermediate diffusion steps. To facilitate this training, we investigate noise-data augmentation sampling strategies and introduce two new benchmark datasets: Noisy-Conceptual-Captions and Noisy-GenAI-Bench. When applied as an early-stopping criterion for BoN, NoisyCLIP at half cost matches or beats frozen CLIP at full cost. Ultimately, this transforms alignment assessment from an expensive final check into a continuous monitoring tool, drastically reducing compute costs without sacrificing semantic fidelity.

Vasco Ramos, Regev Cohen, Idan Szpektor, Joao Magalhaes• 2025

Related benchmarks

TaskDatasetResultRank
Best-of-N SelectionNoisy-Concept-Captions Denoised Latent, iterations 21-30
VQAScore71.8
4
Factual ConsistencyNoisy-Concept-Captions Denoised Latent, iterations 21-30
R@150.6
4
Image-text alignmentGenAI-Bench Basic
Alignment Score26.2
3
Image-text alignmentGenAI-Bench Advanced
Alignment Score0.25
3
Best-of-N SelectionNoisy-Concept-Captions Final Image iteration 50--
2
Showing 5 of 5 rows

Other info

Follow for update