위키득계정 ↗
기술

Titans 장기 기억 모델

Titans 장기 기억 모델기술
연구 공개
2024년 12월
주요 구성
어텐션과 신경 장기 기억
기억 관점
단기 기억과 장기 기억
제안한 결합 변형
세 가지
Titans는 어텐션과 신경망 기반 장기 기억을 결합하는 모델 구조 계열이다. 현재 문맥의 직접적인 관계와 과거의 정보를 기억하는 기능을 함께 활용하려는 접근이다.

두 방식의 기억

순환 모델은 정보를 고정 크기의 상태로 압축하고 어텐션은 문맥 창 안의 직접적인 의존 관계를 다룬다. Titans 연구는 두 방식의 성질을 기억의 관점에서 비교했다.

과거 문맥을 학습하기

신경 장기 기억 모듈이 과거 문맥을 기억하도록 학습한다. 이를 통해 어텐션이 현재 문맥에 집중하면서 오래전 정보도 활용하도록 설계했다.

결합 구조의 변형

어텐션을 제한된 범위에서 정밀하게 동작하는 단기 기억으로 보고 신경 기억을 더 지속적인 장기 기억으로 해석한다. 논문은 두 모듈의 결합 방법에 대한 세 변형을 제시했다.

다양한 과제의 평가

연구는 언어 모델링과 상식 추론 및 유전체와 시계열 과제로 평가했다. 긴 문맥에서 특정 정보를 찾는 시험도 수행하며 결합 구조의 효과를 비교했다.

관련 문서

xLSTM / 유도 헤드 / LongBench / RULER / 트랜스포머 / 언어 모델