Основные понятия
본 논문에서는 단계별 보상 없이 궤적-반환 쌍으로부터 학습하여 장기적인 계획을 수행하는 새로운 생성 모델인 잠재 계획 변환기(LPT)를 제안합니다. LPT는 잠재 변수를 활용하여 궤적 생성과 최종 반환을 연결하고, 잠재 공간에서의 추론을 통해 일관성 있는 계획을 생성합니다.
Аннотация
잠재 계획 변환기(LPT): 잠재 공간 추론으로서의 계획
본 연구 논문에서는 단계별 보상 없이 궤적-반환 쌍 데이터를 사용하여 장기적인 계획 문제를 해결하는 새로운 접근 방식을 제시합니다. 저자들은 궤적 생성과 최종 반환을 연결하는 잠재 변수를 활용하는 잠재 계획 변환기(LPT)라는 새로운 모델을 소개합니다. LPT는 궤적-반환 쌍에 대한 최대 가능성 추정을 통해 학습되며, 잠재 변수의 사후 샘플링을 통해 제한된 컨텍스트 내에서도 일관된 추상화를 형성하기 위해 하위 궤적을 자연스럽게 통합합니다. 테스트 시에는 정책 실행 전에 예상되는 반환으로부터 잠재 변수를 추론하여 추론으로서의 계획이라는 아이디어를 실현합니다.
1. 문제 제기
기존의 강화 학습 방법은 에이전트가 환경과 상호 작용하면서 얻는 단계별 보상에 크게 의존합니다. 그러나 실제 환경에서는 단계별 보상을 설계하는 것이 어렵거나 불가능한 경우가 많습니다. 이러한 문제를 해결하기 위해 본 논문에서는 단계별 보상 없이 궤적-반환 쌍 데이터만을 사용하여 에이전트를 학습하는 방법을 제안합니다.
2. 잠재 계획 변환기(LPT)
LPT는 잠재 변수 z를 사용하여 궤적 τ와 반환 y의 결합 분포를 모델링하는 생성 모델입니다.
잠재 변수: 잠재 변수 z는 궤적 τ를 잠재 공간에 나타낸 벡터로, 계획을 의미합니다.
궤적 생성기: 궤적 생성기 pβ(τ|z)는 잠재 변수 z가 주어졌을 때 궤적 τ의 조건부 확률 분포를 나타냅니다. 본 논문에서는 인과 관계 트랜스포머를 사용하여 궤적 생성기를 모델링합니다.
반환 예측기: 반환 예측기 pγ(y|z)는 잠재 변수 z가 주어졌을 때 반환 y의 조건부 확률 분포를 나타냅니다. 본 논문에서는 다층 퍼셉트론(MLP)을 사용하여 반환 예측기를 모델링합니다.
3. 학습 및 추론
LPT는 궤적-반환 쌍 데이터에 대한 최대 가능성 추정(MLE)을 통해 학습됩니다. 학습 과정에서 잠재 변수 z의 사후 분포 pθ(z0|τ, y)를 샘플링하기 위해 Langevin dynamics를 사용합니다. 테스트 시에는 주어진 예상 반환 y에 대해 베이즈 규칙을 사용하여 잠재 변수 z0를 추론합니다.
4. 실험 결과
저자들은 Gym-Mujoco, Franka Kitchen, Maze2D, Connect Four 등 다양한 벤치마크에서 LPT를 평가했습니다. 실험 결과, LPT는 단계별 보상 없이도 경쟁력 있는 성능을 달성했으며, 특히 궤적 스티칭 및 환경 우발 상황에 대한 적응 능력이 뛰어났습니다.
본 논문의 주요 기여는 다음과 같습니다.
단계별 보상 없이 궤적-반환 쌍 데이터를 사용하여 장기적인 계획 문제를 해결하는 새로운 접근 방식 제시
잠재 변수를 활용하여 궤적 생성과 최종 반환을 연결하는 LPT 모델 제안
다양한 벤치마크에서 LPT의 성능을 검증하고, 기존 방법 대비 우수한 성능 확인