Knowledge Augmented Finetuning Matters in both RAG and Agent Based Dialog Systems

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Cai, Yucheng, Wu, Yuxuan, Huang, Yi, Feng, Junlan, Ou, Zhijian
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916814871592960
author Cai, Yucheng
Wu, Yuxuan
Huang, Yi
Feng, Junlan
Ou, Zhijian
author_facet Cai, Yucheng
Wu, Yuxuan
Huang, Yi
Feng, Junlan
Ou, Zhijian
contents Large language models (LLMs) have recently been applied to dialog systems. Despite making progress, LLMs are prone to errors in knowledge-intensive scenarios. Recently, approaches based on retrieval augmented generation (RAG) and agent have emerged to improve the factual accuracy by enhancing the LLMs with knowledge retrieved from external knowledge bases (KBs). This is mostly implemented by prompting the LLMs with instructions, examples and the retrieved knowledge. However, LLMs may have difficulty using the retrieved knowledge effectively for response generation, because they are not well trained to do such generation for specific domains. To mitigate this problem, we propose to finetune the LLMs in the RAG-based and agent-based systems with domain-specific data, together with domain-specific external knowledge, which is called knowledge augmented finetuning (KAFT). We base our study on the MobileCS2 dataset, a real-life customer service dialog dataset that features intensive knowledge interactions, to systematically compare the prompting and KAFT techniques in the RAG-based and agent-based systems. Experiment results show that KAFT substantially surpasses prompting in both RAG and agent systems, particularly in terms of factual accuracy. To the best of our knowledge, this paper represents the first solid empirical work to investigate the KAFT idea.
format Preprint
id arxiv_https___arxiv_org_abs_2506_22852
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Knowledge Augmented Finetuning Matters in both RAG and Agent Based Dialog Systems
Cai, Yucheng
Wu, Yuxuan
Huang, Yi
Feng, Junlan
Ou, Zhijian
Computation and Language
Large language models (LLMs) have recently been applied to dialog systems. Despite making progress, LLMs are prone to errors in knowledge-intensive scenarios. Recently, approaches based on retrieval augmented generation (RAG) and agent have emerged to improve the factual accuracy by enhancing the LLMs with knowledge retrieved from external knowledge bases (KBs). This is mostly implemented by prompting the LLMs with instructions, examples and the retrieved knowledge. However, LLMs may have difficulty using the retrieved knowledge effectively for response generation, because they are not well trained to do such generation for specific domains. To mitigate this problem, we propose to finetune the LLMs in the RAG-based and agent-based systems with domain-specific data, together with domain-specific external knowledge, which is called knowledge augmented finetuning (KAFT). We base our study on the MobileCS2 dataset, a real-life customer service dialog dataset that features intensive knowledge interactions, to systematically compare the prompting and KAFT techniques in the RAG-based and agent-based systems. Experiment results show that KAFT substantially surpasses prompting in both RAG and agent systems, particularly in terms of factual accuracy. To the best of our knowledge, this paper represents the first solid empirical work to investigate the KAFT idea.
title Knowledge Augmented Finetuning Matters in both RAG and Agent Based Dialog Systems
topic Computation and Language
url https://arxiv.org/abs/2506.22852