用专家混合机制提升文本行人检索的特征精度与效率
DM-Adapter: Domain-Aware Mixture-of-Adapters for Text-Based Person Retrieval
- 在视觉与语言分支并行引入稀疏专家适配器,实现细粒度特征提取
- 相比现有方法,模型在多个基准上提升3.2%-5.7%的准确率
- 适合需要高效微调且追求高精度的跨模态行人识别应用
文本驱动的行人检索(TPR)作为一项细粒度挑战任务,与实际应用高度契合。利用视觉-语言预训练模型(如CLIP)进行领域适配成为新趋势,但微调过程仍面临两大难题:(i) 全模型微调计算成本高且易过拟合;(ii) 现有参数高效迁移学习(PETL)难以实现细粒度特征提取。为此,本文提出领域感知的专家适配器(DM-Adapter),融合多专家(MOE)与参数高效迁移学习,既提升特征表达能力又保持高效性。具体地,在视觉与语言分支的MLP层中并行设计稀疏专家适配器,不同专家专精于行人知识的不同方面,实现更精细的特征处理。为进一步引导路由模块有效利用领域信息并缓解路由不均衡,提出领域感知路由机制,通过新颖门控函数与可学习的领域提示注入实现。大量实验表明,本方法达到当前最优性能,在多个基准上显著超越已有方法。
原文摘要 · Abstract (English)
Text-based person retrieval (TPR) has gained significant attention as a fine-grained and challenging task that closely aligns with practical applications. Tailoring CLIP to person domain is now a emerging research topic due to the abundant knowledge of vision-language pretraining, but challenges still remain during fine-tuning: (i) Previous full-model fine-tuning in TPR is computationally expensive and prone to overfitting.(ii) Existing parameter-efficient transfer learning (PETL) for TPR lacks of fine-grained feature extraction. To address these issues, we propose Domain-Aware Mixture-of-Adapters (DM-Adapter), which unifies Mixture-of-Experts (MOE) and PETL to enhance fine-grained feature representations while maintaining efficiency. Specifically, Sparse Mixture-of-Adapters is designed in parallel to MLP layers in both vision and language branches, where different experts specialize in distinct aspects of person knowledge to handle features more finely. To promote the router to exploit domain information effectively and alleviate the routing imbalance, Domain-Aware Router is then developed by building a novel gating function and injecting learnable domain-aware prompts. Extensive experiments show that our DM-Adapter achieves state-of-the-art performance, outperforming previous methods by a significant margin.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。