벤치마크 · agentic
RE-Bench
RE-Bench(METR의 Research Engineering Benchmark)는 AI 에이전트가 개방형 AI R&D / ML 엔지니어링 과제를 인간 전문가와 비교해 얼마나 잘 수행하는지를 측정한다. 각 시도는 과제별 연속 점수를 받으며, 0은 주어진 시작 해법, 1은 강력한 인간 참조 해법에 대응하도록 정규화된다.
자세히 보기
- 예시
- 최적화된 GPU kernel 작성: 특정 계산(예: prefix-sum)의 참조 구현이 주어지면, 기능적으로 동등하면서 가능한 한 빠르게 실행되는 커스텀 kernel을 만든다. 다른 환경으로는 embedding 가중치가 뒤섞인 모델의 성능 복구나 fine-tuning 스크립트 속도 향상 등이 있다.
- 채점 방식
- 7개 환경 각각에는 연속 점수(예: runtime, loss, 정확도)를 반환하는 자동 채점 함수가 포함된다. 원시 점수는 환경별로 정규화되어 0 = 주어진 시작 해법, 1 = 강력한 참조 해법이 된다(점수는 1을 넘을 수 있다). 에이전트는 실행 중 스스로 채점기를 돌릴 수 있으며, 고정된 시간/연산 예산 내에서 도달한 최고 점수가 인정되고, 종합 성능은 동일한 총 시간 예산에서 인간 전문가 점수와 비교된다.
- 검증 방식
- 해법은 과제의 sandbox에서 실행되고 결정론적 채점 함수로 채점될 때에만 인정되며, 예산 내 최고의 유효 점수가 기록된다. 인간 기준선은 같은 환경을 수행한 숙련된 ML 연구자·엔지니어에게서 나오며(공개 데이터는 61명 전문가의 71회 8시간 시도), 이를 통해 동등한 비교가 가능하다. 결과는 여러 시도와 seed에 걸쳐 집계된다.
- 왜 중요한가
- 이는 프런티어 AI 안전 정책이 가장 주목하는 역량 — AI가 AI R&D를 자동화하는 것 — 을, 현실적이고 개방형인 엔지니어링 문제, 객관적 지표, 인간 전문가와의 직접 비교를 사용해 겨냥한다. 그래서 다른 벤치마크가 측정하는 고전적 ML 기술이 아니라, 자율적 R&D 역량의 위험한 수준을 예측하고 평가하기 위한 대리 지표가 된다.
예제 풀이
문제
「Optimize LLM Foundry finetuning script」: 고정된 GPU에서 언어 모델을 fine-tuning하는 동작하는 스크립트가 주어진다. 학습 동작을 바꾸지 않으면서 그 wall-clock runtime을 최대한 줄여라 — 최종 loss는 여전히 허용 오차 내에서 참조와 일치해야 한다. 무엇이든 수정할 수 있으며, 자신의 runtime을 측정하기 위해 채점 함수를 실행할 수 있다.
해답
먼저 스크립트를 profiling해 병목을 찾고, 결과를 바꾸지 않는 runtime 최적화를 적용한다: 안전한 곳에서의 저정밀/혼합 정밀도(bf16), fused 또는 최적화된 kernel(FlashAttention, fused optimizer), torch.compile, 더 빠른 데이터 로딩 파이프라인, 그리고 중복 연산과 불필요한 CUDA 동기화 제거. 최종 답: 출력이 여전히 허용 오차 내에서 참조와 일치하는 가장 빠른 스크립트 — 유일한 해법은 없으며, 점수는 그 제약 하에서 달성한 runtime 단축량이다.
풀이
이것이 옳은 이유는 환경이 동작/loss 제약 하에서 실측 wall-clock runtime으로 채점하기 때문으로, 출력을 허용 오차 내로 유지하면서 runtime을 줄이는 어떤 변경이든 정규화 점수를 높인다(0 = 시작 스크립트, 1 = 강력한 참조). 채점은 자동이다: 채점기가 제출된 스크립트를 다시 실행하고 그 runtime을 기록한다.
이 벤치마크에 대해 아직 검증된 점수가 없습니다.