mlsanityData to 7 Oct 2026

Diffusion Templates: A Unified Plugin Framework for Controllable Diffusion

arXiv 2604.24351 ↗First posted 27 Apr 2026cs.LG cs.AI cs.CVcited by 3

On the map: Diffusion models → Diffusion foundations and sampling

Abstract (arXiv)

Controllable diffusion methods have substantially expanded the practical utility of diffusion models, but they are typically developed as isolated, backbone-specific systems with incompatible training pipelines, parameter formats, and runtime hooks. This fragmentation makes it difficult to reuse infrastructure across tasks, transfer capabilities across backbones, or compose multiple controls within a single generation pipeline. We present Diffusion Templates, a unified and open plugin framework that decouples base-model inference from controllable capability injection. The framework is organized around three components: Template models that map arbitrary task-specific inputs to an intermediate capability representation, a Template cache that functions as a standardized interface for capability injection, and a Template pipeline that loads, merges, and injects one or more Template caches into the base diffusion runtime. Because the interface is defined at the systems level rather than tied to a specific control architecture, heterogeneous capability carriers such as KV-Cache and LoRA can be supported under the same abstraction. Based on this design, we build a diverse model zoo spanning structural control, brightness adjustment, color adjustment, image editing, super-resolution, sharpness enhancement, aesthetic alignment, content reference, local inpainting, and age control. These case studies show that Diffusion Templates can unify a broad range of controllable generation tasks while preserving modularity, composability, and practical extensibility across rapidly evolving diffusion backbones. All resources will be open sourced, including code, models, and datasets.

Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 42 of its 49 references that are in the map. First posted on arXiv 27 Apr 2026.

Builds on

42 of 49 references.

Closest ancestors · 9

Same direction, newest first

  1. 2024SANA: Efficient High-Resolution Image Synthesis with Linear Diffusion Transformers14 Oct
  2. 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesis5 Mar
  3. 2023PixArt-α: Fast Training of Diffusion Transformer for Photorealistic Text-to-Image Synthesis30 Sep
  4. 2023SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesis4 Jul
  5. 2022Scalable Diffusion Models with Transformers19 Dec
  6. 2022Classifier-Free Diffusion Guidance26 Jul
  7. 2021High-Resolution Image Synthesis with Latent Diffusion Models20 Dec
  8. 2020Denoising Diffusion Implicit Models6 Oct
  9. 2020Denoising Diffusion Probabilistic Models19 Jun

Other references · 33

  1. 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
  2. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
  3. 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
  4. 2022ReAct: Synergizing Reasoning and Acting in Language Modelscited by 8,355
  5. 2025Qwen3 Technical Reportcited by 8,195
  6. 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
  7. 2023Adding Conditional Control to Text-to-Image Diffusion Modelscited by 5,466
  8. 2023Toolformer: Language Models Can Teach Themselves to Use Toolscited by 4,072
  9. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
  10. 2022DreamBooth: Fine Tuning Text-to-Image Diffusion Models for Subject-Driven Generationcited by 3,286
  11. 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,551
  12. 2025Wan: Open and Advanced Large-Scale Video Generative Modelscited by 2,536
21 more
  1. 2022An Image is Worth One Word: Personalizing Text-to-Image Generation using Textual Inversioncited by 2,231
  2. 2024HunyuanVideo: A Systematic Framework For Large Video Generative Modelscited by 1,597
  3. 2023IP-Adapter: Text Compatible Image Prompt Adapter for Text-to-Image Diffusion Modelscited by 1,439
  4. 2023The Rise and Potential of Large Language Model Based Agents: A Surveycited by 1,422
  5. 2023T2I-Adapter: Learning Adapters to Dig out More Controllable Ability for Text-to-Image Diffusion Modelscited by 1,339
  6. 2025SigLIP 2: Multilingual Vision-Language Encoders with Improved Semantic Understanding, Localization, and Dense Featurescited by 1,127
  7. 2025Qwen-Image Technical Reportcited by 973
  8. 2021Real-ESRGAN: Training Real-World Blind Super-Resolution with Pure Synthetic Datacited by 961
  9. 2023Pick-a-Pic: An Open Dataset of User Preferences for Text-to-Image Generationcited by 950
  10. 2024SnapKV: LLM Knows What You are Looking for Before Generationcited by 848
  11. 2023H2O: Heavy-Hitter Oracle for Efficient Generative Inference of Large Language Modelscited by 805
  12. 2023Tool Learning with Foundation Modelscited by 302
  13. 2024Hunyuan-DiT: A Powerful Multi-Resolution Diffusion Transformer with Fine-Grained Chinese Understandingcited by 294
  14. 2026LTX-2: Efficient Joint Audio-Visual Foundation Modelcited by 213
  15. 2024Mooncake: A KVCache-centric Disaggregated Architecture for LLM Servingcited by 193
  16. 2024GenAI Arena: An Open Evaluation Platform for Generative Modelscited by 81
  17. 2024Preble: Efficient Distributed Prompt Scheduling for LLM Servingcited by 68
  18. 2024InferCept: Efficient Intercept Support for Augmented Large Language Model Inferencecited by 32
  19. 2025EliGen: Entity-Level Controlled Image Generation with Regional Attentioncited by 32
  20. 2025AttriCtrl: Fine-Grained Control of Aesthetic Attribute Intensity in Diffusion Modelscited by 5
  21. 2024ArtAug: Enhancing Text-to-Image Generation through Synthesis-Understanding Interactioncited by 1

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 3.

3 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2025FreSca: Scaling in Frequency Space Enhances Diffusion Models
  2. 2026On the Controllability-Fidelity Frontier in Diffusion Editing
  3. 2026Improving Controllable Generation: Faster Training and Better Performance via x0-Supervision
  4. 2025UniCon: Unidirectional Information Flow for Effective Control of Large-Scale Diffusion Models
  5. 2024Ctrl-Adapter: An Efficient and Versatile Framework for Adapting Diverse Controls to Any Diffusion Model
  6. 2025SliderSpace: Decomposing the Visual Capabilities of Diffusion Models
  7. 2023Differential Diffusion: Giving Each Pixel Its Strength
  8. 2023Image is All You Need to Empower Large-scale Diffusion Models for In-Domain Generation