끊임없이 배우는 AI: DRAGO 알고리즘이 가져올 미래
Sun, Fu, Littman, 그리고 Konidaris가 개발한 DRAGO 알고리즘은 지속적인 모델 기반 강화 학습에서 지식 유지를 위한 혁신적인 해결책을 제시합니다. Synthetic Experience Rehearsal과 Regaining Memories Through Exploration이라는 두 가지 핵심 구성 요소를 통해 과거 지식을 효과적으로 유지하고 새로운 과제에 적응하는 능력을 향상시킵니다. 실험 결과, DRAGO는 다양한 지속적 학습 시나리오에서 우수한 성능을 보이며 AI의 지속적인 학습 능력 발전에 기여할 것으로 기대됩니다.

인공지능(AI) 분야에서 지속적인 학습(Continual Learning)은 뜨거운 감자입니다. 단일 과제만 학습하는 것이 아니라, 여러 과제를 순차적으로 학습하며 이전 지식을 유지하는 능력이 AI의 발전에 필수적이기 때문입니다. 최근, Sun, Fu, Littman, 그리고 Konidaris가 이끄는 연구팀이 DRAGO라는 혁신적인 알고리즘을 통해 이 문제에 도전장을 던졌습니다.
DRAGO는 모델 기반 강화 학습(Model-Based Reinforcement Learning)에 초점을 맞추고 있습니다. 단순히 경험을 통해 학습하는 대신, AI가 환경을 모델링하고 그 모델을 기반으로 학습을 진행하는 방식입니다. 하지만 기존 모델 기반 강화 학습은 새로운 과제를 학습할 때 이전 과제에 대한 지식을 잊어버리는 '파국적 망각(Catastrophic Forgetting)' 문제에 직면했습니다.
DRAGO는 이 문제를 해결하기 위해 두 가지 핵심 구성 요소를 제시합니다. 첫째, Synthetic Experience Rehearsal은 과거 과제에서 얻은 지식을 합성 경험으로 재구성하여 AI가 이전에 학습한 내용을 강화하도록 돕습니다. 이는 기존의 데이터 저장 방식과 달리, 메모리 공간을 효율적으로 사용하며 지속적인 학습을 가능하게 합니다. 마치 과거의 기억을 되살리는 연습과 같습니다.
둘째, Regaining Memories Through Exploration은 내재적 보상 메커니즘을 통해 AI가 이전 과제에서 중요했던 상태들을 다시 방문하도록 유도합니다. 이는 마치 사람이 중요한 경험을 되짚어보며 복습하는 것과 같습니다. 이를 통해 AI는 과거의 지식을 더욱 확실하게 기억하고, 새로운 과제에 대한 학습에도 도움을 받습니다.
연구팀은 다양한 실험을 통해 DRAGO가 여러 과제에 걸쳐 지식을 효과적으로 유지하며 우수한 성능을 보임을 입증했습니다. 이는 지속적인 학습 분야에 큰 혁신을 가져올 가능성을 시사합니다. DRAGO는 단순히 알고리즘 이상의 의미를 지닙니다. 끊임없이 배우고 발전하는, 인간과 같은 유연성을 가진 AI 시대를 앞당길 핵심 기술로서 주목받을 것입니다. 앞으로 DRAGO가 어떻게 발전하고, 우리의 삶에 어떤 영향을 미칠지 기대됩니다.
Reference
[arxiv] Knowledge Retention for Continual Model-Based Reinforcement Learning
Published: (Updated: )
Author: Yixiang Sun, Haotian Fu, Michael Littman, George Konidaris
http://arxiv.org/abs/2503.04256v1