Chen, M; Wang, G; Xue, J-H; Ding, Z; Sun, L; (2021) Enhance Via Decoupling: Improving Multi-Label Classifiers With Variational Feature Augmentation. In: Proceedings of the IEEE International Conference on Image Processing (ICIP) 2021. (pp. pp. 1329-1333). Institute of Electrical and Electronics Engineers (IEEE)

Preview

Text MingChen-ICIP2021-accepted.pdf - Accepted Version Download (729kB) | Preview

Abstract

Multi-label classification remains a challenging problem due to the inherent label imbalance issue, which brings overfitting of minor categories to modern deep models. In this paper, to tackle this issue, we propose a novel method named Variational Feature Augmentation (VFA) to enhance the deep neural networks for multi-label classification. Our method decouples the feature vectors extracted by the backbone network into multiple low-dimensional spaces via a novely proposed Variational Feature Decoupling Module. The decoupled feature vectors are then re-combined with a shuffle operation and a Feature Augmentation Layer to enrich the minor co-occurrence relations, mitigating the label imbalance. Different from most other methods, VFA does not modify the network architecture or introduce extra computation cost in inference phase. We conduct comprehensive experiments on four benchmarks of two visual multi-label classification tasks, pedestrian attribute recognition and multi-label image recognition, and the results demonstrate the effectiveness and generality of the proposed VFA.

Download activity - last month

Export as CSVXMLJSON

Download activity - last 12 months

Export as CSVXMLJSON

Downloads by country - last 12 months

Export as CSVXMLJSON

Archive Staff Only

View Item