mlsanityData to 7 Oct 2026

InstructMoLE: Instruction-Guided Mixture of Low-rank Experts for Multi-Conditional Image Generation

arXiv 2512.21788 ↗First posted 25 Dec 2025cs.CV cs.AIcited by 0

On the map: Diffusion models → Diffusion foundations and sampling

Abstract (arXiv)

Parameter-Efficient Fine-Tuning of Diffusion Transformers (DiTs) for diverse, multi-conditional tasks often suffers from task interference when using monolithic adapters like LoRA. The Mixture of Low-rank Experts (MoLE) architecture offers a modular solution, but its potential is usually limited by routing policies that operate at a token level. Such local routing can conflict with the global nature of user instructions, leading to artifacts like spatial fragmentation and semantic drift in complex image generation tasks. To address these limitations, we introduce InstructMoLE, a novel framework that employs an Instruction-Guided Mixture of Low-Rank Experts. Instead of per-token routing, InstructMoLE utilizes a global routing signal, Instruction-Guided Routing (IGR), derived from the user's comprehensive instruction. This ensures that a single, coherently chosen expert council is applied uniformly across all input tokens, preserving the global semantics and structural integrity of the generation process. To complement this, we introduce an output-space orthogonality loss, which promotes expert functional diversity and mitigates representational collapse. Extensive experiments demonstrate that InstructMoLE significantly outperforms existing LoRA adapters and MoLE variants across challenging multi-conditional generation benchmarks. Our work presents a robust and generalizable framework for instruction-driven fine-tuning of generative models, enabling superior compositional control and fidelity to user intent.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 36 of its 45 references that are in the map. First posted on arXiv 25 Dec 2025.

Builds on

36 of 45 references.

Closest ancestors · 6

Same direction, newest first

  1. 2025Expert Race: A Flexible Routing Strategy for Scaling Diffusion Transformer with Mixture of Experts20 Mar
  2. 2025EasyControl: Adding Efficient and Flexible Control for Diffusion Transformer10 Mar
  3. 2024EC-DIT: Scaling Diffusion Transformers with Adaptive Expert-Choice Routing2 Oct
  4. 2024Scaling Diffusion Transformers to 16 Billion Parameters16 Jul
  5. 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesis5 Mar
  6. 2022Scalable Diffusion Models with Transformers19 Dec

Other references · 30

  1. 2014Microsoft COCO: Common Objects in Contextcited by 17,498
  2. 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,764
  3. 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,413
  4. 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,014
  5. 2021Perceiver: General Perception with Iterative Attentioncited by 1,089
  6. 2025FLUX.1 Kontext: Flow Matching for In-Context Image Generation and Editing in Latent Spacecited by 960
  7. 2024Parameter-Efficient Fine-Tuning for Large Models: A Comprehensive Surveycited by 641
  8. 2024ELLA: Equip Diffusion Models with LLM for Enhanced Semantic Alignmentcited by 455
  9. 2025Step1X-Edit: A Practical Framework for General Image Editingcited by 417
  10. 2025In-Context Edit: Enabling Instructional Image Editing with In-Context Generation in Large Scale Diffusion Transformercited by 199
  11. 2023UniControl: A Unified Diffusion Model for Controllable Visual Generation In the Wildcited by 197
  12. 2024Mixture of LoRA Expertscited by 179
18 more
  1. 2025Less-to-More Generalization: Unlocking More Controllability by In-Context Generationcited by 153
  2. 2024MixLoRA: Enhancing Large Language Models Fine-Tuning with LoRA-based Mixture of Expertscited by 144
  3. 2023Mixture of Cluster-conditional LoRA Experts for Vision-language Instruction Tuningcited by 106
  4. 2025DreamO: A Unified Framework for Image Customizationcited by 82
  5. 2025GPT-IMAGE-EDIT-1.5M: A Million-Scale, GPT-Generated Image Datasetcited by 68
  6. 2024MoE-I2: Compressing Mixture of Experts Models through Inter-Expert Pruning and Intra-Expert Low-Rank Decompositioncited by 57
  7. 2025XVerse: Consistent Multi-Subject Control of Identity and Semantic Attributes via DiT Modulationcited by 55
  8. 2024Theory on Mixture-of-Experts in Continual Learningcited by 50
  9. 2025InfiniteYou: Flexible Photo Recrafting While Preserving Your Identitycited by 34
  10. 2025Omni-Effects: Unified and Spatially-Controllable Visual Effects Generationcited by 27
  11. 2023Octavius: Mitigating Task Interference in MLLMs via LoRA-MoEcited by 25
  12. 2025textttComplex-Edit: CoT-Like Instruction Generation for Complexity-Controllable Image Editing Benchmarkcited by 25
  13. 2023HALOC: Hardware-Aware Automatic Low-Rank Compression for Compact Neural Networkscited by 20
  14. 2023COMCAT: Towards Efficient Compression and Customization of Attention-Based Vision Modelscited by 14
  15. 2024Routing Experts: Learning to Route Dynamic Experts in Multi-modal Large Language Modelscited by 13
  16. 2025A Stronger Mixture of Low-Rank Experts for Fine-Tuning Foundation Modelscited by 9
  17. 2024COAP: Memory-Efficient Training with Correlation-Aware Gradient Projectioncited by 8
  18. 2025EcoSpa: Efficient Transformer Training with Coupled Sparsitycited by 1

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025Routing Matters in MoE: Scaling Diffusion Transformers with Explicit Routing Guidance
  2. 2026Sparse Mixture-of-Experts Routing in Visual Diffusion Transformers:Diagnosis, Boundary Calibration and Evolutionary Roadmap from Routing Collapse to Selective Deadlock
  3. 2024A Simple Approach to Unifying Diffusion-based Conditional Generation
  4. 2025Disentangling Instruction Influence in Diffusion Transformers for Parallel Multi-Instruction-Guided Image Editing
  5. 2023VIEScore: Towards Explainable Metrics for Conditional Image Synthesis Evaluation
  6. 2026Edit Where You Mean: Region-Aware Adapter Injection for Mask-Free Local Image Editing
  7. 2024UNIC-Adapter: Unified Image-instruction Adapter with Multi-modal Transformer for Image Generation
  8. 2026Nucleus-Image: Sparse MoE for Image Generation