Australian National University

Wenyi Pi

Neo4j APOC Library Use Case

 Author

Wenyi Pi (

 ORCID:

0009–0002–2884–277)

 Introduction

In the realm of Neo4j, the APOC (Awesome Procedures on Cypher) library stands as a powerful tool. Previously, We have talked about the importance of APOC in optimising Cypher queries and improving query efficiency in our article Exploring Methods of Cypher Query Optimisations.

Introduction to APOC: Enhancing Neo4j Capabilities

Xuzeng He

Latest findings in multiple research directions for handling graph prediction and optimization Author · Xuzeng He (

 ORCID:

0009–0005–7317–7426) A graph, in short, is a description of items linked by relations, where the items of a graph are called

 nodes

(or vertices) and their relations are called

 edges

(or links). Examples of graphs can include social networks (e.g. Instagram) or knowledge

Recent Advances in using Machine Learning with Graphs

Vaibhav Khobragade

Author: Vaibhav Khobragade (

 ORCID:

0009–0009–8807–5982) Introduction Large language models (LLMs) are becoming increasingly popular in natural language processing for their superior competence in various applications.

Enhancing Language Models: The Role of Knowledge Graph Augmentation in Overcoming LLM Challenges

Enhancing Data Interactivity with LLMs and Neo4j Knowledge Graphs Author Wenyi Pi (

 ORCID

: 0009–0002–2884–2771) Introduction Since OpenAI launched ChatGPT, a

 large language model (LLM)

based chatbot, in 2023, it has set off a technological wave.

The integration of large language models (LLMs) with Neo4j-based knowledge graphs

An Introduction to RA-CM3, MuRAG and RACE Author  Xuzeng He (ORCID: 0009-0005-7317-7426)  Generative Artificial Intelligence (GAI) has demonstrated impressive performances in tasks such as text generation and text-to-image generation.

Recent Advances in using Retrieving Multimodal Information for Augmented Generation

Amanda Kau

Unlocking the Power of Questions — A deep dive into Question Answering Systems

Author:

 Amanda Kau (ORCID:


 0009–0004–4949–9284


 )

Virtual assistants have popped up on numerous websites over the years.

Harnessing The Power of Knowledge Graphs in Question Answering

Qingqin Fang

An Introduction to Retrieval Augmented Generation (RAG) and Knowledge Graph

Author Qingqin Fang (ORCID: 0009–0003–5348–4264)

 Introduction

Large Language Models (LLMs) have transformed the landscape of natural language processing, demonstrating exceptional proficiency in generating text that closely resembles human language.

Hallucination in Large Language Models and Two Effective Alleviation Pathways

Yunzhong Zhang

Efficient creation of a stoplight report with data dashboard images

Author: Yunzhong Zhang (ORCID: 0009–0002–8177–419X) Comparing data dashboards is crucial for understanding trends and performance differences. Traditionally, this task required manual effort, which was slow and sometimes inaccurate. Now, thanks to OpenAI’s GPT-4 with Vision (GPT-4V), we are able to automate and improve this process.

How to use GPT-4V For Stoplight Report

Dhruv Gupta

Improving the performance of Large Language Models Author  Dhruv Gupta (ORCID: 0009-0004-7109-5403)  ChatGPT, which first came out in late 2022, took the world by storm. Since then, various LLM models and LLM based products such as Meta’s Llama and Google’s Gemini have emerged, demonstrating the power of LLMs.

RAG: The next big thing after LLMs?

Harvesting the Power and Knowledge of Large Language Models


 Author

Amanda Kau

 (ORCID:

0009–0004–4949–9284

 )

Introduction Knowledge Graphs are networks that represent data in a graphical format. The beauty of Knowledge Graphs lies in their representation of concepts, events and entities as nodes, and the relationships between them as edges.

Automated Knowledge Graph Construction with Large Language Models

Swinburne University of Technology

Nakul Nambiar

Zhuochen Wu

Authors:  Nakul Nambiar (ORCID: 0009-0009-9720-9233) Zhuochen Wu (ORCID: 0009-0000-5642-5348)  Research Graph is a structured representation of research objects that captures information about entities and the relationships between Researcher, Organisation, Publication, Grant and Research Data.

How to use GPT API to export a research graph from PDF publications

Dhruv Gupta

Prompt Engineering — Part 2


 Using intelligence to use artificial Intelligence: A deep dive into Prompt Engineering

Author Dhruv Gupta

 (ORCID:

0009–0004–7109–5403

 )

Introduction In the previous article we discussed what prompt engineering and some of the techniques used for prompt engineering.

Prompt Engineering — Part 2

Enhancing Open-Domain Conversational Question Answering with Knowledge-Enhanced Models and Knowledge Graphs


 How knowledge-enhanced language models and knowledge graphs are advancing open-domain conversational question answering

Author:

 Wenyi Pi (ORCID:


 0009-0002-2884-2771


 )

When searching for information on the website, it is common to come across a flood of

Enhancing Open-Domain Conversational Question Answering with Knowledge-Enhanced Models and…

Recent Advances in Using Machine Learning with Graphs — Part 2 Latest findings in multiple research directions for handling graph construction and network security issues Author · Xuzeng He (

 ORCID:

0009–0005–7317–7426) Introduction A graph, in short, is a description of items linked by relations, where the items of a graph are called nodes (or vertices) and their relations are called edges (or links). Examples of

Recent Advances in Using Machine Learning with Graphs — Part 2

Exploring AI’s Ethical Terrain: Addressing Bias, Security, and Beyond

Author: Vaibhav Khobragade (

 ORCID:

0009–0009–8807–5982) Large language models (LLMs) like OpenAI’s GPT-4, Meta’s LLaMA, and Google Gemini (previously called Bard) have showcased their vast capabilities, from passing bar exams and crafting articles to generating images and website code.

Ethics and AI: Confronting the Challenges Ahead

How to efficiently retrieve information for different applications

Author Wenyi Pi (ORCID: 0009-0002-2884-2771) This article aims to explore various ways in which Retrieval-Augmented Generation (RAG) can be utilised to retrieve information and generate responses effectively within the dialogue system. The rationale behind utilising RAG as well as potential ways in which it can be employed effectively will be covered.

Efficient Information Retrieval and Response Generation with Retrieval-Augmented Generation (RAG)

Amir Aryani

Authors  Nakul Nambiar (ORCID: 0009-0009-9720-9233) Amir Aryani (ORCID: 0000-0002-4259-9774)  Knowledge graphs, which offer a structured representation of data and its relationships, are revolutionising how we organise and access information.

Mini RAG using Neo4j

Aishwarya Nambissan

A novel approach to improving the efficiency of text search in graph databases utilizing Neo4j, OpenAI, and Typesense. Authors  Nakul Nambiar (ORCID: 0009–0009–9720–9233) Aishwarya Nambissan (ORCID: 0009–0003–3823–6609)  The ability to use cutting-edge tools and frameworks is essential for staying ahead in the ever-changing field of technology.

Typesense and Neo4j in a hybrid information retrieval solution

Amir Aryani

A brief overview of different types of clustering techniques and their algorithms. Authors Aishwarya Nambissan (ORCID: 0009-0003-3823-6609) Amir Aryani (ORCID: 0000-0002-4259-9774)

 Background

Clustering is a fascinating technique used in machine learning, where patterns or data points are grouped based on their similarities. It’s like finding hidden connections among different data points without predefined labels.

19 Clustering Techniques

Understanding the Evolutionary Journey of LLMs Author Wenyi Pi (

 ORCID

: 0009–0002–2884–2771) Introduction When we talk about large language models (LLMs), we are actually referring to a type of advanced software that can communicate in a human-like manner. These models have the amazing ability to understand complex contexts and generate content that is coherent and has a human feel.

