Effective Collaborative Representation Learning for Multilabel Text Categorization

Hao Wu

doi:10.1109/TNNLS.2021.3069647

Effective Collaborative Representation Learning for Multilabel Text Categorization

2022-10-01

18

PRE

AI

DOI：10.1109/TNNLS.2021.3069647

Original

Original request for help

Save

Abstract

En 中文

With the booming of deep learning, massive attention has been paid to developing neural models for multilabel text categorization (MLTC). Most of the works concentrate on disclosing word-label relationship, while less attention is taken in exploiting global clues, particularly with the relationship of document-label. To address this limitation, we propose an effective collaborative representation learning (CRL) model in this article. CRL consists of a factorization component for generating shallow representations of documents and a neural component for deep text-encoding and classification. We have developed strategies for jointly training those two components, including an alternating-least-squares-based approach for factorizing the pointwise mutual information (PMI) matrix of label-document and multitask learning (MTL) strategy for the neural component. According to the experimental results on six data sets, CRL can explicitly take advantage of the relationship of document-label and achieve competitive classification performance in comparison with some state-of-the-art deep methods.

Keywords:

Text categorization

Training

Predictive models

Electronic mail

Data models

Collaboration

Biological system modeling

Collaborative representation learning (CRL)

matrix factorization

multitask learning (MTL)

neural networks

text categorization

Effective Collaborative Representation Learning for Multilabel Text Categorization

Abstract

AI Summary

Journal

Organization