샘플 효율 극대화: 가중치 정규화를 통한 강화학습 확장성 향상
본 연구는 가중치 정규화를 통해 CrossQ 강화학습 알고리즘의 확장성을 크게 향상시켰습니다. 높은 UTD 비율에서도 안정적인 학습과 경쟁력 있는 성능을 달성하여, 실제 응용 분야에서의 강화학습 활용 가능성을 높였습니다.

강화학습(Reinforcement Learning)은 놀라운 발전을 이루었지만, 실제 응용 분야에서는 여전히 샘플 효율이 큰 걸림돌입니다. 최근 CrossQ는 낮은 업데이트-데이터 비율(UTD, update-to-data ratio)로 최첨단 샘플 효율을 달성하며 주목받았습니다. 하지만, Daniel Palenicek, Florian Vogt, Jan Peters 등의 연구진은 CrossQ의 높은 UTD 비율 환경에서의 확장성에 대한 심층적인 연구를 진행했습니다.
연구 결과, 높은 UTD 비율에서는 학습 과정의 불안정성이 심화되는 것을 확인했습니다. 이러한 문제를 해결하기 위해 연구진은 CrossQ 프레임워크에 가중치 정규화(weight normalization) 기법을 통합했습니다. 가중치 정규화는 학습 과정을 안정화시키고, 가소성(plasticity) 손실을 방지하며, 효과적인 학습률을 일정하게 유지하는 효과가 있습니다.
본 연구에서 제시된 접근법은 증가하는 UTD 비율에도 안정적으로 확장 가능하며, DeepMind Control Suite와 Myosuite 벤치마크의 25개 복잡한 연속 제어 작업(특히, 복잡한 개와 인간형 환경)에서 경쟁력 있는 성능을 달성했습니다. 네트워크 재설정과 같은 과감한 조치 없이도 샘플 효율과 확장성을 향상시키는 간단하면서도 강력한 방법을 제시한 것입니다. 이는 모델 기반이 아닌 강화학습에서 샘플 효율 향상 및 확장성 문제 해결에 중요한 돌파구가 될 것으로 기대됩니다.
핵심 내용 요약:
- 문제: 기존 CrossQ의 높은 UTD 비율 환경에서의 확장성 문제
- 해결책: 가중치 정규화 기법 통합
- 결과: DeepMind Control Suite와 Myosuite 벤치마크에서 경쟁력 있는 성능 달성, 네트워크 재설정 없이 샘플 효율 및 확장성 향상
- 의의: 모델 기반이 아닌 강화학습의 샘플 효율 및 확장성 향상에 기여하는 중요한 연구 결과
Reference
[arxiv] Scaling Off-Policy Reinforcement Learning with Batch and Weight Normalization
Published: (Updated: )
Author: Daniel Palenicek, Florian Vogt, Jan Peters
http://arxiv.org/abs/2502.07523v1