Learning Guarantee of Reward Modeling Using Deep Neural Networks

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Luo, Yuanhang, Ge, Yeheng, Han, Ruijian, Shen, Guohao
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866915282353651712
author Luo, Yuanhang
Ge, Yeheng
Han, Ruijian
Shen, Guohao
author_facet Luo, Yuanhang
Ge, Yeheng
Han, Ruijian
Shen, Guohao
contents In this work, we study the learning theory of reward modeling with pairwise comparison data using deep neural networks. We establish a novel non-asymptotic regret bound for deep reward estimators in a non-parametric setting, which depends explicitly on the network architecture. Furthermore, to underscore the critical importance of clear human beliefs, we introduce a margin-type condition that assumes the conditional winning probability of the optimal action in pairwise comparisons is significantly distanced from 1/2. This condition enables a sharper regret bound, which substantiates the empirical efficiency of Reinforcement Learning from Human Feedback and highlights clear human beliefs in its success. Notably, this improvement stems from high-quality pairwise comparison data implied by the margin-type condition, is independent of the specific estimators used, and thus applies to various learning algorithms and models.
format Preprint
id arxiv_https___arxiv_org_abs_2505_06601
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Learning Guarantee of Reward Modeling Using Deep Neural Networks
Luo, Yuanhang
Ge, Yeheng
Han, Ruijian
Shen, Guohao
Machine Learning
In this work, we study the learning theory of reward modeling with pairwise comparison data using deep neural networks. We establish a novel non-asymptotic regret bound for deep reward estimators in a non-parametric setting, which depends explicitly on the network architecture. Furthermore, to underscore the critical importance of clear human beliefs, we introduce a margin-type condition that assumes the conditional winning probability of the optimal action in pairwise comparisons is significantly distanced from 1/2. This condition enables a sharper regret bound, which substantiates the empirical efficiency of Reinforcement Learning from Human Feedback and highlights clear human beliefs in its success. Notably, this improvement stems from high-quality pairwise comparison data implied by the margin-type condition, is independent of the specific estimators used, and thus applies to various learning algorithms and models.
title Learning Guarantee of Reward Modeling Using Deep Neural Networks
topic Machine Learning
url https://arxiv.org/abs/2505.06601