HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Fine-grained Post-training for Improving Retrieval-based Dialogue Systems

{Jungyun Seo Youngjoong Ko Byoungjae Kim Taesuk Hong Janghoon Han}

Fine-grained Post-training for Improving Retrieval-based Dialogue Systems

Abstract

Retrieval-based dialogue systems display an outstanding performance when pre-trained language models are used, which includes bidirectional encoder representations from transformers (BERT). During the multi-turn response selection, BERT focuses on training the relationship between the context with multiple utterances and the response. However, this method of training is insufficient when considering the relations between each utterance in the context. This leads to a problem of not completely understanding the context flow that is required to select a response. To address this issue, we propose a new fine-grained post-training method that reflects the characteristics of the multi-turn dialogue. Specifically, the model learns the utterance level interactions by training every short context-response pair in a dialogue session. Furthermore, by using a new training objective, the utterance relevance classification, the model understands the semantic relevance and coherence between the dialogue utterances. Experimental results show that our model achieves new state-of-the-art with significant margins on three benchmark datasets. This suggests that the fine-grained post-training method is highly effective for the response selection task.

Benchmarks

BenchmarkMethodologyMetrics
conversational-response-selection-on-douban-1BERT-FP
MAP: 0.644
MRR: 0.680
P@1: 0.512
R10@1: 0.324
R10@2: 0.542
R10@5: 0.870
conversational-response-selection-on-eBERT-FP
R10@1: 0.870
R10@2: 0.956
R10@5: 0.993
conversational-response-selection-on-rrsBERT-FP
MAP: 0.702
MRR: 0.712
P@1: 0.543
R10@1: 0.488
R10@2: 0.708
R10@5: 0.927
conversational-response-selection-on-rrs-1BERT-FP
NDCG@3: 0.609
NDCG@5: 0.709
conversational-response-selection-on-ubuntu-1BERT-FP
R10@1: 0.911
R10@2: 0.962
R10@5: 0.994

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp