Personalizing Dialogue Agents: I have a dog, do you have pets too?
Saizheng Zhang, Emily Dinan, Jack Urbanek, Arthur Szlam, Douwe Kiela, Jason Weston
Code Available — Be the first to reproduce this paper.
ReproduceCode
- github.com/facebookresearch/ParlAIOfficialIn paperpytorch★ 10,627
- github.com/ThilinaRajapakse/simpletransformerspytorch★ 4,235
- github.com/SivilTaram/Persona-Dialogue-Generationpytorch★ 307
- github.com/devjwsong/gpt2-dialogue-generation-pytorchpytorch★ 176
- github.com/devJWSong/gpt2-chatbot-pytorchpytorch★ 176
- github.com/af1tang/personaGPTpytorch★ 110
- github.com/illidanlab/personaGPTpytorch★ 110
- github.com/lemuria-wchen/DialogVEDpytorch★ 39
- github.com/passing2961/personachatgennone★ 13
- github.com/hkmirza/eacl2024-sci-chat-sharedtasknone★ 0
Abstract
Chit-chat models are known to have several problems: they lack specificity, do not display a consistent personality and are often not very captivating. In this work we present the task of making chit-chat more engaging by conditioning on profile information. We collect data and train models to (i) condition on their given profile information; and (ii) information about the person they are talking to, resulting in improved dialogues, as measured by next utterance prediction. Since (ii) is initially unknown our model is trained to engage its partner with personal topics, and we show the resulting dialogue can be used to predict profile information about the interlocutors.
Tasks
Benchmark Results
| Dataset | Model | Metric | Claimed | Verified | Status |
|---|---|---|---|---|---|
| Persona-Chat | KV Profile Memory | Avg F1 | 11.9 | — | Unverified |