arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2501.08035cs.CLcs.AI

READ:面向有限标注数据下文本分类的基于强化的对抗学习

READ: Reinforcement-based Adversarial Learning for Text Classification with Limited Labeled Data

Microsoft Corporation(微软公司)

Rohit Sharma, Shanu Kumar, Avinash Kumar

AI总结:

针对预训练Transformer模型文本分类需大量标注数据的问题,提出READ方法,融合强化学习文本生成与半监督对抗学习,利用无标注数据生成多样合成文本提升泛化性,在多数据集上性能超越现有最优方法。

摘要:

诸如BERT之类的预训练Transformer模型已在众多文本分类任务中取得大幅进展。但这类模型通常需要海量标注数据才能实现优异性能。获取标注数据往往成本高昂且耗时,而对任意任务而言,借助一些启发式规则收集无标注数据的成本要低得多。因此,本文提出一种方法,以新颖方式融合基于强化学习的文本生成与半监督对抗学习方法,以提升模型性能。我们的方法READ(基于强化的对抗学习)利用无标注数据集,通过强化学习生成多样化的合成文本,再借助对抗学习提升模型的泛化能力。实验结果表明,READ在多个数据集上的表现优于现有最先进方法。

英文摘要:

Pre-trained transformer models such as BERT have shown massive gains across many text classification tasks. However, these models usually need enormous labeled data to achieve impressive performances. Obtaining labeled data is often expensive and time-consuming, whereas collecting unlabeled data using some heuristics is relatively much cheaper for any task. Therefore, this paper proposes a method that encapsulates reinforcement learning-based text generation and semi-supervised adversarial learning approaches in a novel way to improve the model's performance. Our method READ, Reinforcement-based Adversarial learning, utilizes an unlabeled dataset to generate diverse synthetic text through reinforcement learning, improving the model's generalization capability using adversarial learning. Our experimental results show that READ outperforms the existing state-of-art methods on multiple datasets.

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

↑