본문으로 건너뛰기

별도의 격리된 메모리 버퍼를 가진 Transformer 구조 제안

Transformer 백본을 고정한 채 별도의 메모리 가중치만 학습시켜 20개의 독립적인 사실을 높은 정확도로 저장하고 인출하는 BDH 기반 메커니즘이 공개됐다.

커뮤니티 반응

작성자가 직접 실험 결과와 코드를 공유했으며, 기존의 Fast Weights 및 Test-Time Training(TTT) 연구들과의 연관성에 대해 기술적인 흥미를 보이고 있습니다.

주요 논점

01찬성다수

별도의 메모리 버퍼를 통한 정보 저장이 효율적이며 높은 정확도를 보인다는 점을 실험 데이터로 입증했다.

합의점 vs 논쟁점

합의점

  • 제안된 메커니즘이 기존의 Fast Weight Programmer 이론과 최신 TTT 연구의 흐름에 부합한다.
  • 단일 GPU에서 재현 가능한 수준의 효율성을 갖추고 있다.

논쟁점

  • 20개 이상의 사실을 저장할 때의 용량 한계나 대규모 모델로의 확장성은 아직 검증되지 않았다.

실용적 조언

  • 제공된 GitHub 저장소의 README를 참고하여 단일 GPU 환경에서 직접 메모리 인코딩 실험을 재현해 볼 수 있다.
  • 모델 전체를 파인튜닝하는 대신 특정 정보를 메모리 가중치에만 주입하는 방식으로 활용 가능하다.

섹션별 상세

Transformer 백본을 동결한 상태에서 메모리 가중치에 대해서만 300회의 경사 단계를 수행하여 정보를 저장했다. 웜뱃의 배설물 모양이나 특정 지역의 날씨와 같은 구체적인 사실 문장을 쿼리했을 때 0.999 이상의 높은 확률로 정확한 답변을 출력했다. 프로세스를 종료하고 다시 로드한 후에도 동일한 결과가 유지됨을 확인하여 메모리의 지속성을 입증했다.
20개의 서로 연관되지 않은 사실들을 공동으로 인코딩했을 때 20개 모두 정확하게 인출되었으며 확률값의 중앙값은 0.997로 나타났다. 두 가지 주제를 동시에 인코딩하더라도 상호 간섭이나 오염이 0.03 미만으로 매우 낮게 유지됐다. 이는 제안된 메모리 구조가 정보를 독립적이고 안정적으로 저장할 수 있음을 시사한다.
작동 원리는 BDH 메커니즘을 기반으로 하며 매 토큰 단계마다 활성화 값의 외적을 계산하여 메모리에 누적한다. 단순한 토큰 식별자가 아니라 학습된 내용 주소 지정 투영을 사용하여 주소가 전체 인과적 문맥을 반영하도록 설계했다. 이러한 방식은 1991년 Schmidhuber가 제안한 빠른 가중치 프로그래머의 전통을 계승하며 최신 TTT 연구들과 궤를 같이한다.
실험은 1500만 개의 파라미터와 2억 5천만 개의 토큰을 사용하여 단일 소비자용 GPU에서 수행됐다. 정보 인코딩은 한 번에 이루어지는 원샷 방식이 아니라 300단계의 학습 과정을 거쳐 완성된다. 현재 20개 이상의 사실에 대한 용량 확장은 테스트되지 않았으며 오픈소스 라이선스로 코드가 공개되어 재현이 가능하다.

용어 해설

빠른 가중치(Fast Weights)
신경망의 표준 가중치와 달리 추론 과정에서 동적으로 업데이트되는 가중치이다. 입력 시퀀스의 정보를 일시적으로 저장하고 활용하는 메모리 역할을 수행하여 모델의 적응력을 높인다.
외적(Outer Product)
두 벡터를 곱하여 행렬을 생성하는 연산이다. 이 아티클에서는 토큰 단계마다 활성화 값들의 외적을 계산하여 메모리에 정보를 누적하고 저장하는 메커니즘의 핵심으로 사용된다.
내용 주소 지정(Content-Addressing)
데이터의 물리적 위치가 아닌 데이터 자체의 내용을 기반으로 정보를 검색하는 방식이다. 학습된 투영(projection)을 통해 전체 인과적 문맥을 반영하는 주소를 생성하여 메모리에서 정확한 정보를 인출한다.
경사 단계(Gradient Steps)
모델의 오차를 줄이기 위해 가중치를 업데이트하는 반복적인 최적화 과정이다. 본 실험에서는 전체 모델이 아닌 메모리 가중치에 대해서만 300회의 업데이트를 수행하여 정보를 인코딩했다.

언급된 도구

bdh-fast-weights추천링크

Transformer를 위한 별도의 메모리 버퍼 및 빠른 가중치 구현체

AI 분석 전체 내용 보기

AI 요약 · 북마크 · 개인 피드 설정 — 무료

출처 · 인용 안내

원문 발행 2026. 04. 13.수집 2026. 04. 13.출처 타입 REDDIT

인용 시 "요약 출처: AI Trends (aitrends.kr)"를 표기하고, 사실 확인은 원문 보기 기준으로 진행해 주세요. 자세한 기준은 운영 정책을 참고해 주세요.