search

chaitjo / personalized-dialog

Code for the paper 'Personalization in Goal-oriented Dialog' (NeurIPS 2017 Conversational AI Workshop)
https://chaitjo.github.io/personalization-in-dialog/
MIT License
132 stars 45 forks source link

what's the different between 'MemN2N' and 'MemN2N-modified', #2

Closed SeekPoint closed 7 years ago

SeekPoint commented 7 years ago

if I want work on on of them for study and improvement purpose, on which one you suggest based?

chaitjo commented 7 years ago

Hey! Thanks for the interest. The MemN2N model is the one described in FAIR's bAbI Dialog Tasks paper as well as in our paper on arXiv. The MemN2N-modified model is a new model we are currently working on and intend to add to the paper.

The code for the modified model is still not completely finalized, but a rough idea is to split the model's memory into two- one for the profile attributes and one for the conversation history. The model then combines the outputs over these two memories to produce the final output.

You can use either, as the code has not changed a lot between the two models.

SeekPoint commented 7 years ago

great! I will check both of them

chaitjo commented 6 years ago

The modified model, called the split memory architecture, has been added in an updated version of the paper on arXiv: https://arxiv.org/abs/1706.07503

SeekPoint commented 6 years ago

great job