用自适应元学习提升文本检索模型在合成数据上的泛化能力
CAMeL: Cross-modality Adaptive Meta-Learning for Text-based Person Retrieval
- 设计跨模态自适应元学习框架,动态记忆错误样本并平衡新旧任务更新
- 在CUHK-PEDES等真实数据集上超越现有方法,对噪声文本和偏斜图像有强鲁棒性
- 适合研究多模态检索、合成数据训练的开发者,尤其关注模型泛化性
基于文本的人体检索旨在通过文本描述从图像库中识别特定个体。由于标注成本高且涉及隐私,研究者常采用合成数据进行预训练与微调。然而,这些生成数据在图像和文本注释上普遍存在领域偏差,严重影响预训练模型的可扩展性。为此,我们提出一种无领域偏倚的预训练框架——跨模态自适应元学习(CAMeL),以增强模型在预训练阶段的泛化能力,从而支持下游任务。具体而言,我们设计了一系列反映真实场景多样性和复杂性的任务,并引入动态错误样本记忆单元,记录多任务中遭遇的历史错误。为促进多任务适应,还采用自适应双速更新策略,平衡快速适应新任务与缓慢更新历史任务权重。尽管结构简单,该模型在真实世界基准测试(包括CUHK-PEDES、ICFG-PEDES和RSTPReid)上均优于现有最先进方法,并在处理有偏差的合成图像和噪声文本时展现出良好的鲁棒性与可扩展性。代码已开源。
原文摘要 · Abstract (English)
Text-based person retrieval aims to identify specific individuals within an image database using textual descriptions. Due to the high cost of annotation and privacy protection, researchers resort to synthesized data for the paradigm of pretraining and fine-tuning. However, these generated data often exhibit domain biases in both images and textual annotations, which largely compromise the scalability of the pre-trained model. Therefore, we introduce a domain-agnostic pretraining framework based on Cross-modality Adaptive Meta-Learning (CAMeL) to enhance the model generalization capability during pretraining to facilitate the subsequent downstream tasks. In particular, we develop a series of tasks that reflect the diversity and complexity of real-world scenarios, and introduce a dynamic error sample memory unit to memorize the history for errors encountered within multiple tasks. To further ensure multi-task adaptation, we also adopt an adaptive dual-speed update strategy, balancing fast adaptation to new tasks and slow weight updates for historical tasks. Albeit simple, our proposed model not only surpasses existing state-of-the-art methods on real-world benchmarks, including CUHK-PEDES, ICFG-PEDES, and RSTPReid, but also showcases robustness and scalability in handling biased synthetic images and noisy text annotations. Our code is available at https://github.com/Jahawn-Wen/CAMeL-reID.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。