arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2003.11184cs.CLcs.LG
收藏

用于多类分类的对抗性多二进制神经网络

Adversarial Multi-Binary Neural Network for Multi-class Classification

AI Labs, Didi Chuxing(滴滴出行人工智能实验室);AI Labs, Didi Research America(滴滴美国研究院人工智能实验室)

Haiyang Xu, Junwen Chen, Kun Han, Xiangang Li

摘要:

多类文本分类是机器学习和自然语言处理的关键问题之一。新兴的神经网络使用多输出softmax层来处理这个问题,并取得了实质性的进展,但它们没有显式地学习类之间的相关性。在本文中,我们使用多任务框架来解决多类分类问题,其中多类分类器和多个二进制分类器一起训练。此外,我们使用对抗训练来区分类特定的特征和类不可知的特征。该模型受益于更好的特征表示。我们进行了两个大规模的多类文本分类任务的实验,并证明所提出的架构优于基线方法。

英文摘要:

Multi-class text classification is one of the key problems in machine learning and natural language processing. Emerging neural networks deal with the problem using a multi-output softmax layer and achieve substantial progress, but they do not explicitly learn the correlation among classes. In this paper, we use a multi-task framework to address multi-class classification, where a multi-class classifier and multiple binary classifiers are trained together. Moreover, we employ adversarial training to distinguish the class-specific features and the class-agnostic features. The model benefits from better feature representation. We conduct experiments on two large-scale multi-class text classification tasks and demonstrate that the proposed architecture outperforms baseline approaches.

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

↑