LiFT Loops a Shared DiT Core: FID 3.34 Lower With 60% Fewer Parameters on ImageNet

burny_tech · x · 2026-10-21

LiFT (Loop Flow Transformers) is a family of looped generative models that scales computation by repeatedly applying a shared Diffusion Transformer (DiT) core with only light architectural changes. Each recurrent step trains against a single regression target — a point on the straight path from the model's initial estimate to the flow-matching target — indexed by a continuous depth coordinate, so a trained model can loop far beyond its training depth without retraining or early exits. On ImageNet 256×256, LiFT-L/2 achieves FID 3.34 lower than a dense DiT-XL/2 baseline with 60% fewer parameters, 32% fewer training FLOPs, and 52% fewer inference FLOPs — inference compute can grow independently of parameter count.

Related event: LiFT: Looping Shared DiT Core Cuts Parameters 60%, Improves FID(2 posts)→

Original post →

More from Multimodal

Multimodal channel →