Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Harnessing the Power of Semi-Structured Knowledge and LLMs with Triplet-Based Prefiltering for Question Answering

About

Large Language Models (LLMs) frequently lack domain-specific knowledge and even fine-tuned models tend to hallucinate. Hence, more reliable models that can include external knowledge are needed. We present a pipeline, 4StepFocus, and specifically a preprocessing step, that can substantially improve the answers of LLMs. This is achieved by providing guided access to external knowledge making use of the model's ability to capture relational context and conduct rudimentary reasoning by themselves. The method narrows down potentially correct answers by triplets-based searches in a semi-structured knowledge base in a direct, traceable fashion, before switching to latent representations for ranking those candidates based on unstructured data. This distinguishes it from related methods that are purely based on latent representations. 4StepFocus consists of the steps: 1) Triplet generation for extraction of relational data by an LLM, 2) substitution of variables in those triplets to narrow down answer candidates employing a knowledge graph, 3) sorting remaining candidates with a vector similarity search involving associated non-structured data, 4) reranking the best candidates by the LLM with background data provided. Experiments on a medical, a product recommendation, and an academic paper search test set demonstrate that this approach is indeed a powerful augmentation. It not only adds relevant traceable background information from information retrieval, but also improves performance considerably in comparison to state-of-the-art methods. This paper presents a novel, largely unexplored direction and therefore provides a wide range of future work opportunities. Used source code is available at https://github.com/kramerlab/4StepFocus.

Derian Boer, Fabian Koch, Stefan Kramer• 2024

Related benchmarks

TaskDatasetResultRank
Knowledge Graph RetrievalSTaRK-MAG Synthetic 1.0
Hits@153.8
20
Knowledge Graph RetrievalSTaRK-Prime Synthetic 1.0
Hits@10.393
20
Knowledge Graph RetrievalSTaRK-Amazon Synthetic 1.0
Hits@147.6
20
RetrievalSTaRK AMAZON Synthetic
Recall@2056.2
20
RetrievalSTaRK MAG Synthetic
Recall@2065.8
20
RetrievalSTaRK PRIME Synthetic
Recall@2055.9
20
Showing 6 of 6 rows

Other info

Follow for update