Core Concepts
JaxUED는 JAX를 활용하여 현대 UED 알고리즘의 최소 의존성 구현을 제공하며, 이를 통해 기존 CPU 기반 구현 대비 약 100배 속도 향상을 달성합니다. 연구자들의 UED 연구 가속화를 목표로 합니다.
Abstract
JaxUED는 UED 연구자들을 위한 라이브러리입니다. 핵심 특징은 다음과 같습니다:
- 최소 환경 인터페이스 설계
- 일반적인 UED 알고리즘의 단일 파일 참조 구현, 연구자들의 빠른 실험 지원
- 기존 코드베이스와 비교하여 구현의 품질 검증
- 기존 UED 문헌과 달리, Domain Randomization의 놀라운 효과성 발견
JaxUED는 DR, PLR, PLR⊥, ACCEL, PAIRED 등의 구현을 제공합니다. 이를 통해 연구자들은 UED 알고리즘의 내부 작동 방식을 이해하고 새로운 아이디어를 빠르게 프로토타이핑할 수 있습니다.
Stats
DR 방식의 경우 25개 벽 제한 하에서 0.54의 평균 해결률을 달성했습니다.
PAIRED 방식의 경우 60개 벽 제한 하에서 0.61의 평균 해결률을 달성했습니다.
PLR 방식의 경우 60개 벽 제한 하에서 0.72의 평균 해결률을 달성했습니다.
Quotes
"JaxUED는 UED 알고리즘 개발에 깊이 관여하려는 연구자들을 위한 것입니다."
"우리의 최소 의존성 참조 구현은 현재 최첨단 UED 방법의 내부 작동 방식을 드러내며, 새로운 아이디어의 빠른 프로토타이핑을 가능하게 합니다."
"우리는 기존 UED 문헌과 달리, Domain Randomization이 현재 최첨단 UED 방법과 경쟁적인 성능을 보인다는 것을 발견했습니다."