A Lightweight Velocity Correction Framework for Improving Instruction Following in Few-Step Text-to-Image Generation
Few-step text-to-image generators produce high-quality images with few network forward passes, yet struggle with complex prompts involving multiple subjects, interactions, spatial relations, counting, and attribute binding.
ProofPaper ↗
Key points
- We propose a lightweight correction framework for a frozen 6B-parameter generator.
- The generator predicts velocities that determine how its intermediate image representation is updated; a corrector with fewer than 20M parameters learns residual adjustments at selected sampling steps.
- To our knowledge, this is the first framework for lightweight velocity-residual adaptation of fully frozen few-step text-to-image generators that preserves their original sampling schedule and forward-pass count.
- On our target complex-scene distribution, the framework achieves a +17.60-percentage-point NetWin over frozen Z-Image-Turbo in pairwise instruction-following evaluation, with only 26.9 ms (0.84%) additional end-to-end latency per image and no increase in backbone evaluations.
Sources (1)
- [1]A Lightweight Velocity Correction Framework for Improving Instruction Following in Few-Step Text-to-Image GenerationarXiv (AI, ML, NLP, CV, robotics, multi-agent) · Oct 9, 02:43 AM
Few-step text-to-image generators produce high-quality images with few network forward passes, yet struggle with complex prompts involving multiple subjects, interactions, spatial relations, counting, and attribute binding.
We propose a lightweight correction framework for a frozen 6B-parameter generator.
Extractive summary: sentences quoted from the sources.
Before this
- Oct 9, 2026Rethinking Relative Policy Differences Transfer of On-Policy Distillation
- Oct 9, 2026Learning Representations to Guide Pixel Diffusion Transformers
- Oct 8, 2026One Block, Multiple Depths: Recurrent Vision Transformers with Depth-Programmed Experts
- Oct 8, 2026SpatialOPSD: Self-Distilling Spatial Intelligence from Verified Coding Agent Traces
- Oct 8, 2026Distilling Routed 3D Privilege for Spatial Reasoning in Vision-Language Models
- Oct 8, 2026ViSkill: Reinforcing VLM Agents with Evolving Visual-Native Skills