对比了多种文本分类方法,预训练模型表现最优。
Text Classification: Neural Networks VS Machine Learning Models VS Pre-trained Models
- 比较七种预训练模型、三种神经网络和三种机器学习模型。
- GloVe嵌入优于TF-IDF,BERT/DistilBERT显著超越其他方法。
- 适合关注NLP模型选型的开发者与研究者参考。
文本分类是当前常见任务,已有多种高效方法可用。变压器(Transformer)架构通过自注意力机制,在自然语言处理中引发变革,能更有效捕捉长距离依赖关系,优于传统神经网络(如RNN、MLP)。本文对比了七种预训练模型、三种标准神经网络和三种机器学习模型。对神经网络与机器学习模型,还比较了两种嵌入技术:TF-IDF与GloVe,结果显示GloVe表现更优。实验表明,BERT、DistilBERT等预训练模型始终优于标准模型与算法。
原文摘要 · Abstract (English)
Text classification is a very common task nowadays and there are many efficient methods and algorithms that we can employ to accomplish it. Transformers have revolutionized the field of deep learning, particularly in Natural Language Processing (NLP) and have rapidly expanded to other domains such as computer vision, time-series analysis and more. The transformer model was firstly introduced in the context of machine translation and its architecture relies on self-attention mechanisms to capture complex relationships within data sequences. It is able to handle long-range dependencies more effectively than traditional neural networks (such as Recurrent Neural Networks and Multilayer Perceptrons). In this work, we present a comparison between different techniques to perform text classification. We take into consideration seven pre-trained models, three standard neural networks and three machine learning models. For standard neural networks and machine learning models we also compare two embedding techniques: TF-IDF and GloVe, with the latter consistently outperforming the former. Finally, we demonstrate the results from our experiments where pre-trained models such as BERT and DistilBERT always perform better than standard models/algorithms.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。