[CVPR 2026] ELT: Elastic Looped Transformers — Slashing Generative Parameters by 4x via Recurrent Refinement

ELT: Elastic Looped Transformers for Visual Generation

2026-04-01
Sahil Goyal, Swayam Agrawal, Gautham Govind Anil, Prateek Jain, Sujoy Paul, Aditya Kusupati
Summary
Problem
Method
Results
Takeaways

The paper introduces Elastic Looped Transformers (ELT), a parameter-efficient recurrent architecture for visual generation that reuses transformer weights across multiple iterations. By employing a novel Intra-Loop Self Distillation (ILSD) training strategy, ELT achieves state-of-the-art results including an FID of 2.0 on ImageNet and FVD of 72.8 on UCF-101 while using 4x fewer parameters than standard models like MaskGIT and MAGVIT.

TL;DR

Elastic Looped Transformers (ELT) redefine how we build visual generative models. Instead of stacking hundreds of unique layers, ELT reuses a small set of high-quality transformer blocks iteratively. By using a new training trick called Intra-Loop Self Distillation (ILSD), a single ELT model can scale its quality and speed on-the-fly, achieving SOTA results (FID 2.0) with 75% fewer parameters and up to 3.5x higher throughput.

The Problem: The "Memory Wall" and Massive Models

The current "SOTA" in image and video generation (like DiT or MaskGIT) is synonymous with "Bigger is Better." To increase capacity, we stack more layers. This approach hits two walls:

  1. Memory Wall: Loading billions of unique weights from VRAM to the processor is a massive bottleneck.
  2. Inflexibility: Once a model is trained at a specific depth, you can't easily "shorten" it for a mobile device without a massive drop in quality.

Looped/Recurrent transformers are the logical solution—reuse weights to save memory—but historically, they have been "black boxes." Intermediate loops often produce garbage until the very last iteration, making them useless for elastic, real-time scaling.

Methodology: Depth through Recurrence & ILSD

The core "Aha!" moment of ELT is treating the recurrent loop not just as a deep stack, but as a progressive refiner.

1. Weight-Shared Looping

Instead of unique layers, ELT uses a block of layers applied times. This decouples the physical model size (determined by ) from the computational depth (determined by ).

2. Intra-Loop Self Distillation (ILSD)

ELT solves the "unstable intermediate state" problem by forcing the model to be useful at every loop. During training, it uses a Dual-Path system:

  • Teacher Path: Runs the full loops to get the best possible prediction.
  • Student Path: Exits early at a random loop .
  • The Loss: The student is trained to predict the ground truth and mimic the teacher’s mature representation.

ELT Framework Figure 1: The ELT Training Framework utilizing shared parameters and ILSD to enforce meaningful intermediate states.

Experiments: Superior Efficiency

The results are striking. ELT doesn't just "save parameters"—it often outperforms the massive baselines it replaces under similar compute budgets.

Performance Benchmarks

In class-conditional ImageNet 256x256 generation:

  • MaskGIT-XL (Baseline): 446M params FID 2.0.
  • ELT-XL (8 layers x 4 loops): 111M params FID 2.0.

By keeping the "working set" of weights small, the model can stay mostly on-chip, leading to a 3.5x throughput increase on hardware like TPUs because it avoids constant memory swapping.

Comparison of Visual Trajectories Figure 2: Latent trajectories of Standard vs. Elastic Looped Transformers. Standard models only "converge" at the end, while ELT refines the image progressively.

Any-Time Inference: The Real-World Superpower

Because ILSD makes every loop count, ELT enables Any-Time Inference. A single model weights file can be deployed across:

  • Mobile/Edge: Run 2 loops for instant, decent-quality previews.
  • Cloud/Studio: Run 8 loops for high-fidelity, production-grade assets.

As shown in the paper's Pareto front analysis, you can traverse the quality/compute curve at runtime without ever retraining.

Pareto Front Figure 3: Quality (FID) vs. Compute (GFLOPs). ELT stays on the optimal Pareto front, allowing dynamic scaling.

Depth Insight: Scaling Laws for Looping

The paper reveals a crucial nuance: you cannot loop a model too much if the base block is too small. A configuration (one layer looped 32 times) fails because a single layer lacks the representational capacity. The authors found that a "sweet spot" typically involves a block of 8-16 layers iterated 2-4 times.

Conclusion & Future Work

ELT is a paradigm shift for efficient generative AI. It proves that weight-tying is not a compromise—it’s an optimization.

Key Takeaways for Practitioners:

  1. Parameter Efficiency: You can reduce your model footprint by 75% without losing generation quality.
  2. Throughput: Recurrence is a hardware-friendly pattern that minimizes memory latency.
  3. Elasticity: One model, infinite deployment tiers.

Future applications may include combining ELT with One-Step consistency models, where the number of loops becomes the only lever for users to dial in their desired "render quality" vs. "speed" in real-time.

Find Similar Papers

Try Our Examples

  • Which recent papers explore "test-time compute" scaling in Transformers, and how does ELT's recurrence compare to methods like "Chain-of-Thought" or "Latent Reasoning"?
  • Trace the evolution of Universal Transformers and Deep Equilibrium Models (DEQs) in generative tasks—how does Intra-Loop Self Distillation specifically solve the convergence issues found in those earlier works?
  • Are there existing studies applying weight-sharing or looped architectures to large-scale Video-Diffusion-Transformers (ViDiT), and can ELT's any-time inference be adapted for temporal consistency?
Contents
[CVPR 2026] ELT: Elastic Looped Transformers — Slashing Generative Parameters by 4x via Recurrent Refinement
1. TL;DR
2. The Problem: The "Memory Wall" and Massive Models
3. Methodology: Depth through Recurrence & ILSD
3.1. 1. Weight-Shared Looping
3.2. 2. Intra-Loop Self Distillation (ILSD)
4. Experiments: Superior Efficiency
4.1. Performance Benchmarks
5. Any-Time Inference: The Real-World Superpower
6. Depth Insight: Scaling Laws for Looping
7. Conclusion & Future Work