English

DogeRM: Equipping Reward Models with Domain Knowledge through Model Merging

Computation and Language 2024-10-08 v2

Abstract

Reinforcement learning from human feedback (RLHF) is a popular strategy for aligning large language models (LLMs) with desired behaviors. Reward modeling is a crucial step in RLHF. However, collecting paired preference data for training reward models is often costly and time-consuming, especially for domain-specific preferences requiring expert annotation. To address this challenge, we propose the \textbf{Do}main knowled\textbf{ge} merged \textbf{R}eward \textbf{M}odel (DogeRM), a novel framework that integrates domain-specific knowledge into a general reward model by model merging. The experiments demonstrate that DogeRM enhances performance across different benchmarks and provide a detailed analysis showcasing the effects of model merging, showing the great potential of facilitating model alignment.

Keywords

Cite

@article{arxiv.2407.01470,
  title  = {DogeRM: Equipping Reward Models with Domain Knowledge through Model Merging},
  author = {Tzu-Han Lin and Chen-An Li and Hung-yi Lee and Yun-Nung Chen},
  journal= {arXiv preprint arXiv:2407.01470},
  year   = {2024}
}

Comments

In the 2024 Conference on Empirical Methods in Natural Language Processing (EMNLP 2024). The code for our work is available at: https://github.com/MiuLab/DogeRM

R2 v1 2026-06-28T17:25:15.720Z