LiFT loops a shared DiT core to beat dense baseline by 3.34 FID with 60% fewer params

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 core with only light architectural changes. Each recurrent step is trained with a single regression target—a point on a straight path from the model's initial estimate to the flow-matching target. Because targets are indexed by a continuous depth coordinate, a trained model can loop far beyond its training depth with no retraining or early exits. On ImageNet 256x256, LiFT-L/2 achieves an FID 3.34 points lower than the dense DiT-XL/2 baseline with 60% fewer parameters, 32% fewer training FLOPs, and 52% fewer inference FLOPs.

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

Original post →

More from Research

Research channel →