Brief Introduction to the History of Large Language Models (LLMs)

Understanding Sequential Data Modelling with Keras for Time Series Prediction

 Author

Wenyi Pi (

 ORCID

: 0009–0002–2884–2771) Introduction Recurrent Neural Networks (RNNs) are a special type of neural networks that are suitable for learning representations of sequential data like text in Natural Language Processing (NLP). We will walk through a complete example of using RNNs for time series prediction, covering

Beginner’s Guide to Recurrent Neural Networks (RNNs) with Keras

Incorporating Knowledge Graphs to explain reasoning processes


 Author

Amanda Kau (

 ORCID:

0009–0004–4949–9284) Introduction Large language models (LLMs) like GPT-4 possess remarkable language abilities, allowing them to function as chatbots, translators, and much more.

Combining Knowledge Graphs With Language Models for Interpretability

Exploring the OpenAlex Data Structure and Visualization Author: Qingqin Fang (

 ORCID:

0009–0003–5348–4264)

 Introduction to OpenAlex

In today’s world, the realm of research papers is brimming with countless hot topics, and the sheer volume of publications can be overwhelming.

Unveiling Research Trends through OpenAlex Visualization

Large Language Models for Fake News Generation and Detection Author Amanda Kau (

 ORCID

: 0009–0004–4949–9284) Introduction In recent years, fake news has become an increasing concern for many, and for good reason. Newspapers, which we once trusted to deliver credible news through accountable journalists, are vanishing en masse along with their writers.

Are Large Language Models Our Allies or Enemies in the Fight Against Fake News?

Unlocking the power of language models: A deep dive into BERT


 Author:


 Dhruv Gupta (ORCID:


 0009–0004–7109–5403


 )

Clive Humby, in 2006 rightly said, “Data is the new oil”. With data being present everywhere, it has never been more valuable.

Language Models: Deep Dive into BERT

Attention mechanism not getting enough attention Author Dhruv Gupta (

 ORCID

: 0009–0004–7109–5403) Introduction As discussed in this article, RNNs were incapable of learning long-term dependencies. To solve this issue both LSTMs and GRUs were introduced. However, even though LSTMs and GRUs did a fairly decent job for textual data they did not perform well.

Transformers Models in NLP

Exploring the potentials and limitations of Vision Language Models

Author:

 Amanda Kau (ORCID:


 0009–0004–4949–9284


 )

The human brain is more extraordinary than any machine we could build. From an early age, many of us gain the ability to comprehend what our eyes tell us and articulate it. Furthermore, we combine evidence from all our senses to reason.

How Much Can Vision Language Models Really “See”?

A novel compression technique ensuring comparable performance with 70% less parameters

Author Amanda Kau (

 ORCID

: 0009–0004–4949–9284) Introduction The sizes of large language models (LLMs) have been steadily increasing over the last few years.

CompactifAI: Large Language Models Don’t Actually Have To Be Large

From Naive to Modular: Tracing the Evolution of Retrieval-Augmented Generation

Author · Vaibhav Khobragade (

 ORCID:

0009–0009–8807–5982) Introduction Large Language Models (LLMs) have achieved remarkable success.

Three Paradigms of RAG

An introduction to tools supporting PubMed Author: Xuzeng He (

 ORCID:

0009–0005–7317–7426) PubMed is a free Web literature search service developed and maintained by the National Center for Biotechnology Information (

 NCBI

), and it is also a part of NCBI’s Entrez retrieval system that provides access to a diverse set of 38 databases.

Enhancing PubMed: from a Medical Database to beyond

Aland Astudillo

How to use GROBID to extract text from PDF  Author  Aland Astudillo (ORCID: 0009-0008-8672-3168)  GROBID is a powerful and useful tool based on machine learning that can extract text information from PDF files and other files to a structured format. One of the key challenges in knowledge mining from academic articles is reading the content of PDF files.

How to use GROBID

Using Mistral for Data tagging Author · Xuzeng He (

 ORCID:

0009–0005–7317–7426) Introduction Data tagging, in simple terms, is the process of assigning labels or tags to your data so that they are easier to retrieve or analyse.

How to use Large Language Models to tag your data: A complete tutorial

Integrating temporal data into static knowledge graphs

 Author

Amanda Kau

 (ORCID:


 0009–0004–4949–9284


 )

Introduction Knowledge graphs (KGs) have proven to be an effective method of data representation that is increasingly popular. In KGs, entities and concepts are represented as nodes, while the relationships between nodes are depicted as edges.

Dynamic Knowledge Graphs: A Next Step For Data Representation?

Transformative Advances in Language Models through External Knowledge Integration

Author: Qingqin Fang (

 ORCID:

0009–0003–5348–4264)

 Introduction

In the dynamic field of natural language processing, the integration of external knowledge has emerged as a pivotal strategy for enhancing the performance of language models.

Deeper understanding of In-Context Retrieval-Augmented Language Models

Author  Amir Aryani (ORCID: 0000-0002-4259-9774) Introduction   In this article we look at Research Graph as an information model , and an approach to connect and capture the connections between research outputs, researchers and research activities. We explore the metadata model, and we discuss how to capture this graph in a Neo4j Graph Database.

Research Graph 101

Latest findings in multiple research directions for tackling reasoning and common sense challenges Author: Xuzeng He (

 ORCID:

0009–0005–7317–7426) Knowledge Graphs, such as Wikidata, contain rich relational information between entities and have been widely used as a structured format for storing and representing relational information.

Large Language Models and Knowledge Graphs: Ways to combine them

Using intelligence to use artificial Intelligence: A deep dive into Prompt Engineering


 Author


 Dhruv Gupta (ORCID:


 0009–0004–7109–5403


 )

Introduction Large Language Models (LLMs) have become the new normal in the field of Natural Language Processing (NLP). With their improved performance and generative power, people around the world are relying on it for

Prompt Engineering

Boosting Performance for Knowledge Graphs with Neo4j APOC Library

 Author

Wenyi Pi (ORCID: 0009–0002–2884–2771) Introduction A

 knowledge graph

(graph database) captures information about main entities in a domain and the relationships between them. It was an augmented feature store for connected data which gave access to compute, access and operationalise structure features.

Exploring Methods of Cypher Query Optimisations

The AI Helper Turning Mountains of Data into Bite-Sized Instructions

Author Aland Astudillo (ORCID: 0009-0008-8672-3168) LLMs have been changing the way the entire world deals with problems and day-by-day tasks. To make them better for specific applications, they need huge amounts of data and complex and expensive approaches to training them.

What is LLMLingua?

Refining AI Vision: How Retrieval-Augmented Generation Transforms Image Captioning in Large Language Models   Leveraging External Knowledge to Enhance the Descriptive Capabilities of AI Systems Author  Vaibhav Khobragade (ORCID: 0009–0009–8807–5982) Introduction   Large Language Models (LLMs) are artificial intelligence models that are trained on massive amounts of text data in order to generate human-like language and produce coherent

Refining AI Vision: How Retrieval-Augmented Generation Transforms Image Captioning in Large…

Tools and Platform for Integration of Knowledge Graph with RAG pipelines.

Authors Aland Astudillo (ORCID: 0009-0008-8672-3168) Aishwarya Nambissan (ORCID: 0009-0003-3823-6609) Many users of chatbots such as ChatGPT, have encountered the problem of receiving inappropriate or incompatible responses. There are several reasons why this might happen.

Unveiling the Synergy: Retrieval Augmented Generation (RAG) Meets Knowledge Graphs

Author  Amir Aryani: (ORCID: 0000-0002-4259-9774) Definition   A research collaboration network is a group of researchers, and practitioners, or both, working together on joint research activities. These networks often span across disciplines, geographic boundaries, and sectors, enabling participants to share resources, expertise, and data to address common research goals more effectively than they could individually.

