초장문 컨텍스트를 위한 새로운 희소 어텐션(sparse attention) 방식
New sparse attention method for ultra-long context
핵심 요약
Qwen 27B 모델에서 정확도 손실 없이 10M 토큰까지 컨텍스트를 확장하는 새로운 희소 어텐션 연구가 공개되었습니다.
- 희소 어텐션 연구 — Qwen 27B 모델에서 정확도 저하 없이 어텐션 연산 효율화함
- 초장문 컨텍스트 — 10M 토큰까지 확장 가능하며 NIAH 등 특정 작업에서 성능 유지함
- VRAM 및 성능 — 기존 방식 대비 VRAM 사용량을 줄이고 프리필 속도를 획기적으로 개선함
- 오픈 소스 공개 — 커널, SGLang 통합, 논문 및 데이터셋 구축 스크립트 모두 공개됨
x.com
원문 사이트로 이동
어텐션의 새로운 접근 방식에 관한 오픈 소스 연구를 진행하는 팀원입니다. 이번 연구에서 밝혀낸 핵심 내용은 다음과 같습니다.
• Qwen 3.8 27B의 어텐션을 희소화(sparsify)해도 정확도 손실이 거의 없음
• 컨텍스트를 10M 토큰까지 확장하면 특정 작업에서는 정확도가 떨어지지만, NIAH 같은 대부분의 작업은 여전히 잘 돌아감 (꽤 흥미로운 결과임)
• 풀 어텐션과 동일한 검증 손실을 기록하면서도 훨씬 빠르게, 오직 다음 토큰 예측 손실만으로 희소 어텐션을 네이티브하게 학습시킬 수 있음
커뮤니티에서 좋아해 줬으면 좋겠네요. 저희는 H200/B300 GPU와 대규모 시스템에 집중하고 있지만, 이 연구 결과는 작은 오픈 웨이트 모델에도 적용 가능해서 로컬 하드웨어에서 더 긴 컨텍스트를 저렴하게 돌릴 수 있게 해줍니다.
커널, SGLang 통합, 논문, 세부 사항까지 전부 다 공개했습니다.
흥미롭고 유용하게 봐주길 바랍니다!
(과장 광고는 아님 - 최대한 투명하게 공개하려고 노력했음)
