分布式视觉指令微调中无需重放即可实现私有专家的模块化路由。
DistMoE: Private-data Rehearsal-free Routing in Mixture-of-Experts for Distributed Instruction Tuning

- 通过公共锚定机制,仅更新路由和轻量适配器,避免私有专家间差异。
- 在多个视觉语言基准上表现接近集中式训练,且支持灵活复用专家。
- 适合数据分散在不同隐私或权限受限客户端的场景,保护数据隐私。
多模态大语言模型在多模态指令遵循方面表现出色,但将其适配到多样化视觉-语言领域通常依赖集中式数据访问和昂贵的联合训练,这在数据分布于私有、特定领域或权限受限客户端时受到限制。为此,我们提出 DistMoE,一种用于分布式视觉指令微调的混合专家(MoE)方法。在语言解码器的每一层,其在公开前馈网络(FFN)基础上增加客户端特定的私有FFN专家,以获取领域特定知识。然而,独立训练导致私有FFN学习不同尺度和幅值的表示,难以合并。为减少客户端偏差,我们引入公共锚定专家组合阶段,仅通过等向性正则化损失更新路由器和轻量级私有投影适配器,在本地客户端数据与公共数据混合上进行训练,从而实现跨客户端无重放的组合。推理时,DistMoE对公共与私有专家进行模块化路由,实现无需显式领域标签的逐标记域组合。在多个视觉-语言基准上的实验表明,DistMoE支持灵活的专家复用、有效的领域适应,并取得具有竞争力的性能,同时保持对客户端知识的模块化控制。代码已开源。
原文摘要 · Abstract (English)
Multimodal Large Language Models (MLLMs) have shown strong multimodal instruction-following ability, but adapting them to diverse visual-language domains typically assumes centralized data access and costly joint training. This is restrictive when data is distributed across private, domain-specific, or permission-limited clients. To this end, we propose DistMoE, a mixture-of-experts (MoE) approach for distributed visual instruction tuning. In each layer of the language decoder it augments the public feedforward network (FFN) with a client-specific private FFN expert, with the goal to acquire domain-specific knowledge. However, independent expert training causes the private FFNs to learn representation of different scale and magnitudes, making merging the experts difficult. To reduce client-specific drift, we introduce a public-anchored expert composition stage that updates only routers and lightweight private projection adapters on a mix of local client data and public data, via an isotropic regularization loss, therefore making it cross-client rehearsal-free composition. During inference, DistMoE performs modular routing over public and private experts, enabling token-wise domain composition without explicit domain labels. Experiments across diverse visual-language benchmarks show that DistMoE enables flexible expert reuse, effective domain adaptation, and competitive performance while preserving modular control over client-specific knowledge. Codes are available at https://github.com/mainaksingha01/DistMoE.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。