用全局指令引导专家选择,让图像生成更忠实于用户意图。
InstructMoLE: Instruction-Guided Mixture of Low-rank Experts for Multi-Conditional Image Generation
- 根据用户完整指令全局选专家,避免局部路由导致的碎片化
- 在多个复杂任务上超越LoRA和传统MoLE,生成更连贯图像
- 适合需要精准控制生成内容的AI绘画与多条件创作场景
针对扩散Transformer模型在多样、多条件任务中进行参数高效微调时,因使用单体适配器(如LoRA)导致的任务干扰问题,提出InstructMoLE框架。该框架采用低秩专家混合(MoLE)架构,但创新性地引入基于用户完整指令的全局路由信号——指令引导路由(IGR),取代传统的逐标记路由,确保所有输入标记统一应用一组协调的专家组合,从而保持生成过程的全局语义一致性和结构完整性。同时设计输出空间正交性损失,增强专家功能多样性,防止表征坍缩。大量实验表明,InstructMoLE在多个复杂多条件生成基准上显著优于现有LoRA适配器及MoLE变体,展现出更强的可组合控制能力与用户意图遵循性。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。