Unleashing the Potential of Pre-Trained Diffusion Models for Generalizable Person Re-Identification

Domain-generalizable re-identification (DG Re-ID) aims to train a model on one or more source domains and evaluate its performance on unseen target domains, a task that has attracted growing attention due to its practical relevance. While numerous methods have been proposed, most rely on discriminat...

Full description

Saved in:
Bibliographic Details
Main Authors: Jiachen Li, Xiaojin Gong
Format: Article
Language:English
Published: MDPI AG 2025-01-01
Series:Sensors
Subjects:
Online Access:https://www.mdpi.com/1424-8220/25/2/552
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1832587514227982336
author Jiachen Li
Xiaojin Gong
author_facet Jiachen Li
Xiaojin Gong
author_sort Jiachen Li
collection DOAJ
description Domain-generalizable re-identification (DG Re-ID) aims to train a model on one or more source domains and evaluate its performance on unseen target domains, a task that has attracted growing attention due to its practical relevance. While numerous methods have been proposed, most rely on discriminative or contrastive learning frameworks to learn generalizable feature representations. However, these approaches often fail to mitigate shortcut learning, leading to suboptimal performance. In this work, we propose a novel method called diffusion model-assisted representation learning with a correlation-aware conditioning scheme (DCAC) to enhance DG Re-ID. Our method integrates a discriminative and contrastive Re-ID model with a pre-trained diffusion model through a correlation-aware conditioning scheme. By incorporating ID classification probabilities generated from the Re-ID model with a set of learnable ID-wise prompts, the conditioning scheme injects dark knowledge that captures ID correlations to guide the diffusion process. Simultaneously, feedback from the diffusion model is back-propagated through the conditioning scheme to the Re-ID model, effectively improving the generalization capability of Re-ID features. Extensive experiments on both single-source and multi-source DG Re-ID tasks demonstrate that our method achieves state-of-the-art performance. Comprehensive ablation studies further validate the effectiveness of the proposed approach, providing insights into its robustness.
format Article
id doaj-art-9c8e695ada9f4e5888b9868e6b151dee
institution Kabale University
issn 1424-8220
language English
publishDate 2025-01-01
publisher MDPI AG
record_format Article
series Sensors
spelling doaj-art-9c8e695ada9f4e5888b9868e6b151dee2025-01-24T13:49:19ZengMDPI AGSensors1424-82202025-01-0125255210.3390/s25020552Unleashing the Potential of Pre-Trained Diffusion Models for Generalizable Person Re-IdentificationJiachen Li0Xiaojin Gong1College of Information Science and Electronic Engineering, Zhejiang University, Hangzhou 310027, ChinaCollege of Information Science and Electronic Engineering, Zhejiang University, Hangzhou 310027, ChinaDomain-generalizable re-identification (DG Re-ID) aims to train a model on one or more source domains and evaluate its performance on unseen target domains, a task that has attracted growing attention due to its practical relevance. While numerous methods have been proposed, most rely on discriminative or contrastive learning frameworks to learn generalizable feature representations. However, these approaches often fail to mitigate shortcut learning, leading to suboptimal performance. In this work, we propose a novel method called diffusion model-assisted representation learning with a correlation-aware conditioning scheme (DCAC) to enhance DG Re-ID. Our method integrates a discriminative and contrastive Re-ID model with a pre-trained diffusion model through a correlation-aware conditioning scheme. By incorporating ID classification probabilities generated from the Re-ID model with a set of learnable ID-wise prompts, the conditioning scheme injects dark knowledge that captures ID correlations to guide the diffusion process. Simultaneously, feedback from the diffusion model is back-propagated through the conditioning scheme to the Re-ID model, effectively improving the generalization capability of Re-ID features. Extensive experiments on both single-source and multi-source DG Re-ID tasks demonstrate that our method achieves state-of-the-art performance. Comprehensive ablation studies further validate the effectiveness of the proposed approach, providing insights into its robustness.https://www.mdpi.com/1424-8220/25/2/552generalizable person re-identificationdiffusion modelprompt learningCLIP
spellingShingle Jiachen Li
Xiaojin Gong
Unleashing the Potential of Pre-Trained Diffusion Models for Generalizable Person Re-Identification
Sensors
generalizable person re-identification
diffusion model
prompt learning
CLIP
title Unleashing the Potential of Pre-Trained Diffusion Models for Generalizable Person Re-Identification
title_full Unleashing the Potential of Pre-Trained Diffusion Models for Generalizable Person Re-Identification
title_fullStr Unleashing the Potential of Pre-Trained Diffusion Models for Generalizable Person Re-Identification
title_full_unstemmed Unleashing the Potential of Pre-Trained Diffusion Models for Generalizable Person Re-Identification
title_short Unleashing the Potential of Pre-Trained Diffusion Models for Generalizable Person Re-Identification
title_sort unleashing the potential of pre trained diffusion models for generalizable person re identification
topic generalizable person re-identification
diffusion model
prompt learning
CLIP
url https://www.mdpi.com/1424-8220/25/2/552
work_keys_str_mv AT jiachenli unleashingthepotentialofpretraineddiffusionmodelsforgeneralizablepersonreidentification
AT xiaojingong unleashingthepotentialofpretraineddiffusionmodelsforgeneralizablepersonreidentification