Update README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,71 @@
|
|
| 1 |
-
|
| 2 |
-
|
| 3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
````markdown
|
| 2 |
+
---
|
| 3 |
+
language:
|
| 4 |
+
- en
|
| 5 |
+
tags:
|
| 6 |
+
- GraphRAG
|
| 7 |
+
- retrieval-augmented-generation
|
| 8 |
+
- reinforcement-learning
|
| 9 |
+
- multi-hop-reasoning
|
| 10 |
+
license: mit
|
| 11 |
+
datasets:
|
| 12 |
+
- 2Wiki
|
| 13 |
+
- MuSiQue
|
| 14 |
+
- HotpotQA
|
| 15 |
+
base_model: Qwen2.5-7B & Qwen2.5-7B-Instruct
|
| 16 |
+
---
|
| 17 |
+
|
| 18 |
+
# GraphRAG-R1
|
| 19 |
+
|
| 20 |
+
**This is the official repository for the paper β[GraphRAG-R1: Graph Retrieval-Augmented Generation with Process-Constrained Reinforcement Learning](https://arxiv.org/abs/2507.23581)β (Accepted to WWW '26).**
|
| 21 |
+
|
| 22 |
+
| **Item** | **Details** |
|
| 23 |
+
|:---|:---|
|
| 24 |
+
| **π Paper** | [GraphRAG-R1: Graph Retrieval-Augmented Generation with Process-Constrained Reinforcement Learning](https://arxiv.org/abs/2507.23581) |
|
| 25 |
+
| **π Conference** | **The ACM Web Conference 2026 (WWW '26)** <br> *April 13β17, 2026, Dubai, United Arab Emirates* |
|
| 26 |
+
| **π» Full Code** | **GitHub:** `https://github.com/ycygit/GraphRAG-R1` |
|
| 27 |
+
| **π€ HF Models** | This repository provides the LoRA adapters. See [Model Weights & Usage](#model-weights--usage). |
|
| 28 |
+
| **π License** | MIT |
|
| 29 |
+
|
| 30 |
+
---
|
| 31 |
+
|
| 32 |
+
## π Quick Links
|
| 33 |
+
- [Abstract](#abstract)
|
| 34 |
+
- [Model Weights & Usage](#model-weights--usage)
|
| 35 |
+
- [Citation](#citation)
|
| 36 |
+
- [Contact](#contact)
|
| 37 |
+
|
| 38 |
+
---
|
| 39 |
+
|
| 40 |
+
## π Abstract
|
| 41 |
+
|
| 42 |
+
GraphRAG-R1 is a Graph Retrieval-Augmented Generation framework enhanced with **Process-Constrained Reinforcement Learning**. It is designed to significantly improve the reasoning capabilities of large language models (LLMs) on complex, multi-hop question answering tasks by integrating structured knowledge graph retrieval with constrained reinforcement learning over the reasoning process.
|
| 43 |
+
|
| 44 |
+
**Official BibTeX Citation:**
|
| 45 |
+
```bibtex
|
| 46 |
+
@inproceedings{yu2025graphrag,
|
| 47 |
+
title={GraphRAG-R1: Graph Retrieval-Augmented Generation with Process-Constrained Reinforcement Learning},
|
| 48 |
+
author={Yu, Chuanyue and Zhao, Kuo and Li, Yuhan and Chang, Heng and Feng, Mingjian and Jiang, Xiangzhe and Sun, Yufei and Li, Jia and Zhang, Yuzhi and Li, Jianxin and others},
|
| 49 |
+
booktitle = {Proceedings of the ACM Web Conference 2026 (WWW '26)},
|
| 50 |
+
year={2026}
|
| 51 |
+
}
|
| 52 |
+
````
|
| 53 |
+
|
| 54 |
+
---
|
| 55 |
+
|
| 56 |
+
## π€ Model Weights & Usage
|
| 57 |
+
|
| 58 |
+
This repository provides **LoRA adapters** for the GraphRAG-R1 framework. To use them, you must load them onto the corresponding base model.
|
| 59 |
+
|
| 60 |
+
**Available Adapters:**
|
| 61 |
+
|
| 62 |
+
1. **For `Qwen2.5-7B`**: Adapter for the base model.
|
| 63 |
+
2. **For `Qwen2.5-7B-Instruct`**: Adapter for the instruction-tuned variant.
|
| 64 |
+
|
| 65 |
+
---
|
| 66 |
+
|
| 67 |
+
## π Contact
|
| 68 |
+
|
| 69 |
+
For questions regarding the model or paper, please open an issue in the future GitHub repository or contact the authors.
|
| 70 |
+
|
| 71 |
+
---
|