One Stone, Four Birds: A Comprehensive Solution for QA System Using Supervised Contrastive Learning

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Wang, Bo, Mine, Tsunenori
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866910666213818368
author Wang, Bo
Mine, Tsunenori
author_facet Wang, Bo
Mine, Tsunenori
contents This paper presents a novel and comprehensive solution to enhance both the robustness and efficiency of question answering (QA) systems through supervised contrastive learning (SCL). Training a high-performance QA system has become straightforward with pre-trained language models, requiring only a small amount of data and simple fine-tuning. However, despite recent advances, existing QA systems still exhibit significant deficiencies in functionality and training efficiency. We address the functionality issue by defining four key tasks: user input intent classification, out-of-domain input detection, new intent discovery, and continual learning. We then leverage a unified SCL-based representation learning method to efficiently build an intra-class compact and inter-class scattered feature space, facilitating both known intent classification and unknown intent detection and discovery. Consequently, with minimal additional tuning on downstream tasks, our approach significantly improves model efficiency and achieves new state-of-the-art performance across all tasks.
format Preprint
id arxiv_https___arxiv_org_abs_2407_09011
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle One Stone, Four Birds: A Comprehensive Solution for QA System Using Supervised Contrastive Learning
Wang, Bo
Mine, Tsunenori
Computation and Language
Artificial Intelligence
Machine Learning
This paper presents a novel and comprehensive solution to enhance both the robustness and efficiency of question answering (QA) systems through supervised contrastive learning (SCL). Training a high-performance QA system has become straightforward with pre-trained language models, requiring only a small amount of data and simple fine-tuning. However, despite recent advances, existing QA systems still exhibit significant deficiencies in functionality and training efficiency. We address the functionality issue by defining four key tasks: user input intent classification, out-of-domain input detection, new intent discovery, and continual learning. We then leverage a unified SCL-based representation learning method to efficiently build an intra-class compact and inter-class scattered feature space, facilitating both known intent classification and unknown intent detection and discovery. Consequently, with minimal additional tuning on downstream tasks, our approach significantly improves model efficiency and achieves new state-of-the-art performance across all tasks.
title One Stone, Four Birds: A Comprehensive Solution for QA System Using Supervised Contrastive Learning
topic Computation and Language
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2407.09011