High-order diversity feature learning for pedestrian attribute recognition.

Journal: Neural networks : the official journal of the International Neural Network Society
Published Date:

Abstract

Pedestrian attribute recognition (PAR) involves accurately identifying multiple attributes present in pedestrian images. There are two main approaches for PAR: part-based method and attention-based method. The former relies on existing segmentation or region detection methods to localize body parts and learn corresponding attribute-specific feature from the corresponding regions, where the performance heavily depends on the accuracy of body region localization. The latter adopts the embedded attention modules or transformer attention to exploit detailed feature. However, it can focus on certain body regions but often provide coarse attention, failing to capture fine-grained details, the learned feature may also be interfered with by irrelevant information. Meanwhile, these methods overlook the global contextual information. This work argues for replacing coarse attention with detailed attention and integrating it with global contextual feature from ViT to jointly represent attribute-specific regions. To tackle this issue, we propose a High-order Diversity Feature Learning (HDFL) method for PAR based on ViT. We utilize a polynomial predictor to design an Attribute-specific Detailed Feature Exploration (ADFE) module, which can construct the high-order statistics and gain more fine-grained feature. Our ADFE module is a parameter-friendly method that provides flexibility in deciding its utilization during the inference phase. A Soft-redundancy Perception Loss (SPLoss) is proposed to adaptively measure the redundancy between feature of different orders, which can promote diverse characterization of features. Experiments on several PAR datasets show that our method achieves a new state-of-the-art (SOTA) performance. On the most challenging PA100K dataset, our method outperforms previous SOTA by 1.69% and achieves the highest mA of 84.92%.

Authors

  • Junyi Wu
  • Yan Huang
    Department of Neurology, University of Texas Health Science Center at Houston, Houston, TX.
  • Min Gao
    Department of Biliary Surgery, West China Hospital of Sichuan University, Chengdu, China.
  • Yuzhen Niu
    Shandong Provincial Research Center for Bioinformatic Engineering and Technique, School of Life Sciences, Shandong University of Technology, Zibo 255049, China.
  • Yuzhong Chen
    School of Material Science and Engineering, Shandong University, Jinan, China.
  • Qiang Wu
    Department of Radiotherapy, West China Hospital, Sichuan University, Chengdu, PR China; Lung Cancer Center, West China Hospital, Sichuan University, Chengdu, PR China.