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

G1: Teaching LLMs to Reason on Graphs with Reinforcement Learning

About

Although Large Language Models (LLMs) have demonstrated remarkable progress, their proficiency in graph-related tasks remains notably limited, hindering the development of truly general-purpose models. Previous attempts, including pretraining graph foundation models or employing supervised fine-tuning, often face challenges such as the scarcity of large-scale, universally represented graph data. We introduce G1, a simple yet effective approach demonstrating that Reinforcement Learning (RL) on synthetic graph-theoretic tasks can significantly scale LLMs' graph reasoning abilities. To enable RL training, we curate Erd\~os, the largest graph reasoning dataset to date comprising 50 diverse graph-theoretic tasks of varying difficulty levels, 100k training data and 5k test data, all drived from real-world graphs. With RL on Erd\~os, G1 obtains substantial improvements in graph reasoning, where our finetuned 3B model even outperforms Qwen2.5-72B-Instruct (24x size). RL-trained models also show strong zero-shot generalization to unseen tasks, domains, and graph encoding schemes, including other graph-theoretic benchmarks as well as real-world node classification and link prediction tasks, without compromising general reasoning abilities. Our findings offer an efficient, scalable path for building strong graph reasoners by finetuning LLMs with RL on graph-theoretic tasks, which combines the strengths of pretrained LLM capabilities with abundant, automatically generated synthetic data, suggesting that LLMs possess graph understanding abilities that RL can elicit successfully. Our implementation is open-sourced at https://github.com/PKU-ML/G1, with models and datasets hosted on Hugging Face collections https://huggingface.co/collections/PKU-ML/g1-683d659e992794fc99618cf2 for broader accessibility.

Xiaojun Guo, Ang Li, Yifei Wang, Stefanie Jegelka, Yisen Wang• 2025

Related benchmarks

TaskDatasetResultRank
Graph Edit DistanceGED Small-scale
Accuracy52
12
Graph Edit Distance (GED)Large-scale graphs
Accuracy0.03
12
NP-hard graph reasoningSmall-scale NP-hard graph problems Average
Accuracy40
12
Traveling Salesman ProblemTSP Small-scale
Accuracy38
12
Maximum Clique Problem (MCP)Large-scale graphs
Accuracy3
12
NP-hard Graph Problems (Aggregate)Large-scale graphs
Accuracy2
12
Maximum Clique ProblemMCP Small-scale
Accuracy30
12
Traveling Salesperson Problem (TSP)Large-scale graphs
Accuracy0.00e+0
12
Showing 8 of 8 rows

Other info

Follow for update