Gated Value Network for Multilabel Classification.

Yimin Hou,Sen Wan,Feng Bao,Zhiquan Ren,Yunfeng Dong,Qionghai Dai,Yue Deng
DOI: https://doi.org/10.1109/tnnls.2020.3019804
IF: 14.255
2021-01-01
IEEE Transactions on Neural Networks and Learning Systems
Abstract:We introduce a gated value network (GVN) for general multilabel classification (MLC) tasks. GVN was motivated by deep value network (DVN) that directly exploits the "compatibility" metric as the learning pursuit for MLC. Meanwhile, it further improves traditional DVN on twofold. First, GVN relaxes the complex variable optimization steps in DVN inference by incorporating a feedforward predictor for straightforward multilabel prediction. Second, GVN also introduces the gating mechanism to block confounding factors from the input data that allows more precise compatibility evaluations for data and their potential multilabels. The whole GVN framework is trained in an end-to-end manner with policy gradient approaches. We show the effectiveness and generalization of GVN on diverse learning tasks, including document classification, audio tagging, and image attribute prediction.
What problem does this paper attempt to address?