Skip to main navigation Skip to search Skip to main content

Pedestrian Attribute Recognition as Label-balanced Multi-label Learning

  • Yibo Zhou
  • , Hai Miao Hu*
  • , Yirong Xiang
  • , Xiaokang Zhang
  • , Haotian Wu
  • *Corresponding author for this work
  • Beihang University
  • University of Manchester

Research output: Contribution to journalConference articlepeer-review

Abstract

Rooting in the scarcity of most attributes, realistic pedestrian attribute datasets exhibit unduly skewed data distribution, from which two types of model failures are delivered: (1) label imbalance: model predictions lean greatly towards the side of majority labels; (2) semantics imbalance: model is easily overfitted on the under-represented attributes due to their insufficient semantic diversity.To render perfect label balancing, we propose a novel framework that successfully decouples label-balanced data re-sampling from the curse of attributes co-occurrence, i.e., we equalize the sampling prior of an attribute while not biasing that of the co-occurred others.To diversify the attributes semantics and mitigate the feature noise, we propose a Bayesian feature augmentation method to introduce true in-distribution novelty.Handling both imbalances jointly, our work achieves best accuracy on various popular benchmarks, and importantly, with minimal computational budget.

Original languageEnglish
Pages (from-to)61964-61978
Number of pages15
JournalProceedings of Machine Learning Research
Volume235
StatePublished - 2024
Event41st International Conference on Machine Learning, ICML 2024 - Vienna, Austria
Duration: 21 Jul 202427 Jul 2024

Fingerprint

Dive into the research topics of 'Pedestrian Attribute Recognition as Label-balanced Multi-label Learning'. Together they form a unique fingerprint.

Cite this