Research Collaboration Network

Improving the performance and application of Large Language Models

Author Amanda Kau (ORCID: 0009-0004-4949-9284) Large language models (LLMs) like GPT-4, the engine of products like ChatGPT, have taken centre stage in recent years due to their astonishing capabilities. Yet, they are far from perfect.

Understanding Retrieval Pitfalls: Challenges Faced by Retrieval Augmented Generation (RAG) models

Exploring the Potential of Temporal Feature-Logic Embedding (TFLEX) in Complex Query Resolution

Author · Vaibhav Khobragade (

 ORCID:

0009–0009–8807–5982)

 Introduction

Artificial intelligence (AI) and knowledge representation in the field of temporal knowledge graphs are rapidly gaining interest.

Harnessing Temporal Dynamics: Advanced Reasoning using Temporal Knowledge Graphs

Exploring innovative Strategies in Combating Misinformation with Enhanced Multimodal Understanding Author Wenyi Pi (

 ORCID

: 0009–0002–2884–2771) Introduction Misinformation refers to false or inaccurate information that is often given to someone in a deliberate attempt to make them believe something that is not true. This has a significantly negative impact on public health, political stability and social trust and harmony.

Multimodal Large Language Models for Misinformation Detection and Reasoning

The Three Oldest Pillars of NLP

Author Dhruv Gupta (

 ORCID

: 0009–0004–7109–5403) Introduction Natural Language Processing (NLP) has almost become synonymous with Large Language Models (LLMs), Generative AI, and fancy chatbots. With the ever-increasing amount of textual data and exponential growth in computational knowledge, these models are improving every day.

RNNs vs GRUs vs LSTMs

A Unified and Collaborative Framework for LLM Author · Qingqin Fang (

 ORCID:

0009–0003–5348–4264) Introduction In today’s rapidly evolving field of artificial intelligence, large language models (LLMs) are demonstrating unprecedented potential. Particularly, the Retrieval-Augmented Generation (RAG) architecture has become a hot topic in AI technology due to its unique technical capabilities.

RAG 2.0 is Coming?

Bridging Human Perception and AI’s Future: The Convergence of Visual Understanding and Semantic Networks Author · Vaibhav Khobragade (

 ORCID:

0009–0009–8807–5982)

 Introduction

The fusion of Vision-Language Models (

 VLMs

), Generative Models, and Knowledge Graphs (

 KGs

) is reshaping how artificial intelligence (AI) understands and interacts with the world.

Tuning Vision-Language Models and Generative Models with Knowledge Graph

Understanding the Power and Applications of Natural Language Processing

 Author

Dhruv Gupta (

 ORCID:

0009–0004–7109–5403) Introduction We are living in the era of generative AI. In an era where you can ask AI models almost anything, they will most certainly have an answer to the query. With the increased computational power and the amount of textual data, these models are bound to improve their performance.

NLU vs. NLG: Unveiling the Two Sides of Natural Language Processing

Understanding how RNNs work and its applications


 Author

Wenyi Pi (

 ORCID

: 0009–0002–2884–2771) Introduction In the ever-evolving landscape of artificial intelligence (AI), bridging the gap between humans and machines has seen remarkable progress. Researchers and enthusiasts alike have tirelessly worked across numerous aspects of this field, bringing about amazing advancements.

An Introduction to Recurrent Neural Networks (RNNs)

Exploring the Boundaries of Creativity and Responsibility in the Age of AI-Driven Media

 Author

Vaibhav Khobragade (

 ORCID:

0009–0009–8807–5982)

 Introduction

In 2024, the discipline of Generative AI takes a big step forward with the launch of revolutionary models that convert text into dynamic films, altering the landscape of digital content creation.

Generative AI’s Leap into Video in 2024 and its Ethical Horizon

Addressing Misinformation, Bias, and Privacy Concerns

Author:

 Wenyi Pi (ORCID:


 0009–0002–2884–2771


 )

Recently, the spread of false or misinformation has become an important concern for people all over the world, especially in the era of big data and multimedia, where people are faced with a flood of information that makes it difficult to determine its authenticity.

The Ethical and Social Impact of Artificial Intelligence

Latest findings for the use of Knowledge Graph in the field of QA in multiple research directions Author: Xuzeng He (

 ORCID:

0009–0005–7317–7426) Question Answering (QA), the ability to interact with data using natural language questions and obtaining accurate results, has been a long-standing challenge in computer science dating back to the 1960s.

Question Answering enhanced by Knowledge Graph

Unlocking the power of knowledge graphs in research catalogues: A deep dive into OpenAlex


 Author:


 Dhruv Gupta (ORCID:


 0009–0004–7109–5403


 )

Clive Humby, in 2006 rightly said, “Data is the new oil”. With data being present everywhere, it has never been more valuable.

Knowledge Graphs: Deep Dive into OpenAlex

Techniques to integrate Knowledge Graphs into Language Models

 Author

Amanda Kau

 (ORCID:


 0009–0004–4949–9284


 )

Introduction Both knowledge graphs (KGs) and pre-trained language models (PLMs) have gained popularity due to their ability to comprehend world knowledge and their broad applicability.

A Deep Dive Into Knowledge Graph Enhanced Pre-trained Language Models

Automated Knowledge Graph Construction with Large Language Models — Part 2

 Harvesting the Power and Knowledge of Large Language Models

Author Amanda Kau (

 ORCID

:

 0009–0004–4949–9284

) Introduction Knowledge graphs (KGs) are a structured representation of data in a graphical format, in which entities are represented by nodes and are connected by edges representing relationships

Automated Knowledge Graph Construction with Large Language Models — Part 2

An improvement architecture superior to the Transformer, proposed by Meta Author · Qingqin Fang (

 ORCID:

0009–0003–5348–4264) Introduction Recently, researchers from Meta and the University of Southern California have introduced a model called Megalodon. They claim that this model can expand the context window of language models to handle millions of tokens without overwhelming your memory.

The longer the context, the better? Unlimited Context Length in Megalodon

Latest findings in pre-training graphs and using them for link recommendation Author · Xuzeng He (

 ORCID:

0009–0005–7317–7426) Introduction A graph, in short, is a description of items linked by relations, where the items of a graph are called nodes (or vertices) and their relations are called edges (or links). Examples of graphs can include social networks (e.g. Instagram) or knowledge graphs (e.g. Wikipedia). In Instagram

Working with Graphs: Pre-training and Application

Unlocking the Future of AI: The Transformative Journey of Large Language Models Author · Vaibhav Khobragade (

 ORCID:

0009–0009–8807–5982)

 Introduction

Human language development is innate and evolves throughout life. Machines lack this ability to evolve without advanced AI algorithms.

The Journey of Large Language Models: Evolution, Application, and Limitations

Understanding Knowledge Networks From A Graph Perspective


 Author:


 Amanda Kau (ORCID:


 0009–0004–4949–9284


 )

Since 2020, over ten million scholarly articles have been published annually. To put that into perspective, say all ten million articles were released on the first day of the year.

Uncovering The Secrets Behind The Most Influential Scholarly Publications With AceMap

Understanding the Balance between Internal Knowledge and External Sources

Author Qingqin Fang (

 ORCID:

0009–0003–5348–4264) Introduction Previous research often emphasized the limitations of LLM’s information acquisition pathways, focusing on enhancing its capabilities in this regard.

When RAG and LLM Conflict: Who Will AI Listen to?

Solutions to Enhance LLM Performance in Long Contexts Author · Qingqin Fang (

 ORCID:

0009–0003–5348–4264) Introduction In the era of AI breakthroughs, large language models (LLMs) are not just advancements; they are revolutions, transforming how we interact with technology, from casual conversations with chatbots to the intricate mechanisms behind sophisticated data analysis tools.

