4 个月前

多任务深度神经网络在自然语言理解中的应用

多任务深度神经网络在自然语言理解中的应用

摘要

在本文中,我们介绍了一种多任务深度神经网络(MT-DNN),用于跨多个自然语言理解(NLU)任务学习表示。MT-DNN不仅利用了大量跨任务数据,还受益于一种正则化效应,从而生成更为通用的表示,以便更好地适应新任务和新领域。MT-DNN扩展了Liu等人(2015)提出的模型,通过引入预训练的双向Transformer语言模型,即BERT(Devlin等人,2018)。MT-DNN在十个NLU任务上取得了新的最先进结果,包括SNLI、SciTail以及GLUE基准中的九个任务中的八个,将GLUE基准分数提升至82.7%(绝对提升了2.2%)。我们还使用SNLI和SciTail数据集证明了MT-DNN所学习到的表示能够在显著减少领域内标签的情况下实现领域适应。代码和预训练模型已在https://github.com/namisan/mt-dnn公开发布。

代码仓库

phueb/BabyBertSRL
pytorch
GitHub 中提及
gaohuan2015/NLPTool
pytorch
GitHub 中提及
namisan/mt-dnn
官方
pytorch
GitHub 中提及
om00839/machine-suneung
pytorch
GitHub 中提及
ABaldrati/MT-BERT
pytorch
GitHub 中提及
xycforgithub/MultiTask-MRC
pytorch
GitHub 中提及
phueb/CHILDES-SRL
pytorch
GitHub 中提及

基准测试

基准方法指标
linguistic-acceptability-on-colaMT-DNN
Accuracy: 68.4%
natural-language-inference-on-multinliMT-DNN
Matched: 86.7
Mismatched: 86.0
natural-language-inference-on-scitailMT-DNN
Accuracy: 94.1
natural-language-inference-on-snliMT-DNN
% Test Accuracy: 91.6
% Train Accuracy: 97.2
Parameters: 330m
natural-language-inference-on-snliNtumpha
% Test Accuracy: 90.5
% Train Accuracy: 99.1
Parameters: 220
paraphrase-identification-on-quora-questionMT-DNN
Accuracy: 89.6
F1: 72.4
sentiment-analysis-on-sst-2-binaryMT-DNN
Accuracy: 95.6

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
多任务深度神经网络在自然语言理解中的应用 | 论文 | HyperAI超神经