#179 · Primary category: Deep Learning Frameworks

RL4LMs

dialogue-generation language-modeling machine-translation natural-language-processing nlp reinforcement-learning summarization table-to-text text-generation

A modular RL library to fine-tune language models to human preferences

Project last updated:03/01/24

GitHub Stars

2.4K

Forks

201

Contributors

8

License

Apache-2.0

Why we included this project

RL4LMs provides the building blocks for reinforcement-learning fine-tuning of language models, so you can focus on the reward and policy rather than the loop plumbing. You pick an on-policy algorithm like PPO or NLPO, a reward function from the library's broad set of NLG metrics, and a causal or seq2seq model as the policy. Predefined YAML configs make runs reproducible, and the library has been benchmarked across tasks like summarization, translation, dialogue, and table-to-text generation. It's a solid starting point for RLHF-style experiments or custom reward design, though the code is research-oriented and last updated in early 2024, so expect to adapt it for production.

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