CUDA Agent: 고성능 커널 생성 에이전트
[Paper] CUDA Agent: Large-Scale Agentic RL for High-Performance CUDA Kernel Generation
·2026.08.18 15:33
핵심 내용
강화 학습 기반의 CUDA Agent가 기존 모델 대비 뛰어난 CUDA 커널 생성 성능을 입증했다.
자세히 보기
GPU 커널 최적화는 현대 딥러닝의 핵심이지만, 고도의 하드웨어 전문 지식이 필요한 매우 까다로운 작업입니다.
CUDA Agent는 대규모 에이전트 강화 학습(Agentic RL)을 통해 CUDA 커널 생성 전문성을 확보하는 시스템입니다. 이 시스템은 다음 세 가지 핵심 구성 요소를 갖추고 있습니다:
- 확장 가능한 데이터 합성 파이프라인
- 자동 검증 및 프로파일링을 통해 신뢰할 수 있는 보상 신호를 제공하는 스킬 증강 CUDA 개발 환경
- 안정적인 학습을 지원하는 강화 학습 알고리즘
성능 측정 결과, CUDA Agent는 KernelBench의 모든 레벨(Level-1, 2, 3)에서 기존 방식 대비 92%~100% 더 빠른 속도를 기록했습니다. 특히 가장 어려운 Level-3 설정에서는 Claude Opus 4.5 및 Gemini 3 Pro와 같은 강력한 상용 모델보다 약 40% 더 높은 성능을 달성했습니다.
이 한국어 요약은 AI가 자동으로 만들었습니다. 원문의 주장과 맥락은 원문에서 확인해 주세요. 저작권은 원저작자에게 있습니다.