LGAIFeb 22, 2024

COPR: Continual Human Preference Learning via Optimal Policy Regularization

arXiv:2402.14228v310 citationsh-index: 10
Originality Incremental advance
AI Analysis

This addresses the challenge of adapting RLHF to dynamic human preferences for AI alignment, though it is incremental as it builds on existing continual learning and optimal policy theories.

The paper tackles the problem of catastrophic forgetting in continual learning for aligning large language models with evolving human preferences, proposing the COPR method which outperforms baselines on a new benchmark in reward-based, GPT-4, and human evaluations.

Reinforcement Learning from Human Feedback (RLHF) is commonly utilized to improve the alignment of Large Language Models (LLMs) with human preferences. Given the evolving nature of human preferences, continual alignment becomes more crucial and practical in comparison to traditional static alignment. Nevertheless, making RLHF compatible with Continual Learning (CL) is challenging due to its complex process. Meanwhile, directly learning new human preferences may lead to Catastrophic Forgetting (CF) of historical preferences, resulting in helpless or harmful outputs. To overcome these challenges, we propose the Continual Optimal Policy Regularization (COPR) method, which draws inspiration from the optimal policy theory. COPR utilizes a sampling distribution as a demonstration and regularization constraints for CL. It adopts the Lagrangian Duality (LD) method to dynamically regularize the current policy based on the historically optimal policy, which prevents CF and avoids over-emphasizing unbalanced objectives. We also provide formal proof for the learnability of COPR. The experimental results show that COPR outperforms strong CL baselines on our proposed benchmark, in terms of reward-based, GPT-4 evaluations and human assessment. Furthermore, we validate the robustness of COPR under various CL settings, including different backbones, replay memory sizes, and learning orders.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes