LegoDiffusion: Micro-Serving Text-to-Image Diffusion Workflows

Lingyun Yang, Suyi Li, Tianyu Feng, Xiaoxiao Jiang, Zhipeng Di +8 more
4/9/2026
cs.DCcs.AI
diffusion-flow

No confident code implementation found for this paper.

Abstract

Text-to-image generation executes a diffusion workflow comprising multiple models centered on a base diffusion model. Existing serving systems treat each workflow as an opaque monolith, provisioning, placing, and scaling all constituent models together, which obscures internal dataflow, prevents model sharing, and enforces coarse-grained resource management. In this paper, we make a case for micro-serving diffusion workflows with LegoDiffusion, a system that decomposes a workflow into loosely coupled model-execution nodes that can be independently managed and scheduled. By explicitly managing individual model inference, LegoDiffusion unlocks cluster-scale optimizations, including per-model scaling, model sharing, and adaptive model parallelism. Collectively, LegoDiffusion outperforms existing diffusion workflow serving systems, sustaining up to 3x higher request rates and tolerating up to 8x higher burst traffic.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{yang2026legodiffusion,
  title  = {LegoDiffusion: Micro-Serving Text-to-Image Diffusion Workflows},
  author = {Lingyun Yang and Suyi Li and Tianyu Feng and Xiaoxiao Jiang and Zhipeng Di and Weiyi Lu and Kan Liu and Yinghao Yu and Tao Lan and Guodong Yang and Lin Qu and Liping Zhang and Wei Wang},
  year   = {2026},
  eprint = {2604.08123},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2604.08123v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2604.08123.svg)](https://codeofpaper.com/paper/2604.08123)

Discussion