HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

Dialogue Response Selection with Hierarchical Curriculum Learning

Yixuan Su; Deng Cai; Qingyu Zhou; Zibo Lin; Simon Baker; Yunbo Cao; Shuming Shi; Nigel Collier; Yan Wang

Dialogue Response Selection with Hierarchical Curriculum Learning

Abstract

We study the learning of a matching model for dialogue response selection. Motivated by the recent finding that models trained with random negative samples are not ideal in real-world scenarios, we propose a hierarchical curriculum learning framework that trains the matching model in an "easy-to-difficult" scheme. Our learning framework consists of two complementary curricula: (1) corpus-level curriculum (CC); and (2) instance-level curriculum (IC). In CC, the model gradually increases its ability in finding the matching clues between the dialogue context and a response candidate. As for IC, it progressively strengthens the model's ability in identifying the mismatching information between the dialogue context and a response candidate. Empirical studies on three benchmark datasets with three state-of-the-art matching models demonstrate that the proposed learning framework significantly improves the model performance across various evaluation metrics.

Code Repositories

yxuansu/HCL
Official
pytorch

Benchmarks

BenchmarkMethodologyMetrics
conversational-response-selection-on-douban-1SA-BERT+HCL
MAP: 0.639
MRR: 0.681
P@1: 0.514
R10@1: 0.330
R10@2: 0.531
R10@5: 0.858
conversational-response-selection-on-eSA-BERT+HCL
R10@1: 0.721
R10@2: 0.896
R10@5: 0.993
conversational-response-selection-on-rrsSA-BERT+HCL
MAP: 0.671
MRR: 0.683
P@1: 0.503
R10@1: 0.454
R10@2: 0.659
R10@5: 0.917

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