Understanding the performance gap between online and offline alignment algorithms
Fuente:
arXiv
Saved in:
| Main Authors: | , , , , , , , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2024
|
| Subjects: | |
| Online Access: | |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866929343317409792 |
|---|---|
| author | Tang, Yunhao Guo, Daniel Zhaohan Zheng, Zeyu Calandriello, Daniele Cao, Yuan Tarassov, Eugene Munos, Rémi Pires, Bernardo Ávila Valko, Michal Cheng, Yong Dabney, Will |
| author_facet | Tang, Yunhao Guo, Daniel Zhaohan Zheng, Zeyu Calandriello, Daniele Cao, Yuan Tarassov, Eugene Munos, Rémi Pires, Bernardo Ávila Valko, Michal Cheng, Yong Dabney, Will |
| contents | Reinforcement learning from human feedback (RLHF) is the canonical framework for large language model alignment. However, rising popularity in offline alignment algorithms challenge the need for on-policy sampling in RLHF. Within the context of reward over-optimization, we start with an opening set of experiments that demonstrate the clear advantage of online methods over offline methods. This prompts us to investigate the causes to the performance discrepancy through a series of carefully designed experimental ablations. We show empirically that hypotheses such as offline data coverage and data quality by itself cannot convincingly explain the performance difference. We also find that while offline algorithms train policy to become good at pairwise classification, it is worse at generations; in the meantime the policies trained by online algorithms are good at generations while worse at pairwise classification. This hints at a unique interplay between discriminative and generative capabilities, which is greatly impacted by the sampling process. Lastly, we observe that the performance discrepancy persists for both contrastive and non-contrastive loss functions, and appears not to be addressed by simply scaling up policy networks. Taken together, our study sheds light on the pivotal role of on-policy sampling in AI alignment, and hints at certain fundamental challenges of offline alignment algorithms. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2405_08448 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Understanding the performance gap between online and offline alignment algorithms Tang, Yunhao Guo, Daniel Zhaohan Zheng, Zeyu Calandriello, Daniele Cao, Yuan Tarassov, Eugene Munos, Rémi Pires, Bernardo Ávila Valko, Michal Cheng, Yong Dabney, Will Machine Learning Artificial Intelligence Reinforcement learning from human feedback (RLHF) is the canonical framework for large language model alignment. However, rising popularity in offline alignment algorithms challenge the need for on-policy sampling in RLHF. Within the context of reward over-optimization, we start with an opening set of experiments that demonstrate the clear advantage of online methods over offline methods. This prompts us to investigate the causes to the performance discrepancy through a series of carefully designed experimental ablations. We show empirically that hypotheses such as offline data coverage and data quality by itself cannot convincingly explain the performance difference. We also find that while offline algorithms train policy to become good at pairwise classification, it is worse at generations; in the meantime the policies trained by online algorithms are good at generations while worse at pairwise classification. This hints at a unique interplay between discriminative and generative capabilities, which is greatly impacted by the sampling process. Lastly, we observe that the performance discrepancy persists for both contrastive and non-contrastive loss functions, and appears not to be addressed by simply scaling up policy networks. Taken together, our study sheds light on the pivotal role of on-policy sampling in AI alignment, and hints at certain fundamental challenges of offline alignment algorithms. |
| title | Understanding the performance gap between online and offline alignment algorithms |
| topic | Machine Learning Artificial Intelligence |
| url | https://arxiv.org/abs/2405.08448 |