SANA 1.5: Efficient Scaling of Training-Time and Inference-Time Compute in Linear Diffusion Transformer
About
This paper presents SANA-1.5, a linear Diffusion Transformer for efficient scaling in text-to-image generation. Building upon SANA-1.0, we introduce three key innovations: (1) Efficient Training Scaling: A depth-growth paradigm that enables scaling from 1.6B to 4.8B parameters with significantly reduced computational resources, combined with a memory-efficient 8-bit optimizer. (2) Model Depth Pruning: A block importance analysis technique for efficient model compression to arbitrary sizes with minimal quality loss. (3) Inference-time Scaling: A repeated sampling strategy that trades computation for model capacity, enabling smaller models to match larger model quality at inference time. Through these strategies, SANA-1.5 achieves a text-image alignment score of 0.81 on GenEval, which can be further improved to 0.96 through inference scaling with VILA-Judge, establishing a new SoTA on GenEval benchmark. These innovations enable efficient model scaling across different compute budgets while maintaining high quality, making high-quality image generation more accessible. Our code and pre-trained models are released.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Text-to-Image Generation | GenEval | Overall Score81 | 506 | |
| Text-to-Image Generation | GenEval | Overall Score79.4 | 391 | |
| Text-to-Image Generation | GenEval | GenEval Score81 | 360 | |
| Text-to-Image Generation | DPG-Bench | Overall Score85 | 265 | |
| Text-to-Image Generation | GenEval (test) | Two Obj. Acc93 | 221 | |
| Text-to-Image Generation | GenEval | Overall Score72 | 218 | |
| Text-to-Image Generation | DPG-Bench | DPG Score84.7 | 131 | |
| Text-to-Image Generation | GenEval | GenEval Score0.8 | 88 | |
| Text-to-Image Generation | GenEval 1.0 (test) | Overall Score80.62 | 85 | |
| Text-to-Image Generation | R2I-Bench | Causal Accuracy21 | 28 |