Few-shot dialogue state tracking (DST) with Large Language Models (LLM) relies on an effective and efficient conversation retriever to find similar in-context examples for prompt learning. Previous works use raw dialogue context as search keys and queries, and a retriever is fine-tuned with annotated dialogues to achieve superior performance. However, the approach is less suited for scaling to new domains or new annotation languages, where fine-tuning data is unavailable. To address this problem, we handle the task of conversation retrieval based on text summaries of the conversations. A LLM-based conversation summarizer is adopted for query and key generation, which enables effective maximum inner product search. To avoid the extra inference cost brought by LLM-based conversation summarization, we further distill a light-weight conversation encoder which produces query embeddings without decoding summaries for test conversations. We validate our retrieval approach on MultiWOZ datasets with GPT-Neo-2.7B and LLaMA-7B/30B. The experimental results show a significant improvement over relevant baselines in real few-shot DST settings.
翻译:少样本对话状态追踪(DST)结合大语言模型(LLM)依赖于高效且有效的对话检索器,以寻找相似的上下文示例进行提示学习。以往研究采用原始对话上下文作为检索键和查询,并通过标注对话微调检索器以取得优异性能。然而,该方法难以扩展至新领域或新标注语言(在这些场景中微调数据不可用)。针对此问题,我们基于对话的文本摘要处理对话检索任务。采用基于LLM的对话摘要生成器进行查询和键的生成,从而实现高效的最大内积搜索。为避免LLM对话摘要带来的额外推理成本,我们进一步蒸馏出轻量级对话编码器,该编码器在测试对话中无需解码摘要即可生成查询嵌入。我们在MultiWOZ数据集上使用GPT-Neo-2.7B和LLaMA-7B/30B验证了检索方法的有效性。实验结果表明,在真实少样本DST场景下,该方法相较于相关基线具有显著性能提升。