Utterance-to-Utterance Interactive Matching Network for Multi-Turn Response Selection in Retrieval-Based Chatbots

This repository contains the source code and datasets for the IEEE/ACM Transactions on Audio, Speech and Language Processing paper Utterance-to-Utterance Interactive Matching Network for Multi-Turn Response Selection in Retrieval-Based Chatbots by Gu et al.

Dependencies

Python 2.7
Tensorflow 1.4.0

Datasets

Your can download the processed datasets used in our paper here and unzip it to the folder of data.
Ubuntu_V2

Train a new model

cd scripts
bash ubuntu_train.sh

The training process is recorded in log_train_UbuntuV2.txt file.

Test a trained model

bash ubuntu_test.sh

The testing process is recorded in log_test_UbuntuV2.txt file. And your can get a ubuntu_test_out.txt file which records scores for each context-response pair. Run the following command your can compute the metric of Recall.

python compute_recall.py

Cite

If you use the code and datasets, please cite the following paper: "Utterance-to-Utterance Interactive Matching Network for Multi-Turn Response Selection in Retrieval-Based Chatbots" Jia-Chen Gu, Zhen-Hua Ling, Quan Liu. TASLP

@ARTICLE{8910440,
         author={J. {Gu} and Z. {Ling} and Q. {Liu}},
         journal={IEEE/ACM Transactions on Audio, Speech, and Language Processing},
         title={Utterance-to-Utterance Interactive Matching Network for Multi-Turn Response Selection in Retrieval-Based Chatbots},
         year={2020},
         volume={28},
         number={},
         pages={369-379},
         keywords={Context modeling;Encoding;Buildings;Neural networks;Training;Dialogue;response selection;interactive matching network;utterance-to-utterance},
         doi={10.1109/TASLP.2019.2955290},
         ISSN={2329-9304},
         month={},}
}