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

面向行人重识别预训练的跨视频身份关联方法

Cross-video Identity Correlating for Person Re-identification Pre-training

Huazhong University of Science and Technology(华中科技大学);Huawei Noah’s Ark Lab(华为诺亚方舟实验室)

Jialong Zuo, Ying Nie, Hanyu Zhou, Huaxin Zhang, Haoyu Wang, Tianyu Guo, Nong Sang, Changxin Gao

AI总结:

本文提出跨视频身份关联预训练框架 CION,通过渐进式多级去噪和身份引导自蒸馏挖掘跨视频身份不变性,以更少样本取得领先性能,并发布含 32 个模型的 ReIDZoo 模型库。

摘要:

近年来的研究表明,在大规模从互联网视频中提取的行人图像上进行预训练,是学习更优行人重识别表示的有效途径。然而,这些研究大多局限于实例级或单视频轨迹片段级的预训练,忽略了同一行人在不同视频图像中的身份不变性,而这正是行人重识别的关键关注点。为解决这一问题,我们提出了一种跨视频身份关联预训练(CION)框架。通过定义一个同时考虑身份内一致性和身份间判别性的噪声概念,CION 将跨视频图像中的身份关联建模为一个渐进式多级去噪问题来求解。此外,我们提出了一种身份引导的自蒸馏损失,通过挖掘行人图像中的身份不变性来实现更优的大规模预训练。我们进行了大量实验,验证了 CION 在效率和性能上的优越性。CION 在使用更少训练样本的情况下取得了显著领先的性能。例如,与先前最先进的方法 ISR 相比,使用相同的 ResNet50-IBN 骨干网络,CION 在 Market1501 和 MSMT17 上分别取得了 93.3% 和 74.3% 的更高 mAP,而仅利用了 8% 的训练样本。最后,凭借 CION 展现出的优越模型无关能力,我们贡献了一个名为 ReIDZoo 的模型库,以满足该领域多样化的研究和应用需求。该模型库包含一系列 CION 预训练模型,涵盖多种结构和参数规模,共计 32 个模型、10 种不同结构,包括 GhostNet、ConvNext、RepViT、FastViT 等。代码和模型将在 https://github.com/Zplusdragon/CION_ReIDZoo 公开发布。

英文摘要:

Recent researches have proven that pre-training on large-scale person images extracted from internet videos is an effective way in learning better representations for person re-identification. However, these researches are mostly confined to pre-training at the instance-level or single-video tracklet-level. They ignore the identity-invariance in images of the same person across different videos, which is a key focus in person re-identification. To address this issue, we propose a Cross-video Identity-cOrrelating pre-traiNing (CION) framework. Defining a noise concept that comprehensively considers both intra-identity consistency and inter-identity discrimination, CION seeks the identity correlation from cross-video images by modeling it as a progressive multi-level denoising problem. Furthermore, an identity-guided self-distillation loss is proposed to implement better large-scale pre-training by mining the identity-invariance within person images. We conduct extensive experiments to verify the superiority of our CION in terms of efficiency and performance. CION achieves significantly leading performance with even fewer training samples. For example, compared with the previous state-of-the-art~\cite{ISR}, CION with the same ResNet50-IBN achieves higher mAP of 93.3\% and 74.3\% on Market1501 and MSMT17, while only utilizing 8\% training samples. Finally, with CION demonstrating superior model-agnostic ability, we contribute a model zoo named ReIDZoo to meet diverse research and application needs in this field. It contains a series of CION pre-trained models with spanning structures and parameters, totaling 32 models with 10 different structures, including GhostNet, ConvNext, RepViT, FastViT and so on. The code and models will be made publicly available at https://github.com/Zplusdragon/CION_ReIDZoo.

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

↑