LISA: Linear-Indexed Sparse Attention for Efficient Long-Context Reasoning
- 1.재사전학습 없이 교체 가능한 플러그인 희소 어텐션 모듈 LISA 제안
- 2.O(n) 선형 어텐션과 상위 M개 토큰을 고르는 Lightning Indexer를 게이팅으로 융합, 복잡도 O(nM)
- 3.교사 모델 어텐션 분포를 증류하는 2단계 학습과 헤드별 KL 손실로 Indexer 정렬
- 4.DeepSeek 증류 Qwen에서 16K 컴텍스트 추론 50% 가속, AIME·MATH-500 평균 +5.6%
왜 중요한가?
긴 사고사슬(CoT) 추론 모델의 O(n²) 어텐션 비용이 프로덕션 배포의 병목인 상황에서, 속도 50% 개선과 벤치마크 성능 향상을 동시에 달성한 점이 핵심이다. 처음부터 재학습이 필요 없는 플러그인 방식이라 기존 추론 모델에 바로 적용할 수 있다.
🏷️ 언급 프로젝트
본문 미리보기
arXiv:2607.19358v1 Announce Type: new Abstract: Recent advances in long chain-of-thought reasoning models such as DeepSeek-R1 have led to increasingly longer inference context lengths under the test-time scaling paradigm. However, the O(n^2) computational complexity of standard self-attention causes inference costs to grow sharply with long sequences, limiting the deployment of long-CoT reasoning in production settings. To address this, we propose LISA (Linear-Indexed Sparse Attention), a plug-
전체 내용이 궁금하다면?
원문을 직접 읽어보세요