paper-with-me

Papers

Iterative Data Smoothing: Mitigating Reward Overfitting and Overoptimization in RLHF

2024-01-29 · Banghua Zhu, Michael I. Jordan, Jiantao Jiao

Reinforcement Learning from Human Feedback (RLHF) is a pivotal technique that aligns language models closely with human-centric values. The initial phase of RLHF involves learning human values using a reward model from ranking data. It is observed that the performance of the reward model degrades after one epoch of training, and optimizing too much against the learned reward model eventually hinders the true objective. This paper delves into these issues, leveraging the theoretical insights to design improved reward learning algorithm termed 'Iterative Data Smoothing' (IDS). The core idea is that during each training epoch, we not only update the model with the data, but also update the date using the model, replacing hard labels with soft labels. Our empirical findings highlight the superior performance of this approach over the traditional methods.

📄 PDF Abstract BibTeX arXiv:2401.16335

Code (0)

등록된 구현이 없습니다.

Similar Papers 제목 키워드 기반

Rethinking Label Smoothing on Multi-hop Question Answering

2022-12-19 · Zhangyue Yin, Yuxin Wang, Xiannian Hu, Yiguang Wu 외

Multi-Hop Question Answering (MHQA) is a significant area in question answering, requiring multiple reasoning components, including document retrieval, supporting sentence prediction, and answer span extraction. In this …

Image ClassificationMachine Reading ComprehensionMachine TranslationMulti-hop Question Answering+5

Efficient Preference-based Reinforcement Learning via Aligned Experience Estimation

2024-05-29 · Fengshuo Bai, Rui Zhao, Hongming Zhang, Sijia Cui 외

Preference-based reinforcement learning (PbRL) has shown impressive capabilities in training agents without reward engineering. However, a notable limitation of PbRL is its dependency on substantial human feedback. This …

reinforcement-learningReinforcement Learning

The Energy Loss Phenomenon in RLHF: A New Perspective on Mitigating Reward Hacking

2025-01-31 · Yuchun Miao, Sen Zhang, Liang Ding, Yuqi Zhang 외

This work identifies the Energy Loss Phenomenon in Reinforcement Learning from Human Feedback (RLHF) and its connection to reward hacking. Specifically, energy loss in the final layer of a Large Language Model (LLM) grad…

Large Language Model

Towards Mitigating Architecture Overfitting on Distilled Datasets

2023-09-08 · Xuyang Zhong, Chen Liu

Dataset distillation methods have demonstrated remarkable performance for neural networks trained with very limited training data. However, a significant challenge arises in the form of \textit{architecture overfitting}:…

Dataset DistillationKnowledge Distillation

LeanPO: Lean Preference Optimization for Likelihood Alignment in Video-LLMs

2025-06-05 · Xiaodong Wang, Jinfa Huang, Li Yuan, Peixi Peng

Most Video Large Language Models (Video-LLMs) adopt preference alignment techniques, e.g., DPO~\citep{rafailov2024dpo}, to optimize the reward margin between a winning response ($y_w$) and a losing response ($y_l$). Howe…