allenai/reward-bench

RewardBench: the first evaluation tool for reward models.

해결하는 문제

RewardBench는 LLM을 인간 선호에 정렬하는 데 사용되는 보상 모델(RM)의 능력과 안전성을 평가하기 위한 표준화된 방법을 제공합니다. 이는 Direct Preference Optimization(DPO)으로 학습된 모델과 생성형 "LLM-as-a-judge" 모델을 포함해 이러한 모델에 대한 일관된 벤치마킹이 부족한 문제를 해결합니다.

작동 방식

이 프로젝트는 보상 모델이 "chosen"과 "rejected" 응답을 얼마나 잘 구분하는지 테스트하기 위한 도구와 데이터셋 모음을 제공합니다. 포함 내용은 다음과 같습니다:

  • 추론 코드: 다양한 보상 모델(예: Starling, PairRM, OpenAssistant)을 실행하기 위한 통합 코드.
  • 평가 스크립트: 표준 보상 모델(run_rm.py), DPO 모델(run_dpo.py), 그리고 업데이트된 RewardBench 2(run_v2.py)를 위한 전용 스크립트.
  • 생성형 지원: 로컬 vLLM 추론 또는 API(OpenAI, Anthropic, Google Gemini, Together)를 통해 LLM을 심사자로 사용하는 도구.
  • 분석 도구: 결과 시각화, 섹션별 점수(Chat, Safety, Reasoning) 계산, 보상 모델 앙상블 테스트를 위한 유틸리티.

대상 사용자

  • AI 연구자: 새로운 보상 모델이나 선호 최적화 기법을 개발하고 테스트하는 사람.
  • ML 엔지니어: RLHF 파이프라인에 배포하기 전에 보상 모델의 안전성과 품질을 평가하는 사람.
  • 모델 개발자: 공개 리더보드와 자신의 모델을 비교하여 성능을 가늠하는 사람.

하이라이트

  • 폭넓은 모델 지원: 시퀀스 분류기, DPO/KTO 모델, 생성형 심사자와 호환됩니다.
  • RewardBench 2: "best-of-4" 및 "Ties" 데이터셋을 위한 고급 데이터 처리를 포함합니다.
  • 유연한 배포: uv 또는 pip를 통한 설치를 지원하며, 재현 가능한 연구를 위한 Docker 이미지를 제공합니다.
  • 통합: 평가 결과와 모델 메타데이터를 Hugging Face Hub로 직접 푸시할 수 있습니다.

관련

  • 프로젝트
  • 프로젝트
  • 프로젝트
  • 프로젝트
  • 프로젝트