Daily Arxiv

전 세계에서 발간되는 인공지능 관련 논문을 정리하는 페이지 입니다.
본 페이지는 Google Gemini를 활용해 요약 정리하며, 비영리로 운영 됩니다.
논문에 대한 저작권은 저자 및 해당 기관에 있으며, 공유 시 출처만 명기하면 됩니다.

Retrieval-Augmented Decision Transformer: External Memory for In-context RL

Created by
  • Haebom

저자

Thomas Schmied, Fabian Paischer, Vihang Patil, Markus Hofmarcher, Razvan Pascanu, Sepp Hochreiter

개요

본 논문은 강화학습(RL) 환경에서의 문맥 내 학습(ICL)의 한계를 극복하기 위해, 과거 경험에서 현재 상황에 관련된 부분 경로만을 검색하는 메모리 메커니즘을 사용하는 Retrieval-Augmented Decision Transformer (RA-DT)를 제안합니다. RA-DT는 훈련이 필요없는 도메인 비의존적인 검색 구성요소를 사용하며, 그리드 월드 환경, 로봇 시뮬레이션, 절차적으로 생성된 비디오 게임에서 기존 방법보다 우수한 성능을 보입니다. 특히, 짧은 문맥 길이만으로도 높은 성능을 달성합니다. 본 논문은 복잡한 환경에서 기존 ICL 방법의 한계를 밝히고, 향후 연구 방향을 제시하며, 사용된 네 가지 환경에 대한 데이터셋을 공개합니다.

시사점, 한계점

시사점:
강화학습에서 문맥 내 학습의 효율성을 높이는 새로운 방법(RA-DT) 제시.
복잡한 환경에서의 긴 에피소드 문제를 효과적으로 해결.
도메인 비의존적인 검색 메커니즘을 통해 다양한 환경에 적용 가능성 증대.
관련 데이터셋 공개를 통한 향후 연구 활성화 기여.
한계점:
제안된 방법의 성능 평가가 특정 환경(그리드 월드, 로봇 시뮬레이션, 비디오 게임)에 국한됨.
더욱 복잡하고 다양한 환경에서의 일반화 성능 검증 필요.
검색 메커니즘의 효율성 및 확장성에 대한 추가적인 분석 필요.
👍