arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2007.02439cs.LGstat.ML

Pretrained Generalized Autoregressive Model with Adaptive Probabilistic Label Clusters for Extreme Multi-label Text Classification

Lehigh University(理海大学);Xiamen University of Technology(厦门理工学院)

Hui Ye, Zhiyu Chen, Da-Han Wang, Brian D. Davison

英文摘要:

Extreme multi-label text classification (XMTC) is a task for tagging a given text with the most relevant labels from an extremely large label set. We propose a novel deep learning method called APLC-XLNet. Our approach fine-tunes the recently released generalized autoregressive pretrained model (XLNet) to learn a dense representation for the input text. We propose Adaptive Probabilistic Label Clusters (APLC) to approximate the cross entropy loss by exploiting the unbalanced label distribution to form clusters that explicitly reduce the computational time. Our experiments, carried out on five benchmark datasets, show that our approach has achieved new state-of-the-art results on four benchmark datasets. Our source code is available publicly at https://github.com/huiyegit/APLC_XLNet.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