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

GraphWiz: An Instruction-Following Language Model for Graph Problems

About

Large language models (LLMs) have achieved impressive success across several fields, but their proficiency in understanding and resolving complex graph problems is less explored. To bridge this gap, we introduce GraphInstruct, a novel and comprehensive instruction-tuning dataset designed to equip language models with the ability to tackle a broad spectrum of graph problems using explicit reasoning paths. Utilizing GraphInstruct, we build GraphWiz, an open-source language model capable of resolving various graph problem types while generating clear reasoning processes. To enhance the model's capability and reliability, we incorporate the Direct Preference Optimization (DPO) framework into the graph problem-solving context. The enhanced model, GraphWiz-DPO, achieves an average accuracy of 65% across nine tasks with different complexity levels, surpassing GPT-4 which has an average accuracy of 43.8%. Moreover, our research delves into the delicate balance between training data volume and model performance, highlighting the potential for overfitting with increased data. We also explore the transferability of the model's reasoning ability across different graph tasks, indicating the model's adaptability and practical application potential. Our investigation offers a new blueprint and valuable insights for developing LLMs specialized in graph reasoning and problem-solving.

Nuo Chen, Yuhan Li, Jianheng Tang, Jia Li• 2024

Related benchmarks

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

Other info

Follow for update