arXiv:2505.19525cs.LGcs.AI2025-05被引 7

提出新门控机制,让稀疏专家模型更好处理缺失模态数据

Rethinking Gating Mechanism in Sparse MoE: Handling Arbitrary Modality Inputs with Confidence-Guided Gate

  • 分离路由得分与任务置信度,基于真实标签动态调整专家选择
  • 在四个真实数据集上验证,对缺失模态的鲁棒性显著提升
  • 无需额外平衡损失,可缓解专家坍塌,适合多模态实际场景

真实世界多模态学习中,数据缺失常由系统性采集误差或传感器故障导致。稀疏混合专家(SMoE)架构虽具天然处理多模态潜力,但现有方法难以应对缺失模态,导致性能下降和泛化能力差。本文提出ConfSMoE,引入两阶段插补模块,并通过理论分析揭示专家坍塌机制。受此启发,设计新型门控机制:将softmax路由得分解耦为基于真实标签的任务置信度。该方法无需额外负载均衡损失即可有效缓解专家坍塌,且与高斯、拉普拉斯门控等机制的洞察一致。在四个真实数据集、三种实验设置下评估,全面验证了ConfSMoE在缺失模态下的鲁棒性及门控机制的有效性。

原文摘要 · Abstract (English)

Effectively managing missing modalities is a fundamental challenge in real-world multimodal learning scenarios, where data incompleteness often results from systematic collection errors or sensor failures. Sparse Mixture-of-Experts (SMoE) architectures have the potential to naturally handle multimodal data, with individual experts specializing in different modalities. However, existing SMoE approach often lacks proper ability to handle missing modality, leading to performance degradation and poor generalization in real-world applications. We propose ConfSMoE to introduce a two-stage imputation module to handle the missing modality problem for the SMoE architecture by taking the opinion of experts and reveal the insight of expert collapse from theoretical analysis with strong empirical evidence. Inspired by our theoretical analysis, ConfSMoE propose a novel expert gating mechanism by detaching the softmax routing score to task confidence score w.r.t ground truth signal. This naturally relieves expert collapse without introducing additional load balance loss function. We show that the insights of expert collapse aligns with other gating mechanism such as Gaussian and Laplacian gate. The proposed method is evaluated on four different real world dataset with three distinct experiment settings to conduct comprehensive analysis of ConfSMoE on resistance to missing modality and the impacts of proposed gating mechanism.

多模态专家模型门控机制数据缺失

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。