Robust Retrieval Augmented Generation for Zero-shot Slot Filling
Automatically inducing high quality knowledge graphs from a given collection of documents still remains a challenging problem in AI. One way to make headway for this problem is through advancements in a related task known as slot filling. In this task, given an entity query in form of [Entity, Slot, ?], a system is asked to fill the slot by generating or extracting the missing value exploiting evidence extracted from relevant passage(s) in the given document collection. The recent works in the field try to solve this task in an end-to-end fashion using retrieval-based language models. In this paper, we present a novel approach to zero-shot slot filling that extends dense passage retrieval with hard negatives and robust training procedures for retrieval augmented generation models. Our model reports large improvements on both T-REx and zsRE slot filling datasets, improving both passage retrieval and slot value generation, and ranking at the top-1 position in the KILT leaderboard. Moreover, we demonstrate the robustness of our system showing its domain adaptation capability on a new variant of the TACRED dataset for slot filling, through a combination of zero/few-shot learning. We release the source code and pre-trained models.
Code (2)
Tasks
Domain AdaptationFew-Shot LearningKnowledge GraphsPassage RetrievalRetrievalRetrieval-augmented Generationslot-fillingSlot FillingZero-shot Slot FillingSimilar Papers 제목 키워드 기반
Audiobox TTA-RAG: Improving Zero-Shot and Few-Shot Text-To-Audio with Retrieval-Augmented Generation
This work focuses on improving Text-To-Audio (TTA) generation on zero-shot and few-shot settings (i.e. generating unseen or uncommon audio events). Inspired by the success of Retrieval-Augmented Generation (RAG) in Large…
Audio GenerationLarge Language ModelRAGRetrieval+1Zero-shot Slot Filling with DPR and RAG
The ability to automatically extract Knowledge Graphs (KG) from a given collection of documents is a long-standing problem in Artificial Intelligence. One way to assess this capability is through the task of slot filling…
Knowledge Base PopulationKnowledge GraphsRAGRetrieval+4Slot Dependency Modeling for Zero-Shot Cross-Domain Dialogue State Tracking
Zero-shot learning for Dialogue State Tracking (DST) focuses on generalizing to an unseen domain without the expense of collecting in domain data. However, previous zero-shot DST methods ignore the slot dependencies in a…
Dialogue State TrackingZero-Shot LearningGAR-meets-RAG Paradigm for Zero-Shot Information Retrieval
Given a query and a document corpus, the information retrieval (IR) task is to output a ranked list of relevant documents. Combining large language models (LLMs) with embedding-based retrieval models, recent work shows p…
Information RetrievalPassage RetrievalRAGRe-Ranking+2Unsupervised Information Refinement Training of Large Language Models for Retrieval-Augmented Generation
Retrieval-augmented generation (RAG) enhances large language models (LLMs) by incorporating additional information from retrieval. However, studies have shown that LLMs still face challenges in effectively using the retr…
Code GenerationIn-Context LearningLanguage ModelingLanguage Modelling+6