Daily Arxiv

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

From Memories to Maps: Mechanisms of In-Context Reinforcement Learning in Transformers

Created by
  • Haebom
Category
Empty

저자

Ching Fang, Kanaka Rajan

개요

인간과 동물은 최소한의 경험으로 새로운 환경에 적응하는 뛰어난 학습 효율성을 보인다. 기존의 강화 학습 알고리즘은 이러한 능력을 잘 포착하지 못한다. 본 연구에서는 변압기를 사용하여 쥐의 행동을 모방한 계획 작업 환경에서 문맥 내 강화 학습을 수행하고, 모델에서 나타나는 학습 알고리즘을 분석한다. 그 결과, 표현 학습은 문맥 내 구조 학습과 상호 연관성 정렬을 통해 지원되며, 강화 학습 전략은 표준 모델 프리 또는 모델 기반 계획으로 해석되지 않음을 확인했다. 대신, 문맥 내 강화 학습은 모델의 메모리 토큰 내에서 중간 계산을 캐싱하여 지원되며, 의사 결정 시 이러한 캐싱된 정보를 활용한다.

시사점, 한계점

시사점:
메모리가 유연한 행동을 지원하기 위해 원시 경험과 캐시된 계산을 저장하는 계산 자원으로 작용할 수 있음을 시사한다.
모델에서 개발된 표현이 뇌의 해마-내외측후피질 시스템과 관련된 계산과 유사하다는 것을 발견하여, 자연 인지 능력과의 연관성을 제시한다.
인공 및 자연 환경에서 문맥 내 학습의 기반이 되는 빠른 적응에 대한 메커니즘 가설을 제시한다.
한계점:
구체적인 한계점은 논문에 명시되지 않음.
👍