Gene-R1: Reasoning with Data-Augmented Lightweight LLMs for Gene Set Analysis.

Area of Science:

• Bioinformatics
• Computational Biology
• Genomics

Background:

• Gene set analysis (GSA) is crucial for identifying molecular functions of gene groups.
• LLM-powered methods offer functional annotations and insights for gene sets.
• Current LLM approaches for GSA predominantly use proprietary models, raising cost and privacy concerns.

Purpose of the Study:

• To introduce Gene-R1, a novel framework for gene set analysis.
• To equip lightweight, open-source LLMs with advanced reasoning capabilities for GSA.
• To address the gap in research on advanced reasoning strategies for GSA.

Main Methods:

• Developed Gene-R1, a data-augmented learning framework.
• Integrated step-by-step reasoning capabilities into open-source LLMs.
• Conducted experiments on 1,508 in-distribution and 106 out-of-distribution gene sets.

Main Results:

• Gene-R1 achieved substantial performance gains on in-distribution gene sets, matching commercial LLMs.
• On out-of-distribution gene sets, Gene-R1 demonstrated comparable performance to commercial and large-scale LLMs.
• The framework exhibited robust generalizability across diverse gene sources.

Conclusions:

• Gene-R1 effectively enhances open-source LLMs for gene set analysis.
• The framework provides a cost-effective and privacy-preserving alternative to proprietary models.
• Gene-R1 offers a promising approach for accurate and generalizable gene function annotation.