Navigating the Long Context Conundrum: Challenges in Language Models’ Information Processing

Supervised Fine-tuning, Reinforcement Learning from Human Feedback and the latest SteerLM Author · Xuzeng He (

 ORCID:

0009–0005–7317–7426) Introduction Large Language Models (LLMs), usually trained with extensive text data, can demonstrate remarkable capabilities in handling various tasks with state-of-the-art performance. However, people nowadays typically want something more personalised instead of a general solution.

Fine-tuning Large Language Models: A Brief Introduction

An Overview of Constructing a Knowledge Graph Author · Qingqin Fang (

 ORCID:

0009–0003–5348–4264) 1. Introduction 1.1 What is a Knowledge Graph Knowledge Graphs are structured semantic knowledge bases used to rapidly describe concepts and their relationships in the physical world.

Unlocking Intelligence: The Journey from Data to Knowledge Graph

Author Dhruv Gupta (

 ORCID

: 0009–0004–7109–5403) Introduction Large Language Models (LLMs) have become the new face of Natural language processing (NLP). With their generative power and ability to comprehend human language, the human reliance on these models is increasing every day. However, the LLMs have been known to hallucinate and thus produce wrong outputs.

Prompt engineering: A Way to Smartly Use AI

Latest effort in assessing the security of the code generated by large language models Author · Xuzeng He (

 ORCID:

0009–0005–7317–7426) Introduction With the surge of Large Language Models (LLMs) nowadays, there is a rising trend among developers to use Large Language Models to assist their daily code writing. Famous products include GitHub Copilot or simply ChatGPT.

Large Language Models for Code Writing: Security Assessment

Research Graph

Google’s Groundbreaking AI Model Explained
 


 Introduction

In the rapidly evolving landscape of artificial intelligence, Google continues to stand at the forefront, consistently pushing the boundaries of what these technologies can achieve. One of their latest advancements, GEMMA2, represents a significant leap in AI capabilities.

What is Gemma2?

Stories by Research Graph on Medium

**The AI Helper Turning Mountains of Data into Bite-Sized Instructions**

<figure>
<img
src="https://cdn-images-1.medium.com/max/1000/0*XwvfCyjF7dUWQ7Wj" />
<figcaption>Image generated with Google’s Gemini
12–02–2024.</figcaption>
</figure>

### Author

