Repo for "MUR: Momentum Uncertainty Guided Reasoning for Large Language Models"
- [2025/07/22] 🔥🔥🔥 Our paper is released !!!
- [2025/07/19] 🔥🔥🔥 Our github repo is released!!!
MUR reduces computation by over 50% on average across three backbone models, while improving accuracy by 0.62–3.37%.
To use MUR, we can try with the following command.
Firstly, create the environment and install the requirements. This implementation is accelerated and supported by vllm.
# env
conda create -n mur python==3.11.9
conda activate mur
pip install -r requirements.txtNext, simply run different python files:
python [TTS setting]-[per_step_scale|mur].pyFinally, run eval files. To be specific, please eval gpqa_diamond dataset using eval/eval_gpqa_cot.py. Adiitionaly, use eval/math_verifier.py to verify math datasets.
Feel free to contact with me if you have any questions ~~~