动态专家路由让小模型实现实时开放词汇检测,性能超大模型。
Dynamic-DINO: Fine-Grained Mixture of Experts Tuning for Real-time Open-Vocabulary Object Detection
- 用细粒度专家网络分解FFN,实现动态推理
- 仅用156万开源数据训练,超越依赖2000万私有数据的基线
- 深层专家固定搭档,适合资源受限场景部署
Mixture of Experts(MoE)架构在大型视觉语言模型中表现优异,但在基于大规模视觉语言数据集、但模型较小的实时开放词汇目标检测器中的潜力尚未被探索。本文揭示了有趣现象:浅层专家倾向于与多样化同伴协作以扩展搜索空间;深层则形成固定协作结构,每个专家维持2-3个固定伙伴,不同组合专门处理特定模式。为此提出Dynamic-DINO,通过高效的MoE-Tuning策略将Grounding DINO 1.5 Edge从密集模型转变为动态推理框架。设计细粒度分解机制,将基础模型的前馈网络(FFN)拆分为多个小型专家网络,扩大子网搜索空间。为防止微调初期性能下降,提出预训练权重分配策略与专用路由器初始化。推理时仅激活与输入相关的专家,形成紧凑子网。实验表明,仅使用156万开源数据预训练的Dynamic-DINO,性能优于使用私有Grounding20M数据集训练的Grounding DINO 1.5 Edge。
原文摘要 · Abstract (English)
The Mixture of Experts (MoE) architecture has excelled in Large Vision-Language Models (LVLMs), yet its potential in real-time open-vocabulary object detectors, which also leverage large-scale vision-language datasets but smaller models, remains unexplored. This work investigates this domain, revealing intriguing insights. In the shallow layers, experts tend to cooperate with diverse peers to expand the search space. While in the deeper layers, fixed collaborative structures emerge, where each expert maintains 2-3 fixed partners and distinct expert combinations are specialized in processing specific patterns. Concretely, we propose Dynamic-DINO, which extends Grounding DINO 1.5 Edge from a dense model to a dynamic inference framework via an efficient MoE-Tuning strategy. Additionally, we design a granularity decomposition mechanism to decompose the Feed-Forward Network (FFN) of base model into multiple smaller expert networks, expanding the subnet search space. To prevent performance degradation at the start of fine-tuning, we further propose a pre-trained weight allocation strategy for the experts, coupled with a specific router initialization. During inference, only the input-relevant experts are activated to form a compact subnet. Experiments show that, pretrained with merely 1.56M open-source data, Dynamic-DINO outperforms Grounding DINO 1.5 Edge, pretrained on the private Grounding20M dataset.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。