A Transformer-based Unsupervised Clustering Method for Vehicle Re-identification

Weifan Wu, Wei Ke, Hao Sheng

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

Current unsupervised re-identification methods use a clustering-based neural network for training. In the vehicle re-identification field, the feature information between different vehicles is small, and it is not easy to distinguish the detailed features of different vehicles using only the basic clustering algorithm for unsupervised learning. When clustering is performed, the general clustering methods inevitably put different vehicles together due to the high similarity. We propose a new re-identification method to solve these problems. This method is based on clustering and use the unsupervised learning. First, we employ the vision transformer structure as a feature extractor. The vision transformer structure can obtain more discriminative and correlated features than the general convolution. Second, we use a fine-grained clustering method to subdivide the clustered information into different vehicles. We trained our method on two open-source datasets, and finally obtained better test results without additional labeling information.

Original languageEnglish
Title of host publication6th IEEE International Conference on Universal Village, UV 2022
PublisherInstitute of Electrical and Electronics Engineers Inc.
ISBN (Electronic)9781665474771
DOIs
Publication statusPublished - 2022
Event6th IEEE International Conference on Universal Village, UV 2022 - Hybrid, Boston, United States
Duration: 22 Oct 202225 Oct 2022

Publication series

Name6th IEEE International Conference on Universal Village, UV 2022

Conference

Conference6th IEEE International Conference on Universal Village, UV 2022
Country/TerritoryUnited States
CityHybrid, Boston
Period22/10/2225/10/22

Keywords

  • clustering
  • re-identification
  • vision transformer

Fingerprint

Dive into the research topics of 'A Transformer-based Unsupervised Clustering Method for Vehicle Re-identification'. Together they form a unique fingerprint.

Cite this