Microsoft researchers use AI-generated knowledge graphs to improve RAG’s ability to answer questions

Ben Wodecki, Jr. Editor

February 15, 2024

1 Min Read
Illustration of a knowledge graph

At a Glance

  • Basic RAG struggles with context. Microsoft researchers might have found a way to improve it.

Retrieval-Augmented Generation (RAG) is among the most universally used techniques for extending the capabilities of large language models. Now, researchers at Microsoft have come up with a way to improve RAG through knowledge graphs.

RAG enables a model to generate more accurate answers because the user attaches a document or other data source to the query that the model can use for reference.

Microsoft’s researchers propose GraphRAG, which uses AI-generated knowledge graphs to improve question-answering when analyzing complex information.

Basic RAG struggles with context − unable to connect the dots when it comes to answering a question that requires information from disparate sources.

GraphRAG, however, tasks the large language model to create a knowledge graph based on the private dataset. That graph is then used alongside graph machine learning to perform prompt augmentation at query time.

The process, according to Microsoft, results in substantial improvement in answering questions. By using the generated knowledge graph, GraphRAG improves the “retrieval” portion of RAG by populating context windows with higher relevance content, resulting in better answers.

Looking at the example below, both techniques answer the initial exploratory question well. However once faced with a query that requires the system to connect the dots, basic RAG fails to create a response, while GraphRAG creates a lengthy output.

A Microsoft research blog said GraphRAG has been applied to scenarios across social media, news articles and workplace productivity.

Microsoft did not say whether it is going to publish access to GraphRAG. Instead, it expressed intent to collaborate with customers on applying the new technique to “a variety of new domains.”

Read more about:

ChatGPT / Generative AI

About the Author(s)

Ben Wodecki

Jr. Editor

Ben Wodecki is the Jr. Editor of AI Business, covering a wide range of AI content. Ben joined the team in March 2021 as assistant editor and was promoted to Jr. Editor. He has written for The New Statesman, Intellectual Property Magazine, and The Telegraph India, among others. He holds an MSc in Digital Journalism from Middlesex University.

Keep up with the ever-evolving AI landscape
Unlock exclusive AI content by subscribing to our newsletter!!

You May Also Like