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

التفاصيل البيبلوغرافية
العنوان: One Stone, Four Birds: A Comprehensive Solution for QA System Using Supervised Contrastive Learning
المؤلفون: Wang, Bo, Mine, Tsunenori
سنة النشر: 2024
المجموعة: Computer Science
مصطلحات موضوعية: Computer Science - Computation and Language, Computer Science - Artificial Intelligence, Computer Science - 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.
Comment: 14 pages, under review
نوع الوثيقة: Working Paper
URL الوصول: http://arxiv.org/abs/2407.09011
رقم الأكسشن: edsarx.2407.09011
قاعدة البيانات: arXiv