Comparative Performance Evaluation of Large Language Models for Extracting Molecular Interactions and Pathway Knowledge

Gilchan Park, Byung-Jun Yoon, Xihaier Luo, Vanessa López-Marrero, Patrick R. Johnstone, Shinjae Yoo, Francis J. AlexanderJ. Comput. Biol 2023 · OpenReview

TL;DR

This paper evaluates how well 15 different Large Language Models (LLMs) can extract crucial biological information like protein interactions, pathway associations, and gene regulatory relationships from scientific texts. The study found that larger LLMs generally perform better, but all models face challenges with complex or less-studied biological entities, highlighting the need for further advancements in LLM capabilities for biological knowledge discovery.

What Problem Does This Paper Address?

Understanding the intricate web of interactions between molecules (like proteins and genes) and how they form biological pathways is fundamental to deciphering cellular processes, disease mechanisms, and developing new treatments. While extensive databases exist, they are often incomplete and require significant human effort to maintain and update. This laborious process limits our ability to quickly access and utilize this vital information. The paper aims to address this by exploring whether Large Language Models (LLMs), with their advanced natural language processing capabilities, can automate the extraction of this complex molecular and pathway knowledge from scientific literature, making it more accessible and efficient for researchers.

Methods

The researchers evaluated 15 open-source LLMs on three key biological tasks:

  1. Protein-Protein Interaction (PPI) Recognition: Identifying which proteins interact with each other. This involved both a generative task (listing interacting proteins) and a binary task (yes/no, do they interact?).
  2. Pathway Association: Identifying genes associated with specific human pathways, particularly those affected by low-dose radiation (LDR).
  3. Gene Regulatory Relations: Determining the type of relationship (e.g., activation, inhibition) between genes based on text descriptions.

They employed a rigorous evaluation framework, including the use of verified negative PPI samples and carefully selected pathway datasets. The models were tested using various prompt engineering techniques and evaluated using metrics like micro F1, macro F1, and the count of fully matched predictions.

Datasets

The study utilized several established biological databases:

  • STRING DB: For protein-protein interaction data.
  • Negatome 2.0: A curated database of non-interacting protein pairs, used to ensure robust evaluation of PPI prediction.
  • KEGG database: For human pathways, specifically focusing on pathways affected by Low-Dose Radiation (LDR) exposure.
  • INDRA database: For gene regulatory relationships, containing text statements extracted from scientific papers.

The researchers also developed specific benchmarks for each task, including a curated set of LDR-affected KEGG pathways and a multiple-choice question format for gene regulatory relations.

Key Insights and Ideas

  • Bigger is (Often) Better: Larger LLMs generally demonstrated superior performance across most tasks, suggesting that their extensive training data and parameter count are crucial for capturing complex biological relationships.
  • Domain-Specific Models Show Promise: Smaller, domain-specific models like BioGPT-Large and BioMedLM showed significant improvements in certain tasks, sometimes outperforming larger general-purpose models. This highlights the value of specialized training for biological applications.
  • Nuance Matters: The discrepancy between micro F1 and macro F1 scores in PPI tasks suggests that while models might identify some correct interactions (micro F1), they may struggle with a comprehensive understanding of all interactions for a given protein (macro F1).
  • Well-Studied vs. Under-Studied: Models performed better on well-characterized proteins and pathways, likely because these entities are more extensively documented in scientific literature. Less-studied or uncharacterized biological entities posed a significant challenge.
  • Task Matters: Models showed a notable difference in performance between tasks. For instance, they generally performed better at identifying genes related to pathways than at predicting protein-protein interactions, possibly due to the more structured nature of pathway names.

Key Results

  • PPI Task: Mixtral-8x7B-Instruct (46B) achieved the highest performance in generating PPIs, followed by LLaMA2-Chat (70B). For the yes/no PPI task, chat-based LLMs like MPT-Chat (7B) and LLaMA-2-Chat (70B) showed superior performance.
  • Pathway Task: Mixtral-8x7B-Instruct (46B) also led in identifying genes related to LDR-affected pathways, with BioMedLM and Galactica (30B) also performing well.
  • Gene Regulatory Relations Task: Mixtral-8x7B-Instruct (46B) again showed the highest performance, with SOLAR-Instruct (10.7B) achieving the second-best score.
  • Limitations: Models struggled with identifying groups of genes/proteins with diverse functions and recognizing highly correlated gene regulatory relationships. Falcon models, in particular, showed significantly lower performance in some tasks.

Limitations and Future Work

The authors acknowledge that LLMs still face challenges in accurately identifying less-studied biological entities and highly correlated gene regulatory relationships. They suggest that future work should focus on:

  • Prompt Engineering: Exploring advanced techniques like Retrieval Augmented Generation (RAG), Chain-of-Thought (CoT), and Tree-of-Thought (ToT) to incorporate external contextual information into prompts.
  • Parameter-Efficient Fine-Tuning (PEFT): Fine-tuning LLMs on specific downstream tasks to improve prediction outcomes while reducing computational demands.
  • Addressing Hallucinations: While not explicitly detailed as a limitation in the abstract, the paper mentions the need for meticulous fact-checking, implying that LLMs can sometimes generate incorrect or misleading information.

The study provides a valuable benchmark for current LLM capabilities in biological knowledge extraction and points towards promising avenues for future research to enhance their utility in this domain.