arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2507.01351cs.CV

长尾分布感知的混合专家路由器:用于大规模视觉-语言模型的混合专家架构

Long-Tailed Distribution-Aware Router For Mixture-of-Experts in Large Vision-Language Model

School of Software Technology, Zhejiang University(浙江大学软件学院);College of Computer Science and Technology, Zhejiang University(浙江大学计算机科学与技术学院);School of Electronic and Information, Shanghai Dianji University(上海电机学院电子信息学院)

Chaoxiang Cai, Longrong Yang, Minghe Weng, Xuewei Li, Zequn Qin, Xi Li

AI总结:

本文提出LTDR路由器,针对视觉-语言混合专家架构中的长尾分布问题,通过模态特定的分布感知路由和动态专家激活策略,提升模型性能。

摘要:

混合专家(MoE)架构通过将密集网络替换为稀疏网络,在大规模视觉-语言模型(LVLMs)中实现了在激活更少参数的情况下获得可比性能。现有LVLMs的MoE架构主要关注于token到专家的路由(TER),鼓励不同专家专门处理特定token。然而,这些方法通常依赖负载平衡机制,忽视了视觉和语言模态之间的固有分布差异。为了解决这一限制,我们提出了长尾分布感知的路由器(LTDR)用于视觉-语言TER,解决了两个关键挑战:(1)模态特定的分布感知路由。我们观察到语言TER通常遵循相对均匀的分布,而视觉TER表现出长尾分布。这种模态差异促使设计专门的路由策略。 (2)视觉特定的动态专家激活。认识到高信息视觉尾token的重要性,我们引入了一种数据增强启发策略,增加激活专家的数量,确保这些罕见但信息丰富的token得到充分学习。在视觉-语言和视觉基准上,我们的方法实现了稳定的提升,视觉-语言性能提升了1.2%/2.1%,视觉基准提升了1.6%。

英文摘要:

The mixture-of-experts (MoE) architecture, which replaces dense networks with sparse ones, has attracted significant attention in large vision-language models (LVLMs) for achieving comparable performance while activating far fewer parameters. Existing MoE architectures for LVLMs primarily focus on token-to-expert routing (TER), encouraging different experts to specialize in processing specific tokens. However, these methods typically rely on the load balancing mechanism, neglecting the inherent distributional differences between vision and language modalities. To address this limitation, we propose the Long-Tailed Distribution-aware Router (LTDR) for vision-language TER, which tackles two key challenges: (1) Modality-specific distribution-aware routing. We observe that language TER generally follows a relatively uniform distribution, whereas vision TER exhibits a long-tailed distribution. This modality discrepancy motivates the design of specialized routing strategies for each modality. (2) Vision-specific dynamic expert activation. Recognizing the importance of high-information vision tail tokens, we introduce a data-augmentation-inspired strategy that increases the number of activated experts, ensuring sufficient learning for these rare but informative tokens. On vision-language and vision benchmarks, our approach achieves consistent improvements, boosting performance by 1.2% / 2.1% on vision-language and 1.6% on vision benchmarks.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