Search papers, labs, and topics across Lattice.
Swift-Image is a compact unified model designed for text-to-image generation and image editing, achieving high performance with only 6 billion parameters through innovative training techniques. The model employs a progressive training pipeline that enhances visual quality and semantic coverage, alongside a unique Prompt Enhancer to align user requests with visual outputs. Notably, Swift-Image outperforms other open-source models while maintaining efficiency, with a compressed variant that shows negligible quality loss and improved editing performance through few-step distillation.
Achieving leading performance in image generation with only 6 billion parameters, Swift-Image redefines the efficiency frontier for compact models.
We present Swift-Image, a compact unified model for text-to-image generation, single-image editing, and multi-image editing. Our goal is to explore how far a relatively small visual generator can be pushed through systematic training engineering under a constrained computational budget. Swift-Image adopts an efficient 6B single-stream DiT and a progressive training pipeline that evolves from broad semantic coverage to higher resolution, stronger visual quality, and unified generation-editing supervision. For post-training, we employ parallel expert reinforcement learning followed by multi-teacher on-policy distillation to alleviate interference among heterogeneous objectives. We further decouple high-level reasoning from pixel-level rendering with a Prompt Enhancer that translates user requests into generator-aligned visual specifications. For efficient deployment, structural pruning and few-step distillation produce 3B and accelerated variants. Swift-Image achieves leading aggregate performance among evaluated open-source models with only 6B parameters and 243K GPU training hours; the compressed 3B model incurs nearly no loss, while few-step distillation further improves aggregate editing performance with substantially fewer sampling steps. Our study also summarizes practical lessons for architecture, data curriculum, post-training, prompt enhancement, and model compression.
