학술논문

Pseudo-label Assisted Optimization of Multi-branch Network for Cross-domain Person Re-identification
Document Type
Conference
Source
2023 IEEE International Conference on Mechatronics and Automation (ICMA) Mechatronics and Automation (ICMA), 2023 IEEE International Conference on. :13-18 Aug, 2023
Subject
Bioengineering
Components, Circuits, Devices and Systems
Computing and Processing
Robotics and Control Systems
Signal Processing and Analysis
Training
Representation learning
Adaptation models
Mechatronics
Optimization methods
Training data
Feature extraction
Person re-identification
Pseudo-label
Multi-branch network
Cross domain adaptive
Language
ISSN
2152-744X
Abstract
Person re-identification plays an important role in the field of robot intelligence perception and public safety. Its main task is to identify person targets under cross-camera. However, the domain diversity between different datasets poses a clear challenge for adapting a model trained on one dataset to another. Currently, person re-identification methods based on domain adaptive learning and pseudo-label have made good progress on this problem. Unfortunately, inferior pseudo-labels and source domain noise affect the performance. In order to improve the quality of generated pseudo-labels and enhance the feature representation capability of the model, we propose a pseudo-label assisted optimization of a multi-branch person re-identification method. The multi-branch network is able to extract and represent more effective global and local features, and the generated pseudo-labels are optimized by using cosine similarity and DBSCAN clustering on the feature vectors, thus improving the consistency of the supervised information to enhance the cross-domain recognition performance. We also use a loss function combining cross-entropy loss and triplet loss to make the best feature learning. Experiments show that our method performs well in the Market-to-Duke and Duke-to-Market cross-domain recognition tasks.