* Aland Astudillo (ORCID: [0009-0008-8672-3168](https://orcid.org/0009-0008-8672-3168))

LLMs have been changing the way the entire world deals with problems and
day-by-day tasks. To make them better for specific applications, they
need huge amounts of data and complex and expensive approaches to
training them. However, there are some challenges such as limited prompt
size, and limited context windows that make LLM not suitable for some.
This is a major issue for tasks that require huge amounts of
information. LLMLingua has been developed as a framework that helps LLMs
in addressing these limitations. In this article, we will review what
LLMLingua is, what it does, how it does it and what we can expect in the
near future.

Picture this scenario, you're tasked with teaching a group of aspiring
chefs how to prepare a complex gourmet meal. You could throw every
recipe detail and culinary term at them, hoping they absorb it all. But
wouldn't it be more effective to break down the instructions into clear,
concise steps, focusing only on the crucial techniques and ingredients?
This, in essence, is the magic of LLMLingua.

Instead of aspiring chefs, imagine training a team of AI apprentices to
tackle complex tasks. Like those aspiring chefs who thrive in learning
to prepare new dishes, LLMs thrive on information. But feeding LLMs with
mountains of raw data can be overwhelming, leading to slow processing
and limited performance. Enter LLMLingua, the innovative tool acting as
the AI chef, meticulously preparing information into bite-sized
instructions that these AI apprentices can easily digest and master.

<figure>
<img
src="https://cdn-images-1.medium.com/max/1000/0*WauCih49veOTPy9V" />
<figcaption>Image generated with Google’s Gemini
12–02–2024.</figcaption>
</figure>

So, how does LLMLingua craft this culinary magic?

It employs two key techniques, similar to how a skilled chef optimises
a recipe:

1. **Ingredient Trimming:** Imagine each piece of information as an
 ingredient in the recipe. LLMLingua, akin to a seasoned chef,
 identifies and discards unnecessary elements like filler words or
 irrelevant details. This "trimming" streamlines instructions,
 keeping only the essential components, just like using only the
 right spices to enhance a dish's flavour.
2. **Recipe Refining**: Even carefully chosen ingredients need precise
 instructions. LLMLingua refines the wording and phrasing of
 instructions, ensuring the AI apprentices grasp the core meaning
 with perfect clarity. It's like the chef rewriting the recipe in
 clear, concise steps, leaving no room for confusion or
 misinterpretation.

Currently, several variations of LLMLingua exist, each tailored to
specific tasks and LLM architectures. One version optimises prompts for
question-answering models, while another focuses on improving
summarization tasks. These variations highlight the adaptability and
evolving nature of this technology.

But LLMLingua isn't alone in the quest for efficient AI. Techniques like
knowledge distillation and parameter reduction also aim to streamline
models. What sets LLMLingua apart is its focus on prompt manipulation,
offering a unique and flexible approach.

<figure>
<img
src="https://cdn-images-1.medium.com/max/1000/0*hAjLbEGPdLOJyziQ" />
<figcaption>Scheme of the LLMLingua framework. Image created by Aland
Astudillo.</figcaption>
</figure>

Imagine teaching a friend a complex recipe. You wouldn't overwhelm them
with every detail; instead, you'd break it down, focusing on key steps
and ingredients. LLMLingua operates similarly, streamlining information
for large language models (LLMs) to achieve peak performance. Here's a
deeper dive into its technical pipeline:

1. **Input Preprocessing**: The journey begins with the "raw"
 information intended for the LLM. LLMLingua analyses it, identifying
 irrelevant elements like filler words or redundant instructions.
 Think of skimming unnecessary details from a recipe book, keeping
 only the crucial steps. This initial "trimming" helps reduce the
 information load on the LLM.
2. **Tokenization**: Next, LLMLingua breaks down the preprocessed text
 into individual units called "tokens," similar to words in a
 sentence. This facilitates further analysis and manipulation.
 Imagine separating the ingredients listed in your recipe into
 individual units --- flour, eggs, milk, etc.
3. **Budget Control**: LLMLingua employs a "budget controller" to
 ensure compression doesn't compromise information integrity. By
 setting a desired compression ratio, it controls how much
 information can be discarded while maintaining optimal performance
 for the LLM. This is like deciding how much of each ingredient to
 use without altering the final dish drastically.
4. **Iterative Compression**: Now comes the magic. LLMLingua utilises
 an iterative algorithm to refine the tokenized information. Each
 iteration analyses the remaining tokens, identifying opportunities
 for further compression while considering their importance for the
 LLM's task. Think of repeatedly reviewing your recipe steps,
 replacing complex techniques with simpler alternatives whenever
 possible.
5. **Instruction Tuning**: Beyond simply removing elements, LLMLingua
 also refines the wording and phrasing of the remaining instructions.
 Imagine rewriting your recipe steps for clarity and accuracy,
 ensuring your friend understands them perfectly. This ensures the
 LLM receives clear and concise instructions, minimising room for
 misinterpretation.
6. **Output Generation**: Finally, the compressed and refined
 information is presented to the LLM as a "streamlined recipe." The
 LLM processes it efficiently, achieving higher performance with
 faster response times and reduced resource consumption. It's like
 your friend effortlessly executing the optimised recipe, producing a
 delicious dish efficiently.

This is just a simplified overview and the actual LLMLingua pipeline
involves complex algorithms and deep learning techniques that can be
reviewed in the research article (See the references).

**Key concepts: Entropy and perplexity**

Additionally, there are two key concepts that allow the control engine
in the LLMLingua framework: entropy and perplexity. These metrics act as
indicators of information content and difficulty, allowing researchers
to gauge the effectiveness of LLMLingua's compression techniques.

**1. Entropy**

Imagine tossing a coin. With one possible outcome (heads or tails), the
entropy is low. Now, consider a hundred-sided die --- with many more
possibilities --- the entropy is higher. Similarly, entropy in language
measures the unpredictability of the next word given the previous ones.
Longer, more complex sentences typically have higher entropy than
shorter, simpler ones.

LLMLingua aims to reduce the entropy of instructions fed to LLMs. By
removing redundant information and focusing on key elements, it
essentially makes the next word more predictable, like simplifying the
die to fewer sides. This lowers the information load on the LLM, leading
to more efficient processing.

**2. Perplexity**

Perplexity builds upon entropy but is presented as an inverse
probability. A lower perplexity value signifies higher predictability,
indicating that the model can more easily anticipate the next word.
Conversely, a high perplexity suggests complex, unpredictable language,
making it harder for the model to process.

In the context of LLMLingua, a decrease in perplexity after compression
reflects improved efficiency. It means the LLM can understand the
compressed instructions with the same accuracy as the original but with
less effort. This translates to faster response times and lower
computational costs.

It's important to note that:

- Both entropy and perplexity are complex measures with nuances beyond
 this simplified explanation.
- LLMLingua's goal isn't to achieve the absolute lowest entropy or
 perplexity, but to find the optimal balance between compression and
 information fidelity.

So, while entropy and perplexity might seem like abstract concepts, they
play a crucial role in understanding how LLMLingua achieves its
efficiency gains.

**Why is LLMLingua the secret ingredient for AI success?**

LLMs are changing the world, translating languages on the fly, composing
personalised poems, and even answering your questions with remarkable
accuracy. But just like our aspiring chefs, their hunger for information
can create limitations. Imagine translating entire novels
word-for-word --- it's slow, resource-intensive, and ultimately
unsatisfying. LLMLingua tackles this head-on, paving the way for faster,
more efficient, and impactful AI applications:

- **Speedy Chatbots**: Ever feel like waiting forever for a chatbot
 response? LLMLingua compresses your questions, enabling chatbots to
 understand you instantly and respond at lightning speed. It's like
 having a personal AI assistant who's always on top of their game,
 eliminating frustrating wait times.
- **Translation on the Go**: Imagine translating entire documents in
 seconds! LLMLingua empowers translation tools to process information
 with laser precision, breaking down language barriers faster than
 ever. Think of it as a universal translator, seamlessly bridging
 communication gaps across cultures and languages like a skilled
 multilingual chef seamlessly navigating different cuisines.
- **Research Revolution**: Picture sifting through mountains of
 scientific data in minutes. LLMLingua empowers AI assistants to
 analyse complex research papers with unmatched efficiency,
 accelerating scientific breakthroughs and discoveries. It's like
 having a tireless research partner who can summarise vast amounts of
 information, highlighting key findings and saving researchers
 countless hours.
- **AI for Everyone**: From personalised learning experiences to
 efficient task management, LLMLingua can revolutionise how AI
 assistants interact with us. Imagine an AI tutor who personalises
 learning modules based on your individual needs or an assistant who
 manages your schedule with laser-focus, all thanks to the power of
 concise and optimised instructions.

But, wait...how do we solve the problem of the limited context size? Let
me introduce you to **LongLLMLingua**!

### What is LongLLMLingua?

Think of LongLLMLingua as the master chef, overseeing the entire
culinary experience. It builds upon LLMLingua's foundation, applying its
compression techniques not just to single instructions, but to entire
sequences of information. This empowers LLMs to process long
contexts --- think research papers, dialogue history, or complex
narratives --- with impressive efficiency and improved performance.

**How does it work?**

LongLLMLingua operates in three key phases:

- **Coarse Compression**: First, it performs a "rough cut," analysing
 the entire context and identifying large sections it can safely
 discard. Imagine skimming extraneous details from a recipe book,
 focusing only on the core steps for each dish.
- **Fine Compression**: Next, it dives deeper, applying LLMLingua's
 techniques to refine remaining information. Think of meticulously
 preparing each dish, ensuring optimal ingredient selection and precise
 instructions.
- **Reordering for Coherence**: Finally, LongLLMLingua ensures the
 compressed information maintains its original meaning and flow.
 Imagine plating each dish in a visually appealing and sequential
 manner, ensuring a cohesive dining experience.

**Why is LongLLMLingua important?**

Long contexts are crucial in various AI applications:

- **Question Answering**: Imagine needing context from several articles
 to answer a complex question. LongLLMLingua helps LLMs retain key
 information across documents, leading to more accurate answers.
- **Machine Translation**: Imagine translating entire books instead of
 single sentences. LongLLMLingua ensures coherence and preserves
 meaning despite lengthy input.
- **Dialogue Systems**: Imagine chatbots understanding intricate
 conversation history. LongLLMLingua enables them to maintain context,
 leading to more natural and engaging interactions.

Real-world examples of LongLLMLingua in action:

- **LongBench Benchmark**: This benchmark measures LLM performance in
 long context scenarios. When applied to GPT-3.5-Turbo, LongLLMLingua
 achieved a 17.1% performance boost with 4x fewer tokens.
- **ZeroScrolls Benchmark**: This benchmark focuses on reading
 comprehension in long contexts. LongLLingua reduced costs by \$27.4
 per 1,000 samples while maintaining performance.

Remember, LongLLMLingua isn't limited to cooking analogies. It's a
powerful tool revolutionising how LLMs handle long contexts,
accelerating progress in various AI domains. By improving efficiency and
maintaining information fidelity, LongLLMLingua opens doors to a future
where AI interactions are more contextual, seamless, and impactful.

**Conclusion**

In conclusion, both LLMLingua and LongLLMLingua are not just
technological advancements; they're culinary metaphors come to life.
LLMLingua acts as the skilled chef, meticulously preparing information
into bite-sized instructions for AI apprentices. LongLLMLingua takes the
baton, transforming into the master chef, seamlessly handling entire
sequences of information like preparing a multi-course feast. Together,
they're revolutionising how LLMs interact with information, paving the
way for a future where:

- AI responses are faster, more accurate, and contextually relevant.
- Machine translation transcends sentence-by-sentence limitations,
 unlocking global communication on a grand scale.
- AI assistants understand our interactions in their entirety, leading
 to more natural and engaging dialogues.

As these technologies continue to evolve, the possibilities are endless.
Imagine AI tutors adapting to your learning style based on years of
educational data, or chatbots understanding months of conversation
history to anticipate your needs with uncanny accuracy.

The future of AI is hungry for efficiency and understanding, and
LLMLingua and LongLLMLingua are serving up the perfect recipe for
success. By empowering LLMs to process information like culinary
masters, they're opening doors to a new era of seamless interaction,
accelerated discovery, and boundless communication.

**Sources**

- LLMLingua research paper: Jiang et al., 2023. LLMLingua: Compressing
 prompts for accelerated inference of Large Language Models. arXiv
 2310.05736v2 <https://arxiv.org/abs/2310.05736>
- LongLLMLingua research paper: Jiang et al., 2023. LongLLMLingua:
 accelerating and enhancing LLMs in long context scenarios via prompt
 compression. arXiv 2310.06839v1 <https://arxiv.org/abs/2310.06839>
- Microsoft LLMLingua page: LLMLingua: Compressing Prompts for
 Accelerated Inference of Large Language Models --- Microsoft Research
 <https://www.microsoft.com/en-us/research/project/llmlingua/>
- Hugging Face LLMLingua Space: LLMLingua --- a Hugging Face
 <https://huggingface.co/spaces/microsoft/LLMLingua>

![](https://medium.com/_/stat?event=post.clientViewed&referrerSource=full_rss&postId=3a632385b206){width="1"
height="1"}

<figure>
<img
src="https://cdn-images-1.medium.com/max/1024/1*7Fy1vUav7gAiy9CAWanAvQ.png" />
<figcaption>Created using DALLE on 6 Feb 2024</figcaption>
</figure>

#### Authors:

* Nakul Nambiar (ORCID: [0009-0009-9720-9233](https://orcid.org/0009-0009-9720-9233))
* Zhuochen Wu (ORCID: [0009-0000-5642-5348](https://orcid.org/0009-0000-5642-5348))

Research Graph is a structured representation of research objects that
captures information about entities and the relationships between
Researcher, Organisation, Publication, Grant and Research Data.
Currently, the publications are available as PDF files, and due to the
free-form text, it is difficult to parse a PDF file to extract
structured information. In this article, we will try to create a
research graph from a PDF of a publication by extracting relevant
information from the text and organising it into a graph structure
using OpenAI.

<figure>
<img
src="https://cdn-images-1.medium.com/max/1024/1*40j-_agywNgUsJ_h_kXslA.jpeg" />
<figcaption>Pipeline to create the graph from PDF</figcaption>
</figure>

### OpenAI

In this work, we use OpenAI API and the new Assistant functionality of
GPT (Currently in Beta) to transform PDF documents to a set of
structured JSON files based on Research Graph Schema.

#### Assistants API

The Assistants API allows you to build AI assistants within your
applications. An assistant can answer user questions by using models,
tools, and information per predetermined guidelines. It is a beta API
that is actively under development. Using the assistants API, we can use
the OpenAI-hosted tools like code interpreter and knowledge retrieval.
For this blog post, we will focus on knowledge retrieval.

#### Knowledge Retrieval

Sometimes, we need the AI model to answer queries based on unknown
knowledge, like user-provided documents or sensitive information. We can
augment the model with this information using the assistant APIs
knowledge retrieval tool. We can upload files to the assistant, and it
automatically chunks the documents and creates and stores embeddings to
implement vector search on the data.

#### Example

For our example, we will upload PDF files of publications to the OpenAI
assistant and the knowledge retrieval tool to get a JSON output of the
graph schema for the given publications. The publication we used for
this example can be accessed from the following
[link](https://doi.org/10.5166/jroi-2-1-6).

<figure>
<img
src="https://cdn-images-1.medium.com/max/1024/1*CrAVHSNrRmQiBay81_Bi0Q.png" />
<figcaption>Screenshot from the publication (<a
href="https://doi.org/10.5166/jroi-2-1-6">https://doi.org/10.5166/jroi-2-1-6</a>)</figcaption>
</figure>

#### Step 1

Read the input path where the Publications PDFs are stored and the
output path where the JSON output will be stored.

 import configparser
 config = configparser.ConfigParser()
 config.read('{}/config.ini'.format(current_path))
 input_path = config['DEFAULT']['Input-Path']
 output_path = config['DEFAULT']['Output-Path']
 debug = config['DEFAULT']['Debug']

#### Step 2

Get all the PDF files from the Input path.

 onlyfiles = [f for f in os.listdir(input_path) if os.path.isfile(os.path.join(input_path, f))]

#### Step 3

Then, we need to initialise the assistant to use the Knowledge retrieval
tool. To do this, we need to specify the type of tools in the API for
"retrieval". We also specify the instructions for the assistant and the
OpenAI model to be used.

 my_file_ids = []
 if client.files.list().data==[]:
 for f in onlyfiles:
 file = client.files.create(
 file=open(input_path + f, "rb"),
 purpose='assistants'
 )
 my_file_ids.append(file.id)

 # Add the file to the assistant
 assistant = client.beta.assistants.create(
 instructions = "You are a publication database support chatbot. Use pdf files uploaded to best respond to user queries in JSON.",
 model = "gpt-4-1106-preview",
 tools = [{"type": "retrieval"}],
 # Do not attach all files to the assistant, otherwise, it will mismatch the answers even though specify file ID in query messages.
 # We will attach to each message instead
 )

#### Step 4

Then, we specify the information we need to extract from the
publications files and pass them as user queries to the assistant. After
experimenting with the assistant instructions, we found that requesting
JSON format in each user message generates the most consistent output.

 user_msgs = ["Print the title of this paper in JSON",
 "Print the authors of this paper in JSON",
 "Print the abstract section of the paper in JSON",
 "Print the keywords of the paper in JSON",
 "Print the DOI number of the paper in JSON",
 "Print the author affiliations of the paper in JSON",
 "Print the reference section of the paper in JSON"]

#### Step 5

The next is to pass the queries to the assistant to generate the output.
We need to create an individual thread object for each user query, which
contains the query as a user message. Then, we run the thread and
retrieve the assistant's answer.

 all_results = []

 for i in my_file_ids:
 print('\n#####')
 # the JSON result it can extract and parse, hopefully
 file_result = {}
 
 for q in user_msgs:
 # create thread, user message and run objects for each query
 thread = client.beta.threads.create()
 msg = client.beta.threads.messages.create(
 thread_id=thread.id,
 role="user",
 content=q,
 file_ids=[i] # specify the file/publication we want to extract from
 )
 print('\n',q)
 run = client.beta.threads.runs.create(
 thread_id=thread.id,
 assistant_id=assistant.id,
 additional_instructions="If answer cannot be found, print 'False'" # not very useful at the time of this presentation
 )
 # checking run status by retrieving updated object each time
 while run.status in ["queued",'in_progress']:
 print(run.status) 
 time.sleep(5)
 run = client.beta.threads.runs.retrieve(
 thread_id=thread.id,
 run_id=run.id
 )
 # usually a rate limit error
 if run.status=='failed': logging.info("Run failed: ", run)
 if run.status=='completed':
 print("<Complete>")
 # extract updated message object, this includes user messages
 messages = client.beta.threads.messages.list(
 thread_id=thread.id
 )
 for m in messages:
 if m.role=='assistant':
 value = m.content[0].text.value # get the text response
 if "json" not in value:
 if value=='False':logging.info("No answer found for ", str(q))
 else:
 logging.info("Not JSON output, maybe no answer found in the file or model is outdated: ", str(value))
 else:
 # clean the response and try to parse as json
 value = value.split("```")[1].split('json')[-1].strip()
 try: 
 d = json.loads(value)
 file_result.update(d)
 print(d)
 except Exception as e:
 logging.info(f"Query {q} \nFailed to parse string to JSON: ", str(e))
 print(f"Query {q} \nFailed to parse string to JSON: ", str(e))
 
 all_results.append(file_result)

The JSON output generated for the above publication file is:

 [{'title': 'Dodes (diagnostic nodes) for Guideline Manipulation',
 'authors': [{'name': 'PM Putora',
 'affiliation': 'Department of Radiation-Oncology, Kantonsspital St. Gallen, St. Gallen, Switzerland'},
 {'name': 'M Blattner',
 'affiliation': 'Laboratory for Web Science, Zürich, Switzerland'},
 {'name': 'A Papachristofilou',
 'affiliation': 'Department of Radiation Oncology, University Hospital Basel, Basel, Switzerland'},
 {'name': 'F Mariotti',
 'affiliation': 'Laboratory for Web Science, Zürich, Switzerland'},
 {'name': 'B Paoli',
 'affiliation': 'Laboratory for Web Science, Zürich, Switzerland'},
 {'name': 'L Plasswilma',
 'affiliation': 'Department of Radiation-Oncology, Kantonsspital St. Gallen, St. Gallen, Switzerland'}],
 'Abstract': {'Background': 'Treatment recommendations (guidelines) are commonly represented in text form. Based on parameters (questions) recommendations are defined (answers).',
 'Objectives': 'To improve handling, alternative forms of representation are required.',
 'Methods': 'The concept of Dodes (diagnostic nodes) has been developed. Dodes contain answers and questions. Dodes are based on linked nodes and additionally contain descriptive information and recommendations. Dodes are organized hierarchically into Dode trees. Dode categories must be defined to prevent redundancy.',
 'Results': 'A centralized and neutral Dode database can provide standardization which is a requirement for the comparison of recommendations. Centralized administration of Dode categories can provide information about diagnostic criteria (Dode categories) underutilized in existing recommendations (Dode trees).',
 'Conclusions': 'Representing clinical recommendations in Dode trees improves their manageability handling and updateability.'},
 'Keywords': ['dodes',
 'ontology',
 'semantic web',
 'guidelines',
 'recommendations',
 'linked nodes'],
 'DOI': '10.5166/jroi-2-1-6',
 'references': [{'ref_number': '[1]',
 'authors': 'Mohler J Bahnson RR Boston B et al.',
 'title': 'NCCN clinical practice guidelines in oncology: prostate cancer.',
 'source': 'J Natl Compr Canc Netw.',
 'year': '2010 Feb',
 'volume_issue_pages': '8(2):162-200'},
 {'ref_number': '[2]',
 'authors': 'Heidenreich A Aus G Bolla M et al.',
 'title': 'EAU guidelines on prostate cancer.',
 'source': 'Eur Urol.',
 'year': '2008 Jan',
 'volume_issue_pages': '53(1):68-80',
 'notes': 'Epub 2007 Sep 19. Review.'},
 {'ref_number': '[3]',
 'authors': 'Fairchild A Barnes E Ghosh S et al.',
 'title': 'International patterns of practice in palliative radiotherapy for painful bone metastases: evidence-based practice?',
 'source': 'Int J Radiat Oncol Biol Phys.',
 'year': '2009 Dec 1',
 'volume_issue_pages': '75(5):1501-10',
 'notes': 'Epub 2009 May 21.'},
 {'ref_number': '[4]',
 'authors': 'Lawrentschuk N Daljeet N Ma C et al.',
 'title': "Prostate-specific antigen test result interpretation when combined with risk factors for recommendation of biopsy: a survey of urologist's practice patterns.",
 'source': 'Int Urol Nephrol.',
 'year': '2010 Jun 12',
 'notes': 'Epub ahead of print'},
 {'ref_number': '[5]',
 'authors': 'Parmelli E Papini D Moja L et al.',
 'title': 'Updating clinical recommendations for breast colorectal and lung cancer treatments: an opportunity to improve methodology and clinical relevance.',
 'source': 'Ann Oncol.',
 'year': '2010 Jul 19',
 'notes': 'Epub ahead of print'},
 {'ref_number': '[6]',
 'authors': 'Ahn HS Lee HJ Hahn S et al.',
 'title': 'Evaluation of the Seventh American Joint Committee on Cancer/International Union Against Cancer Classification of gastric adenocarcinoma in comparison with the sixth classification.',
 'source': 'Cancer.',
 'year': '2010 Aug 24',
 'notes': 'Epub ahead of print'},
 {'ref_number': '[7]',
 'authors': 'Rami-Porta R Goldstraw P.',
 'title': 'Strength and weakness of the new TNM classification for lung cancer.',
 'source': 'Eur Respir J.',
 'year': '2010 Aug',
 'volume_issue_pages': '36(2):237-9'},
 {'ref_number': '[8]',
 'authors': 'Sinn HP Helmchen B Wittekind CH.',
 'title': 'TNM classification of breast cancer: Changes and comments on the 7th edition.',
 'source': 'Pathologe.',
 'year': '2010 Aug 15',
 'notes': 'Epub ahead of print'},
 {'ref_number': '[9]',
 'authors': 'Paleri V Mehanna H Wight RG.',
 'title': "TNM classification of malignant tumours 7th edition: what's new for head and neck?",
 'source': 'Clin Otolaryngol.',
 'year': '2010 Aug',
 'volume_issue_pages': '35(4):270-2'},
 {'ref_number': '[10]',
 'authors': 'Guarino N.',
 'title': 'Formal Ontology and Information Systems',
 'source': '1998 IOS Press'},
 {'ref_number': '[11]',
 'authors': 'Uschold M Gruniger M.',
 'title': 'Ontologies: Principles Methods and Applications.',
 'source': 'Knowledge Engineering Review',
 'year': '1996',
 'volume_issue_pages': '11(2)'},
 {'ref_number': '[12]',
 'authors': 'Aho A Garey M Ullman J.',
 'title': 'The Transitive Reduction of a Directed Graph.',
 'source': 'SIAM Journal on Computing',
 'year': '1972',
 'volume_issue_pages': '1(2): 131–137'},
 {'ref_number': '[13]',
 'authors': 'Tai K',
 'title': 'The tree-to-tree correction problem.',
 'source': 'Journal of the Association for Computing Machinery (JACM)',
 'year': '1979',
 'volume_issue_pages': '26(3):422-433'}]}]

#### Step 6

File objects and assistant object needs to be cleaned up because they
cost money in 'retrieval' mode. Also, it is a good coding practice.

 for f in client.files.list().data:
 client.files.delete(f.id)

 # Retrieve and delete running assistants
 my_assistants = client.beta.assistants.list(
 order="desc",
 )
 for a in my_assistants.data: 
 response = client.beta.assistants.delete(a.id)
 print(response)

#### Step 7

The next step is to generate a graph visualisation using the Python
[Networkx](https://networkx.org/documentation/stable/tutorial.html) package.

 import networkx as nx
 import matplotlib.pyplot as plt

 G = nx.DiGraph()
 node_colors = []

 key = "jroi/" + all_results[0]['title']
 G.add_nodes_from([(all_results[0]['title'], {'doi': all_results[0]['DOI'], 'title': all_results[0]['title'], 'source': 'jroi', 'key': key})])
 node_colors.append('#4ba9dc')

 for author in all_results[0]['authors']:
 key = "jroi/" + author['name']
 G.add_nodes_from([(author['name'], {'key': key, 'local_id': author['name'], 'full_name': author['name'], 'source': 'jroi'})])
 G.add_edge(all_results[0]['title'], author['name'])
 node_colors.append('#63cc9e')


 for reference in all_results[0]['references']:
 key = "jroi/" + reference['title']
 G.add_nodes_from([(reference['title'].split('.')[0][:25] + '...', {'title': reference['title'], 'source': 'jroi', 'key': key})])
 G.add_edge(all_results[0]['title'], reference['title'].split('.')[0][:25] + '...')
 node_colors.append('#4ba9dc')

 pos = nx.spring_layout(G)
 labels = nx.get_edge_attributes(G, 'label')
 nx.draw(G, pos, with_labels=True, node_size=1000, node_color=node_colors, font_size=7, font_color='black')
 nx.draw_networkx_edge_labels(G, pos, edge_labels=labels)
 plt.savefig("graph_image.png")
 plt.show()

The graph visualisation is as below:

<figure>
<img
src="https://cdn-images-1.medium.com/max/640/1*-a5y19GUSyQZv63KMXKnZA.png" />
<figcaption>Graph Visualisation generated using Networkx</figcaption>
</figure>

**Note:** Please note that the output structure generated by OpenAI may
be different for different executions. So you may need to update the
above code according to that structure.

### Conclusion

In conclusion, leveraging the GPT API to extract a Research Graph from
PDF publications offers a powerful and efficient solution for
researchers and data analysts. This workflow streamlines the process of
converting PDF publications into structured and accessible Research
Graphs. But we must also pay careful attention to the inconsistencies of
the responses generated by LLM. Over time, accuracy and relevance may be
further enhanced by routinely updating and improving the extraction
model.

![](https://medium.com/_/stat?event=post.clientViewed&referrerSource=full_rss&postId=4655f64337db){width="1"
height="1"}

<figure>
<img
src="https://cdn-images-1.medium.com/max/1024/1*Gzmbpq5qJUz2W1zpHd1XGA.png" />
<figcaption>Created using DALL-E on 28 Jan 2024</figcaption>
</figure>

### Authors

* Nakul Nambiar (ORCID: [0009-0009-9720-9233](https://orcid.org/0009-0009-9720-9233))
* Amir Aryani (ORCID: [0000-0002-4259-9774](https://orcid.org/0000-0002-4259-9774))

Knowledge graphs, which offer a structured representation of data and
its relationships, are revolutionising how we organise and access
information. With large amounts of data, it sometimes becomes difficult
to draw insights from it. This blog article examines how to combine
Neo4j, a graph database, with OpenAI's Retrieval-Augmented Generation
(RAG) model to build a robust knowledge management system.

LLMs are pre-trained models and generate outputs based on the prompts we
pass them. Sometimes, the LLM results are precisely what you need, but
other times, they just return random facts from their training data.
Retrieval-augmented generation (RAG) is an AI framework that tries to
improve the response of the LLM models by providing context knowledge
from external sources. Additional benefits of RAG include:

- Enables getting insights into private data without sharing or using it
 for training.
- Reduces the chances of 'Hallucination' and incorrect responses.
- Reduces the need to train the model on new data continuously.

### Sample Data

For our exercise, we will use OpenAI, combined with the Australian
National Graph Database, to create an RAG pipeline.

The [Australian National Graph
Database](https://researchgraph.org/national-graph/) is a vast knowledge
graph with around 11 million nodes. We can't possibly fit all the data
into the OpenAI context. So, in the first phase, we will retrieve a
subset of the graph and then use this graph as a context for the OpenAI
and to answer prompts.

We use a Virtual Machine (VM) with a Neo4j server running the graph
database as a Neo4j server. The [graph
schema](https://researchgraph.org/schema/) includes five primary node
types that describe the research ecosystem: Researchers, Publications,
Datasets, Grants and Organisations.

<figure>
<img
src="https://cdn-images-1.medium.com/max/1024/1*UocexTQHeuX3_VAwSJ03CQ.png"
alt="Research Graph Schema: https://researchgraph.org/schema/" />
<figcaption>Research Graph Schema. <a
href="https://researchgraph.org/schema/">https://researchgraph.org/schema/</a></figcaption>
</figure>

### Step 1

Firstly, we must establish the connection to the Neo4j database. Also,
you might notice that we use an OpenAI API key. This key can be
generated at the OpenAI portal. A tutorial is available
[here](https://elephas.app/blog/how-to-create-openai-api-keys-cl5c4f21d281431po7k8fgyol0).

 import os
 import json
 import pandas as pd
 import configparser
 from neo4j import GraphDatabase

 config = configparser.ConfigParser()
 config.read('config.ini')

 # Set the OpenAI API key env variable manually
 os.environ["OPENAI_API_KEY"] = config['DEFAULT']['openai-key']

 # DB credentials
 url = config['DEFAULT']['url']
 username = config['DEFAULT']['username']
 password = config['DEFAULT']['password']

### Step 2

We get a subset of the graph from the larger dataset by querying Neo4j.
The output of the graph is retrieved in JSON format.

 def run_query(cypher_query):
 with GraphDatabase.driver(url, auth=(username, password)) as driver:
 with driver.session() as session:
 result = session.run(cypher_query)
 return result.data()

 cypher_query = """
 MATCH (n:organisation)-[r]-(g:grant)
 where g.source='arc.gov.au' RETURN * LIMIT 100
 """

 result = run_query(cypher_query)

Note: We have used a limit of 100 nodes in the response to limit the
size of the returning JSON file. This would impact the LLM functionality
if it exceeds the context window size.

We create a system prompt by providing background knowledge about the
graph database and the task. Along with it, we pass the JSON subgraph
as context.

 system_prompt = f'''
 You are a helpful agent designed to fetch information from a graph database.
 Data
 {json.dumps(result)}

 This is the output of the Neo4j cypher which returns nodes and relationships. n is the organisations, g is the grants
 and r is the relationships between the grants and organisations.

 Based on this data answer the user prompts and return the output as json.

 '''

Users can then enter the prompt to get insights from the subgraph. For
example, in the prompt below, we are trying to find the grant with the
most relationship to organisations.

 prompt = "Help me find the grants which is related to the most organisations"

### Step 3

We pass the system and user prompts to the OpenAI model. In this case,
we use "gpt-4--1106-preview". More information is available on the
[OpenAI page on Models](https://platform.openai.com/docs/models).

 from openai import OpenAI
 client = OpenAI()

 def define_query(prompt, model="gpt-4-1106-preview"):
 completion = client.chat.completions.create(
 model=model,
 temperature=0,
 response_format= {
 "type": "json_object"
 },
 messages=[
 {
 "role": "system",
 "content": system_prompt
 },
 {
 "role": "user",
 "content": prompt
 }
 ]
 )
 return completion.choices[0].message.content

OpenAI provides the grant details with the most connections to
organisations as a JSON output.

 print(define_query(prompt))

### Output

This is the output provided by OpenAI RAG.

 {
 "grants": [
 {
 "arc_for_primary": "1117",
 "local_id": "LP200301123",
 "start_year": "2023",
 "purl": "http://purl.org/au-research/grants/arc/LP200301123",
 "source": "arc.gov.au",
 "type": "grant",
 "arc_scheme_name": "Linkage Projects",
 "title": "Making Australia resilient to airborne infection transmission.",
 "arc_announcement_administering_organisation": "Queensland University of Technology",
 "url": "http://dataportal.arc.gov.au/NCGP/API/grants/LP200301123",
 "arc_funding_at_announcement": "898013",
 "arc_grant_status": "Active",
 "arc_for": "040101,1117,111705,120404",
 "r_number": "158880661",
 "end_year": "2026",
 "participant_list": "Lidia Morawska,Duzgun Agdas,Laurie Buys,Keith Grimwood,Graham Johnson,Guy Marks,Alexander Paz,Thomas Rainey,Zoran Ristovski,Kirsten Spann,Bo Xia,Patrick Chambers,Shaun Clough,Stephen Corbett,Charles dePuthod,Neal Durant,Travis Kline,Peter McGarry,Ross Mensforth,Clive Paige,John Penny,Brad Prezant,Uma Rajappa,David Ward,Scott Bell,Giorgio Buonanno,Mark Jermy,Jose Jimenez,Claire Wainwright,Sandra Glaister,greg Bell,Louisa O'Toole",
 "key": "arc/LP200301123",
 "related_organisations_count": 24
 }
 ]
 }

We can compare it visually with the graph from Neo4j and see that RAG
provided the correct output.

<figure>
<img
src="https://cdn-images-1.medium.com/max/1024/1*PkIQl8fupR-TdiwppqdJkg.png" />
</figure>

We have tested this mini pipeline with other examples, and the
simplicity of the pipeline makes it functional as long as the size of
the subgraph produced in Step 2 does not exceed the context window of
the OpenAI model.

### Conclusion

The integration of OpenAI RAG and Neo4j paves the way for innovative
solutions in knowledge graph applications, and recommendation systems.
RAG is efficient in grounding the LLM on the most recent data, and
verified information and reduces the expense of retraining the model
when new data is available.

![](https://medium.com/_/stat?event=post.clientViewed&referrerSource=full_rss&postId=3a183df816ce){width="1"
height="1"}