← 返回资源分享
TransferTransfo: A Transfer Learning Approach for Neural Network Based Conversational Agents
论文
论文
发布时间2019-01-23
发表arXiv:1901.08149
作者:Thomas Wolf,Julien Chaumond,Clement Delangue,Victor Sanh
详细介绍
We introduce a new approach to generative data-driven dialogue systems (e.g.
chatbots) called TransferTransfo which is a combination of a Transfer learning
based training scheme and a high-capacity Transformer model. Fine-tuning is
performed by using a multi-task objective which combines several unsupervised
prediction tasks. The resulting fine-tuned model shows strong improvements over
the current state-of-the-art end-to-end conversational models like memory
augmented seq2seq and information-retrieval models. On the privately held
PERSONA-CHAT dataset of the Conversational Intelligence Challenge 2, this
approach obtains a new state-of-the-art, with respective perplexity, Hits@1 and
F1 metrics of 16.28 (45 % absolute improvement), 80.7 (46 % absolute
improvement) and 19.5 (20 % absolute improvement).
代码仓库 (22)
ErikEkstedt/TurnGPT官方PyTorch
cerebroai/AskItPyTorch
liliaellouz/wabi-sabiPyTorch
xnancy/convai-transformerPyTorch
Vergangenheit/Conv-AIPyTorch
the-pythoncoder/councelChatbot2PyTorch
CP8315-NNP-JLOPEZ/NNP-ChatbotPyTorch
noriyukipy/gptchatPyTorch
jiangjyjy/CDialGPTPyTorch
the-pythoncoder/counsel-chatPyTorch
