Multi-Agent Collaborative Reward Design for Enhancing Reasoning in Reinforcement Learning
Yang, Zhang, Wang et al.
We present CRM (Multi-Agent Collaborative Reward Model), a framework that replaces a single black-box reward model with a coordinated team of specialist evaluators to improve robustness and interpretability in RLHF. Conventional reward models struggle to jointly optimize multiple, sometimes conflicting, preference dimensions (e.g., factuality, helpfulness, safety) and offer limited transparency into why a score is assigned. CRM addresses these issues by decomposing preference evaluation into domain-specific agents that each produce partial signals, alongside global evaluators such as ranker-based and embedding-similarity rewards. A centralized aggregator fuses these signals at each timestep, balancing factors like step-wise correctness, multi-agent agreement, and repetition penalties, yielding a single training reward compatible with standard RL pipelines. The policy is optimized with advantage-based updates (e.g., GAE), while a value model regresses to the aggregated reward, enabling multi-perspective reward shaping without requiring additional human annotations beyond those used to train the evaluators. To support training and assessment, we introduce rewardBench, a benchmark and training suite aligned with the collaborative structure of CRM. Together, CRM and rewardBench provide a practical, modular path to more transparent reward modeling and more stable optimization.
제목: Multi-Agent Collaborative Reward Design for Enhancing Reasoning in Reinforcement Learning
저자: Pei Yang (Gradient), Ke Zhang (Waseda University), Ji Wang (Columbia University), Xiao Chen (Hong Kong Polytechnic University), Yuxin Tang (Rice University & Gradient Network), Eric Yang, Lynn Ai, Bill Shi (Gradient)
본 논문은 CRM(협력 보상 모델, Collaborative Reward Model) 프레임워크를 제안하며, 단일 블랙박스 보상 모델을 대체하는 조율된 전문가 평가 팀을 통해 RLHF(인간 피드백으로부터의 강화학습)의 견고성과 해석 가능성을 향상시킵니다. 기존 보상 모델은 사실성, 유용성, 안전성과 같이 잠재적으로 충돌하는 여러 선호도 차원을 동시에 최적화하기 어려우며, 점수 부여 이유에 대한 투명성이 제한적입니다. CRM은 선호도 평가를 특정 영역의 에이전트로 분해하여 이러한 문제를 해결하며, 각 에이전트는 부분 신호를 생성하고 순위 지정 및 임베딩 유사도 기반의 글로벌 평가기와 함께 작동합니다. 중앙화된 집계기는 각 시간 단계에서 이러한 신호를 융합하여 단계별 정확성, 다중에이전트 일관성 및 반복 페널티 등의 요소를 균형 있게 조정하며, 표준 RL 파이프라인과 호환되는 단일 훈련 보상을 생성합니다. 본 논문은 또한 모듈식, 해석 가능한 보상 모델링을 위한 실용적 경로를 제공하는 RewardBench 벤치마크 스위트를 소개합니다.
Christiano et al. (2017) - Deep reinforcement learning from human preferences
Ouyang et al. (2022) - InstructGPT: Training language models to follow instructions with human feedback
보상 모델링:
Coste et al. (2023) - Reward model ensembles help mitigate overoptimization
Wang et al. (2024) - Interpretable preferences via multi-objective reward modeling
다중에이전트 평가:
Irving et al. (2018) - AI safety via debate
Chan et al. (2023) - ChatEval: Towards better LLM-based evaluators through multi-agent debate
세밀한 피드백:
Zheng et al. (2024) - GRPO: Guided reinforcement preference optimization
Ankner et al. (2024) - Critique-out-loud reward models
종합 평가: 본 논문은 RLHF의 해석 가능성과 추론 능력 향상에 있어 중요한 기여를 하는 혁신적이고 실용적인 다중에이전트 협력 보상 모델링 프레임워크를 제안합니다. 실험 규모 제한 및 구현 세부 사항 부족 등의 문제가 있지만, 핵심 아이디어는 중요한 학술 가치와 응용 전망을 가집니다. 후속 연구에서 더 많은 구현 세부 사항 보충, 실험 규모 확대, 관련 코드 및 모델 오픈소스화를 통해 커뮤니티 발전을 촉진하기를 기대합니다.