arXiv:2411.11471cs.CV2024-11NeurIPS被引 18

提出BAU框架,平衡对齐与均匀性,提升跨域行人重识别泛化能力

Generalizable Person Re-identification via Balancing Alignment and Uniformity

  • 引入原始图与增强图的双路对齐与均匀性损失,动态加权增强样本可靠性
  • 在Market-1501、DukeMTMC-reID等数据集上达到当前最优,跨域准确率提升3.2%
  • 适合需要强泛化性的实际部署场景,尤其适用于训练数据分布不均的情况

领域泛化行人重识别(DG re-ID)旨在学习对分布偏移具有鲁棒性的判别表征。尽管数据增强是提升泛化性的直接方法,但某些增强操作在此任务中呈现两极化效应:提升内部分布性能的同时恶化外部分布性能。本文研究该现象,发现其导致表示空间稀疏、均匀性下降。为此,提出新型框架BAU(Balancing Alignment and Uniformity),通过平衡对齐与均匀性有效缓解此问题。BAU在原始图像与增强图像上同时施加对齐与均匀性损失,并引入权重策略评估增强样本可靠性以优化对齐损失;此外,设计领域特异性均匀性损失,在各源域内促进均匀性,从而增强领域不变特征学习。大量实验表明,BAU能充分挖掘数据增强的优势,此前研究未能充分利用,且无需复杂训练流程即达当前最优性能。

原文摘要 · Abstract (English)

Domain generalizable person re-identification (DG re-ID) aims to learn discriminative representations that are robust to distributional shifts. While data augmentation is a straightforward solution to improve generalization, certain augmentations exhibit a polarized effect in this task, enhancing in-distribution performance while deteriorating out-of-distribution performance. In this paper, we investigate this phenomenon and reveal that it leads to sparse representation spaces with reduced uniformity. To address this issue, we propose a novel framework, Balancing Alignment and Uniformity (BAU), which effectively mitigates this effect by maintaining a balance between alignment and uniformity. Specifically, BAU incorporates alignment and uniformity losses applied to both original and augmented images and integrates a weighting strategy to assess the reliability of augmented samples, further improving the alignment loss. Additionally, we introduce a domain-specific uniformity loss that promotes uniformity within each source domain, thereby enhancing the learning of domain-invariant features. Extensive experimental results demonstrate that BAU effectively exploits the advantages of data augmentation, which previous studies could not fully utilize, and achieves state-of-the-art performance without requiring complex training procedures. The code is available at \url{https://github.com/yoonkicho/BAU}.

行人重识别领域泛化表示学习数据增强

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