#157 · Primary category: MLOps & Evaluation
RLHF-Reward-Modeling
Recipes to train reward model for RLHF.
Project last updated:04/24/25
GitHub Stars
1.5K
Forks
110
Contributors
14
License
Apache-2.0
Why we included this project
Reward model training is often the fiddliest part of an RLHF pipeline, and this repository collects the recipes the RLHFlow team used to train their released models. It covers the classic Bradley-Terry reward model and pairwise preference models, then goes further with research variants like ArmoRM's multi-objective mixture-of-experts aggregation, length-bias-corrected reward modeling, and semi-supervised self-training. The training data, code, and hyperparameters for each approach are included, so you can reproduce the results on your own preference data instead of reconstructing the setup from a paper. The decision-tree reward model is a nice option if you want a scoring signal you can actually inspect rather than a black-box network. For teams already at home with Hugging Face training loops, this is a useful reference for comparing reward model designs side by side.
Articles for this project
No articles for this project yet.
To suggest a topic or contribute an article, contact us.
Related projects in this category
unsloth
Local UI to run and train LLMs and diffusion models. Supports GGUF, MLX, Qwen3.8, Kimi K3, MiniMax-H3, Gemma 4, FLUX and more.
LlamaFactory
Unified Efficient Fine-Tuning of 100+ LLMs & VLMs (ACL 2024)
airflow
Apache Airflow - A platform to programmatically author, schedule, and monitor workflows
langfuse
🪢 Open source AI engineering platform: LLM evals, observability, metrics, prompt management, playground, datasets. Integrates with OpenTelemetry, LangChain, OpenAI SDK, LiteLLM, and more. 🍊YC W23
netron
Visualizer for neural network, deep learning and machine learning models