PlanePaper commited on
Commit
e5d3d75
·
verified ·
1 Parent(s): 69e9bfa

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +58 -0
README.md ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ language:
4
+ - en
5
+ metrics:
6
+ - accuracy
7
+ base_model:
8
+ - deepseek-ai/DeepSeek-R1-Distill-Qwen-14B
9
+ ---
10
+
11
+ # 🚀 GRPO-LEAD: Efficient Reasoning Enhancement for Mathematical Tasks
12
+
13
+ ---
14
+
15
+ ## 📚 Overview
16
+
17
+ **GRPO-LEAD** (**GRPO** with **L**ength-dependent rewards, **E**xplicit penalties, and **A**dvantage reweighting for **D**ifficulty) is an advanced reinforcement learning pipeline designed to fine-tune large language models (LLMs) for concise, accurate, and efficient reasoning in mathematical tasks.
18
+
19
+ ---
20
+
21
+ ## 📊 Performance Benchmarks
22
+
23
+ The following benchmarks were conducted on AIME24 and AIME25 datasets, evaluated with parameters: 14k maximum tokens, temperature of 0.6, min-p of 0.01, and 32 samples per question.
24
+
25
+ | **Model** | **AIME24 Cons@32** | **AIME24 Pass@1** | **AIME24 Avg. Length** | **AIME25 Cons@32** | **AIME25 Pass@1** | **AIME25 Avg. Length** |
26
+ |---------------------|--------------------|-------------------|------------------------|--------------------|-------------------|------------------------|
27
+ | **DeepSeek-Distlled-14B** | 0.800 | 0.614 | 9182 | 0.633 | 0.429 | 10046 |
28
+ | **Light-R1-14B-DS** | 0.833 | 0.641 | 9571 | 0.767 | 0.505 | 10194 |
29
+ | **LEAD-14B (ours)** | **0.867** | **0.650** | **8267** | **0.767** | **0.539** | **8668** |
30
+
31
+ Our GRPO-LEAD model achieves superior consistency and higher accuracy, demonstrating significantly improved reasoning efficiency as evidenced by shorter average reasoning lengths.
32
+
33
+ ---
34
+
35
+ ## 📂 Code and Documentation
36
+
37
+ For complete details, codebase, and usage examples, please visit our GitHub repository:
38
+
39
+ [**📌 GitHub Repository**](https://github.com/aeroplanepaper/GRPO-LEAD)
40
+
41
+ ---
42
+
43
+ ## 📖 Citation
44
+
45
+ If you find our work useful, please cite it as:
46
+
47
+ ```bibtex
48
+ @misc{zhang2025grpoleaddifficultyawarereinforcementlearning,
49
+ title={GRPO-LEAD: A Difficulty-Aware Reinforcement Learning Approach for Concise Mathematical Reasoning in Language Models},
50
+ author={Jixiao Zhang and Chunsheng Zuo},
51
+ year={2025},
52
+ eprint={2504.09696},
53
+ archivePrefix={arXiv},
54
+ primaryClass={cs.CL},
55
+ url={https://arxiv.org/abs/2504.09696},
56
+ }
57
+ ```
58
+ Enjoy exploring GRPO-LEAD! 🚀✨