개발된 상상 임무 기반 학습 기법의 유효성. 그림 UNIST 제공새로운 AI 메타 강화학습 기법이 등장했다. AI 스스로 새로운 임무를 상상하고 예습한다.
UNIST(울산과학기술원) 인공지능대학원 한승열 교수팀은 학습해본 적 없는 새로운 임무에도 적응할 수 있도록 인공지능을 훈련시키는 기법인 TAVT(Task-Aware Virtual Training)를 개발했다고 19일 밝혔다.
이번에 개발된 학습 기법은 AI 스스로 '가상의 임무'를 만들어 이를 미리 학습하게 하는 방식이다.
딥러닝 기반의 표현 학습 모듈과 생성 모듈로 구성돼 있다.
표현 학습 모듈이 서로 다른 임무 간의 유사도(거리)를 정량화해 임무 구조(latent representation)를 파악하면, 생성 모듈이 이를 조합해 새로운 가상의 임무를 만든다.
생성된 가상 임무는 원래 임무의 특징을 보존할 수 있도록 설계돼, 학습한 적 없는 상황에 대한 예습 효과가 있다.
한승열 교수(좌측)와 김정모 연구원(제1저자). 사진 UNIST 제공제1저자인 김정모 연구원은 "기존 강화학습은 하나의 임무에서 최적의 정책을 익히도록 설계돼 새로운 상황에서는 성능이 급격히 떨어진다. 다양한 임무를 경험하게 하는 메타강화학습 기법도 있지만, 훈련 범위를 벗어난 상황(Out-Of-Distribution)에서는 여전히 적응이 쉽지 않다"고 설명했다.
연구팀은 이 학습 기법을 치타, 개미, 2족 보행 등 다양한 로봇 시뮬레이션 환경에 적용해본 결과, 훈련하지 않은 임무에 대한 적응력이 향상됐음을 확인했다.
특히 치타 로봇 시뮬레이션(Cheetah-Vel-OOD) 실험에서는 TAVT 기법을 적용한 경우, 경험해 보지 못한 중간 속도(1.25, 1.75 m/s 등)에서도 목표 속도를 빠르게 파악해 안정적으로 주행을 이어갔다.
반면 기존 메타강화학습 기법을 적용한 로봇은 적응이 느리거나 넘어지는 일이 잦았다.
한승열 교수는 "이번 기법은 AI 에이전트의 임무 일반화 성능을 높일 수 있는 방식으로, 피지컬 AI 로봇이나 자율주행차·드론 등 다양한 상황에서 유연한 대응이 필수적인 분야에 폭넓게 활용될 수 있을 것"이라고 말했다.
연구 결과는 세계 3대 인공지능 학회 중 하나인 '2025 ICML(International Conference on Machine Learning)'에 채택됐다.
2025 ICML은 7월 13일부터 19일까지 캐나다 밴쿠버에서 열렸다.