arXiv:2409.05972cs.CLcs.LG2024-09被引 1

小数据下法律文本分类,传统模型反而比BERT更有效。

A Small Claims Court for the NLP: Judging Legal Text Classification Strategies With Small Datasets

  • 用小样本+无标签数据,对比多种模型在法律文本分类中的表现。
  • 词向量+逻辑回归等经典模型准确率达80.7%,优于BERT。
  • 适合资源有限的法律领域研究者参考。

近年来,语言建模的发展显著降低了文本分类对标注数据的需求。基于Transformer的模型在未标注数据上预训练后,可超越从零开始训练的模型。然而,在需要专家标注的法律领域,微调此类模型仍需大量标注数据。本文研究如何优化小规模标注数据与大量无标签数据的使用,在巴西检察院诉讼记录中对50个预定义主题进行文本分类,目标是自动填充当前需深度法律知识的人工任务。该任务因葡萄牙语法律资源稀缺而更具挑战性。结果表明,逻辑回归、SVM及随机森林、梯度提升等经典监督模型配合word2vec词向量,性能优于BERT模型。但若将BERT作为分类器架构使用,则表现最佳。最优结果来自无监督数据增强(UDA)方法,结合BERT、数据增强与半监督学习,达到80.7%的准确率。

原文摘要 · Abstract (English)

Recent advances in language modelling has significantly decreased the need of labelled data in text classification tasks. Transformer-based models, pre-trained on unlabeled data, can outmatch the performance of models trained from scratch for each task. However, the amount of labelled data need to fine-tune such type of model is still considerably high for domains requiring expert-level annotators, like the legal domain. This paper investigates the best strategies for optimizing the use of a small labeled dataset and large amounts of unlabeled data and perform a classification task in the legal area with 50 predefined topics. More specifically, we use the records of demands to a Brazilian Public Prosecutor's Office aiming to assign the descriptions in one of the subjects, which currently demands deep legal knowledge for manual filling. The task of optimizing the performance of classifiers in this scenario is especially challenging, given the low amount of resources available regarding the Portuguese language, especially in the legal domain. Our results demonstrate that classic supervised models such as logistic regression and SVM and the ensembles random forest and gradient boosting achieve better performance along with embeddings extracted with word2vec when compared to BERT language model. The latter demonstrates superior performance in association with the architecture of the model itself as a classifier, having surpassed all previous models in that regard. The best result was obtained with Unsupervised Data Augmentation (UDA), which jointly uses BERT, data augmentation, and strategies of semi-supervised learning, with an accuracy of 80.7% in the aforementioned task.

法律文本小样本BERT分类

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。