Boosting Reward Model with Preference-Conditional Multi-Aspect Synthetic Data Generation

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Shen, Jiaming, Xu, Ran, Jun, Yennie, Qin, Zhen, Liu, Tianqi, Yang, Carl, Liang, Yi, Baumgartner, Simon, Bendersky, Michael
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866913736899428352
author Shen, Jiaming
Xu, Ran
Jun, Yennie
Qin, Zhen
Liu, Tianqi
Yang, Carl
Liang, Yi
Baumgartner, Simon
Bendersky, Michael
author_facet Shen, Jiaming
Xu, Ran
Jun, Yennie
Qin, Zhen
Liu, Tianqi
Yang, Carl
Liang, Yi
Baumgartner, Simon
Bendersky, Michael
contents Reward models (RMs) are crucial for aligning large language models (LLMs) with human preferences. They are trained using preference datasets where each example consists of one input prompt, two responses, and a preference label. As curating a high-quality human labeled preference dataset is both time-consuming and expensive, people often rely on existing powerful LLMs for preference label generation. This can potentially introduce noise and impede RM training. In this work, we present RMBoost, a novel synthetic preference data generation paradigm to boost reward model quality. Unlike traditional methods, which generate two responses before obtaining the preference label, RMBoost first generates one response and selects a preference label, followed by generating the second more (or less) preferred response conditioned on the pre-selected preference label and the first response. This approach offers two main advantages. First, RMBoost reduces labeling noise since preference pairs are constructed intentionally. Second, RMBoost facilitates the creation of more diverse responses by incorporating various quality aspects (e.g., helpfulness, relevance, completeness) into the prompts. We conduct extensive experiments across three diverse datasets and demonstrate that RMBoost outperforms other synthetic preference data generation techniques and significantly boosts the performance of four distinct reward models.
format Preprint
id arxiv_https___arxiv_org_abs_2407_16008
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Boosting Reward Model with Preference-Conditional Multi-Aspect Synthetic Data Generation
Shen, Jiaming
Xu, Ran
Jun, Yennie
Qin, Zhen
Liu, Tianqi
Yang, Carl
Liang, Yi
Baumgartner, Simon
Bendersky, Michael
Computation and Language
Reward models (RMs) are crucial for aligning large language models (LLMs) with human preferences. They are trained using preference datasets where each example consists of one input prompt, two responses, and a preference label. As curating a high-quality human labeled preference dataset is both time-consuming and expensive, people often rely on existing powerful LLMs for preference label generation. This can potentially introduce noise and impede RM training. In this work, we present RMBoost, a novel synthetic preference data generation paradigm to boost reward model quality. Unlike traditional methods, which generate two responses before obtaining the preference label, RMBoost first generates one response and selects a preference label, followed by generating the second more (or less) preferred response conditioned on the pre-selected preference label and the first response. This approach offers two main advantages. First, RMBoost reduces labeling noise since preference pairs are constructed intentionally. Second, RMBoost facilitates the creation of more diverse responses by incorporating various quality aspects (e.g., helpfulness, relevance, completeness) into the prompts. We conduct extensive experiments across three diverse datasets and demonstrate that RMBoost outperforms other synthetic preference data generation techniques and significantly boosts the performance of four distinct reward models.
title Boosting Reward Model with Preference-Conditional Multi-Aspect Synthetic Data Generation
topic Computation and Language
url https://arxiv.org/abs/2407.16008