라즈베리 파이 제로에서 AI가 혼자 꿈꾸고 게임하는 기술

3줄 요약
1. 라즈베리 파이 피코 2 W에서 구동되는 초소형 월드 모델이 개발되었습니다.
2. AI 신경망 두 개가 협력하여 스스로 게임 화면을 만들고 직접 플레이합니다.
3. 메모리를 극도로 아끼는 루프드 트랜스포머 구조로 마이크로컨트롤러에 이식했습니다.

단돈 몇천 원짜리 칩에서 AI가 게임을 만든다고?

최근 개발자 커뮤니티에서 초소형 마이크로컨트롤러 보드인 '라즈베리 파이 피코 2 W'를 활용한 아주 독특한 프로젝트가 주목받고 있습니다. 거대 서버실이나 고성능 그래픽카드에서나 돌아갈 법한 인공지능(AI) '월드 모델'을 이 아주 작고 저렴한 칩 위에 우겨넣은 것입니다. 심지어 이 AI는 단순히 코드를 실행하는 것을 넘어, 스스로 화면을 그려내고 그 안에서 플레이까지 해냅니다.

이 프로젝트의 핵심은 '세상을 이해하고 예측하는 모델'을 극단적으로 다이어트하는 데 있습니다. 연구자는 메모리가 턱없이 부족한 마이크로컨트롤러 환경에서도 AI가 작동할 수 있도록 가중치를 재사용하는 방식을 고안했습니다. 그 결과, 전력과 자원이 극도로 제한된 환경에서도 환상적인 실험이 가능해졌습니다.

서로 다른 역할을 하는 두 개의 초소형 신경망

이 시스템은 크게 두 가지의 미니 신경망이 짝을 이뤄 작동합니다. 첫 번째 신경망은 '게임 세계' 자체를 창조합니다. 현재의 프레임과 사용자의 조작(액션) 입력을 받아서, 다음에 펼쳐질 화면과 보상 값을 예측해 냅니다. 즉, AI가 실시간으로 게임 엔진 역할을 수행하며 다음 장면을 그려내는 셈입니다.

두 번째 신경망은 그렇게 만들어진 가상의 게임 세계를 즐기는 '플레이어' 역할을 맡습니다. 첫 번째 모델이 그려낸 화면을 보고 다음에 어떤 행동을 취할지 결정합니다. 결과적으로 이 작은 보드는 외부의 도움 없이 자기 스스로 가상의 세계를 상상하고, 그 상상 속 세상에서 캐릭터를 조종하며 학습을 이어 나갑니다.

메모리를 극한으로 아끼는 '루프드 트랜스포머'의 비밀

일반적인 AI 모델은 수많은 층(layer)을 차례대로 쌓아 올리기 때문에 엄청난 메모리를 잡아먹습니다. 하지만 이 프로젝트는 '루프드 트랜스포머(Looped Transformer)' 구조를 채택했습니다. 하나의 트랜스포머 블록을 여러 번 반복해서 적용하되, 똑같은 가중치(weights)를 계속 재사용하는 방식입니다.

가중치를 재사용하면 메모리 사용량이 극적으로 줄어듭니다. 개발자는 FPU(부동소수점 연산장치)를 탑재하고 가격이 저렴한 라즈베리 파이 피코 2 W를 타깃으로 잡았습니다. 여기에 아주 단순화된 미니 게임들을 직접 넘파이(NumPy)로 구현해 가며 AI가 만들어내는 프레임이 실제 규칙과 맞는지 꼼꼼하게 검증했습니다.

스스로 꿈꾸고 학습하는 장치의 미래

여기서 끝이 아닙니다. 개발자는 이 보드에 스트리밍 강화학습(RL) 에이전트까지 추가했습니다. 이제 라즈베리 파이 피코는 누군가 시키지 않아도 자신이 만든 가상의 꿈속 세계 안에서 플레이 방법을 스스로 깨우치고 발전시킬 수 있습니다.

물론 이 기술이 당장 화려한 3차원 대작 게임을 완벽하게 구동해 주는 것은 아닙니다. 하지만 저전력·저비용의 극단적인 임베디드 환경에서도 AI의 핵심 기능인 '환경 예측과 학습'이 가능함을 입증했다는 점에서 큰 의미가 있습니다. 앞으로 IoT 기기나 스마트 센서가 스스로 판단하고 학습하는 길을 열어준 셈입니다.