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

CMTR:用于可见光-红外行人重识别的跨模态Transformer

CMTR: Cross-modality Transformer for Visible-infrared Person Re-identification

School of Computer and Information Technology, Beijing Jiaotong University(北京交通大学计算机与信息技术学院);JD AI Research(京东人工智能研究院)

Tengfei Liang, Yi Jin, Yajun Gao, Wu Liu, Songhe Feng, Tao Wang, Yidong Li

AI总结:

针对可见光-红外跨模态行人重识别任务,提出首个基于Transformer的CMTR方法,通过设计模态嵌入融合token信息并提出模态感知增强损失,显著提升了跨模态特征判别力,性能超越现有CNN方法。

摘要:

可见光-红外跨模态行人重识别是一项具有挑战性的ReID任务,旨在异构的可见光和红外模态之间检索并匹配同一身份的图像。因此,该任务的核心是弥合这两种模态之间的巨大鸿沟。现有的基于卷积神经网络的方法主要面临对模态信息感知不足的问题,无法为身份学习良好的判别性模态不变嵌入,这限制了它们的性能。为了解决这些问题,我们提出了一种基于跨模态Transformer的方法(CMTR)用于可见光-红外行人重识别任务,该方法能够显式挖掘每种模态的信息,并基于此生成更好的判别性特征。具体而言,为了捕获模态特征,我们设计了新颖的模态嵌入,将其与token嵌入融合以编码模态信息。此外,为了增强模态嵌入的表示并调整匹配嵌入的分布,我们基于学习到的模态信息提出了一种模态感知增强损失,以减小类内距离并扩大类间距离。据我们所知,这是将Transformer网络应用于跨模态重识别任务的首项工作。我们在公开的SYSU-MM01和RegDB数据集上进行了大量实验,我们提出的CMTR模型的性能显著超越了现有的基于CNN的杰出方法。

英文摘要:

Visible-infrared cross-modality person re-identification is a challenging ReID task, which aims to retrieve and match the same identity's images between the heterogeneous visible and infrared modalities. Thus, the core of this task is to bridge the huge gap between these two modalities. The existing convolutional neural network-based methods mainly face the problem of insufficient perception of modalities' information, and can not learn good discriminative modality-invariant embeddings for identities, which limits their performance. To solve these problems, we propose a cross-modality transformer-based method (CMTR) for the visible-infrared person re-identification task, which can explicitly mine the information of each modality and generate better discriminative features based on it. Specifically, to capture modalities' characteristics, we design the novel modality embeddings, which are fused with token embeddings to encode modalities' information. Furthermore, to enhance representation of modality embeddings and adjust matching embeddings' distribution, we propose a modality-aware enhancement loss based on the learned modalities' information, reducing intra-class distance and enlarging inter-class distance. To our knowledge, this is the first work of applying transformer network to the cross-modality re-identification task. We implement extensive experiments on the public SYSU-MM01 and RegDB datasets, and our proposed CMTR model's performance significantly surpasses existing outstanding CNN-based methods.

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

↑