Generative Multi-task Learning for Text Classification

Wei Zhao, Hui Gao, Shuhui Chen, Nan Wang
2020 IEEE Access  
Multi-task learning leverages potential correlations among related tasks to extract common features and yield performance gains. In this paper, a generative multi-task learning (MTL) approach for text classification and categorization is proposed, which is composed of a shared encoder, a multilabel classification decoder and a hierarchical categorization decoder. In the two decoders, a label-orderindependent multi-label classification loss function and a hierarchical structure mask matrix are
more » ... troduced. Experiments conducted on the real-world public security dataset show that the proposed approach has obvious advantages over the baseline approaches and can enhance the semantic association between the results of the classification and categorization tasks. INDEX TERMS Generative model, hierarchical classification, multi-label classification, multi-task learning. This work is licensed under a Creative Commons Attribution 4.0 License. For more information, see https://creativecommons.org/licenses/by/4.0/ VOLUME 8, 2020
doi:10.1109/access.2020.2991337 fatcat:bdfsl7hrhjgntomapzezuy4hye