슬라이딩 윈도가 선형 어텐션을 이겼다
재학습 없이, 더 싸게, 더 정확하게 — 마이크로소프트가 던진 반문 하나가 '긴 문맥' 경쟁의 전제를 흔든다.
- 거대 언어모델의 2차(quadratic) 어텐션은 토큰이 늘수록 메모리·비용이 감당 못 할 만큼 커진다. 그 대안으로 각광받은 것이 선형 어텐션(Linear Attention)이다.
- 그러나 이 논문은 싱크(sink)를 더한 슬라이딩 윈도 어텐션(SWA)이라는 훨씬 단순한 방법이 사후학습된 선형 어텐션과 대등하거나 오히려 더 낫다는 것을 보였다.
- 특히 장문맥 추론에서 SWA는 선형 어텐션보다 2–10배 높은 성능을 내며, 재학습이 필요 없고 빠르고 메모리도 적게 쓴다.
1. 문제의 뿌리 — 왜 어텐션은 '지속 불가능'한가
트랜스포머의 심장인 셀프 어텐션은 모든 토큰이 이전의 모든 토큰을 바라본다. 문장이 길어질수록 계산량은 문장 길이의 제곱(n²)으로 불어난다. 더 치명적인 것은 KV 캐시(Key-Value cache)이다. 새 토큰이 하나 생길 때마다 그 키와 값을 메모리에 영구히 쌓아 두어야 하므로, 문맥이 길어질수록 메모리와 에너지 소모가 끝없이 늘어난다. 논문은 이를 두고 "지속 불가능하다(unsustainable)"라고 단언한다.
2. 유력한 대안이었던 '선형 어텐션'의 약속과 함정
이 문제를 풀기 위한 대표적 접근이 선형 어텐션이다. 아이디어는 단순하고 매력적이다. 과거의 모든 키·값을 하나하나 저장하는 대신, 이를 고정 크기의 '상태(state)' 하나로 압축해 두고 새 토큰이 올 때마다 갱신한다. 그러면 문맥이 아무리 길어져도 메모리는 일정하고 비용은 길이에 비례(선형)한다. 기존 모델을 이 방식으로 개조(retrofit)해 사후학습시키는 연구가 크게 유행하였다.
그런데 논문은 날카로운 지적을 던진다. "이 방향은 정작 더 단순한 기준선(baseline)과 제대로 비교된 적이 없다." 화려한 방법론끼리만 겨루었을 뿐, 정말 소박한 대안보다 나은지 검증이 부실했다는 것이다.
3. 반전의 주인공 — '싱크를 둔 슬라이딩 윈도 어텐션(SWA)'
슬라이딩 윈도 어텐션은 이름 그대로다. 모든 과거를 보는 대신 최근 W개의 토큰(창, window)만 바라본다. 창은 새 토큰을 따라 미끄러지듯 이동한다. 덕분에 메모리는 창 크기로 고정되고 계산도 가볍다. 다만 순수한 슬라이딩 윈도만으로는 문장 맨 앞의 중요한 정보(지시문·질문 등)를 잃는 약점이 있다.
여기서 등장하는 것이 싱크(sink)다. 맨 앞 소수의 토큰을 항상 창에 함께 유지하는 장치로, 모델이 습관적으로 크게 주목하는 '초기 토큰'을 버리지 않게 해 안정성을 지킨다. 즉 SWA + 싱크 = 최근 문맥(창) + 최초 문맥(싱크)의 조합이다.
4. 실험 결과 — 장문맥에서 2–10배의 격차
저자들은 여러 종류의 LLM과 다양한 다운스트림 과제에서 두 방식을 맞붙였다. 결과는 명확하였다. SWA는 사후학습된 선형 어텐션과 대등하거나 더 나은 성능을 냈고, 특히 긴 문맥에서 정보를 정확히 찾아내야 하는 추론 과제 — Needle-in-a-Haystack(건초더미 속 바늘 찾기)과 BABILong — 에서는 선형 어텐션보다 2배에서 10배까지 높은 성능을 기록하였다.
5. 세 방식 한눈에 비교
| 구분 | 완전(2차) 어텐션 | 선형 어텐션 | SWA + 싱크 |
|---|---|---|---|
| 계산 비용 | n² (높음) | 선형(낮음) | 선형(낮음) |
| 메모리(KV) | 무한 누적 | 고정 | 고정(창+싱크) |
| 재학습 필요 | 없음 | 사후학습 필요 | 불필요 |
| 장문맥 추론 | 최상(기준) | 약함 | 강함(선형의 2–10배) |
| 구현 난도 | 기본 | 복잡 | 단순 |
6. 실무적 함의 — 무엇을 바꿔야 하는가
저자들의 권고는 직설적이다. "추론 메모리 비용을 줄이려면 선형 모델을 사후학습하지 말고 SWA로 전환하라." 선형 어텐션이 SWA를 따라잡으려면 처음부터 새로 학습하거나 막대한 사후학습을 감내해야 할 가능성이 크다는 것이다. 정리하면 다음과 같다.
첫째, 비용 구조가 달라진다. 이미 학습된 모델에 손대지 않고도 추론 단계에서 메모리와 연산을 크게 줄일 수 있으므로, 서비스 운영비(특히 긴 문맥을 다루는 RAG·에이전트·장문 요약)를 즉시 낮출 여지가 있다. 둘째, 전환 비용이 거의 없다. 재학습이 필요 없다는 점은 곧 실험과 배포의 리스크가 작다는 뜻이다. 셋째, '단순한 기준선'을 먼저 시험하라는 교훈이다. 화려한 신기법을 도입하기 전에 소박한 대안이 이미 충분한지 반드시 확인해야 한다.
※ 본문의 그림은 이해를 돕기 위한 개념도이며, 정확한 실험 수치와 설정은 원문을 확인하기 바란다.
'인공지능시대에 우리 함께 가자' 카테고리의 다른 글
| AI 연구·산업 아침판 다이제스트 (2026년 9월 1일 화요일) (0) | 2026.09.01 |
|---|---|
| 오후 국내 브리핑 (2026년 8월 31일, 월요일) (1) | 2026.08.31 |
| AI 아침판 다이제스트 (2026년 8월 31일 월요일) (0) | 2026.08.31 |
| AI 연구·산업 동향 아침판 다이제스트 (2026년 8월 28일, 금) (0) | 2026.08.29 |
| AI 연구·산업 동향 아침판 다이제스트 (2026년 8월 28일, 금) (0) | 2026.08.28 |