Exploring Dense Retrieval for Dialogue Response Selection

Tian Lan, Deng Cai, Yan Wang, Yixuan Su, Heyan Huang, Xian Ling Mao*

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

2 Citations (Scopus)

Abstract

Recent progress in deep learning has continuously improved the accuracy of dialogue response selection. However, in real-world scenarios, the high computation cost forces existing dialogue response selection models to rank only a small number of candidates, recalled by a coarse-grained model, precluding many high-quality candidates. To overcome this problem, we present a novel and efficient response selection model and a set of tailor-designed learning strategies to train it effectively. The proposed model consists of a dense retrieval module and an interaction layer, which could directly select the proper response from a large corpus. We conduct re-rank and full-rank evaluations on widely used benchmarks to evaluate our proposed model. Extensive experimental results demonstrate that our proposed model notably outperforms the state-of-the-art baselines on both re-rank and full-rank evaluations. Moreover, human evaluation results show that the response quality could be improved further by enlarging the candidate pool with nonparallel corpora. In addition, we also release high-quality benchmarks that are carefully annotated for more accurate dialogue response selection evaluation. All source codes, datasets, model parameters, and other related resources have been publicly available.1

Original languageEnglish
Article number84
JournalACM Transactions on Information Systems
Volume42
Issue number3
DOIs
Publication statusPublished - 22 Jan 2024

Keywords

  • Retrieval-based dialogue system
  • deep semantic hashing
  • dense retrieval
  • dialogue evaluation

Fingerprint

Dive into the research topics of 'Exploring Dense Retrieval for Dialogue Response Selection'. Together they form a unique fingerprint.

Cite this