Search papers, labs, and topics across Lattice.
The paper introduces Elastic Latent Interface Transformer (ELIT), a DiT-compatible mechanism that decouples input image size from computational cost by inserting a learnable, variable-length latent token sequence. ELIT uses lightweight read/write cross-attention layers to move information between spatial tokens and latents, prioritizing important input regions and learning an importance-ordered representation. By dynamically adjusting the number of latents at inference, ELIT enables flexible latency-quality trade-offs, achieving significant gains in FID and FDD scores across various datasets and architectures.
Achieve up to 40% better FID/FDD scores in diffusion transformers by decoupling image resolution from compute via a simple "elastic latent interface" that lets you dial the latency-quality tradeoff at inference time.
Diffusion transformers (DiTs) achieve high generative quality but lock FLOPs to image resolution, limiting principled latency-quality trade-offs, and allocate computation uniformly across input spatial tokens, wasting resource allocation to unimportant regions. We introduce Elastic Latent Interface Transformer (ELIT), a drop-in, DiT-compatible mechanism that decouples input image size from compute. Our approach inserts a latent interface, a learnable variable-length token sequence on which standard transformer blocks can operate. Lightweight Read and Write cross-attention layers move information between spatial tokens and latents and prioritize important input regions. By training with random dropping of tail latents, ELIT learns to produce importance-ordered representations with earlier latents capturing global structure while later ones contain information to refine details. At inference, the number of latents can be dynamically adjusted to match compute constraints. ELIT is deliberately minimal, adding two cross-attention layers while leaving the rectified flow objective and the DiT stack unchanged. Across datasets and architectures (DiT, U-ViT, HDiT, MM-DiT), ELIT delivers consistent gains. On ImageNet-1K 512px, ELIT delivers an average gain of $35.3\%$ and $39.6\%$ in FID and FDD scores. Project page: https://snap-research.github.io/elit/
