선형화의 핵심: 분석 기반 트랜스포머 선형화
요약
본 연구는 인과적 셀프 어텐션의 2차 비용 문제를 해결하기 위해 트랜스포머 선형화 기법을 제안합니다. 모델 품질 보존에 초점을 맞춰, 상태 업데이트 설계와 구조적 개입(싱크 토큰 등)을 분석했습니다. LLaMA 및 Qwen 같은 대규모 모델까지 확장하여 MMLU에서 우수한 성능과 긴 컨텍스트 검색 능력을 입증했습니다.
핵심 포인트
- 인과적 셀프 어텐션의 2차 비용 문제를 해결하는 연구입니다.
- 델타 스타일 네트워크가 게이팅 누적보다 우수함을 이론적으로 설명합니다.
- 싱크 토큰, 짧은 컨볼루션 등 구조적 개입으로 오차를 줄였습니다.
- LLaMA와 Qwen 모델에 적용하여 높은 성능을 입증했습니다.
인과적 셀프 어텐션(causal self-attention)의 2차 비용은 긴 컨텍스트 트랜스포머 추론을 심각하게 병목시킵니다. 수많은 사후(post hoc) 선형화 파이프라인이 존재하지만, 어떤 구성 요소가 모델 품질을 보존하는지 식별하기 어렵습니다. 본 연구는 엄격한 고정 백본(frozen-backbone) 체제에서 상태 업데이트 설계의 영향을 분리하여 분석합니다. 우리는 softmax가 키 의존적(key-dependent), 랭크-1 직교 투영(rank-1 orthogonal projections)에 의존한다는 것을 보여주며, 이로 인해 왜 델타 스타일 네트워크(delta-style networks)가 순수 게이팅 누적(purely gated accumulation)보다 우수한 성능을 보이는지 설명합니다. 우리는 근사 오차의 잠재적인 원인을 식별하고, 싱크 토큰(sink tokens), 짧은 컨볼루션(short convolutions), 고정 예산 캐시 라우팅(fixed-budget cache routing)과 같은 구조적 개입을 도입하여 남아있는 격차를 줄입니다. 우리는 이 선형화 접근 방식을 LLaMA와 Qwen 모델의 최대 32B 파라미터까지 확장했으며, MMLU에서 이전 사후 기준선(post hoc baselines)을 능가하고 복잡한 적응형 캐싱 프레임워크의 긴 컨텍스트 검색 성능과 일치함을 보여줍니다.
AI 자동 생성 콘텐츠
본 콘텐츠는 arXiv cs.LG의 원문을 AI가 자동으로 요약·번역·분석한 것입니다. 원 저작권은 원저작자에게 있으며, 정확한 내용은 반드시 원문을 확인해 주세요.
원문 바로가기