Files
2nd/10_Wiki/Topics/AI_and_ML/Flash Attention.md
T

37 lines
3.1 KiB
Markdown

---
id: [[P-Reinforce|P-Reinforce]]-AUTO-FLAT-001
category: Unified
confidence_score: 1.00
tags: [auto-reinforced, flash-attention, attention-optimization, transformer, gpu-optimization, llm-inference]
last_reinforced: 2026-05-04
---
# [[Flash Attention|Flash Attention]]
## 📌 한 줄 통찰 (The Karpathy Summary)
> "메모리 병목의 해방군: 어텐션의 수학적 원리는 유지하면서, GPU의 SRAM과 HBM 사이의 데이터 이동을 타일링 기법으로 최적화하여 2~4배의 속도 향상과 극적인 메모리 절감을 동시에 달성한 하드웨어 인식 최적화의 정점."
## 📖 구조화된 지식 (Synthesized Content)
FlashAttention은 대규모 언어 모델(LLM)의 컨텍스트 윈도우 확장 시 발생하는 어텐션 메커니즘의 계산 및 메모리 병목 현상을 해결하기 위한 하드웨어 인식(Hardware-aware) 최적화 기법입니다.
1. **핵심 작동 원리**:
* **Tiling (타일링)**: 거대한 어텐션 행렬을 작은 블록(타일) 단위로 나누어, 속도가 빠른 GPU 온칩 SRAM에서 연산을 수행함으로써 느린 HBM(고대역폭 메모리)으로의 접근 횟수를 최소화합니다.
* **Recomputation (재계산)**: 메모리에 거대한 중간 행렬을 저장하는 대신, 역전파(Backpropagation) 시 필요한 값을 필요할 때마다 다시 계산하는 방식을 택해 메모리 복잡도를 $O(n^2)$에서 $O(n)$으로 낮춥니다.
2. **주요 성과**:
* **정확도 유지**: 근본적인 연산 복잡도($O(n^2d)$)는 동일하게 유지하면서도, 실제 연산 속도를 2~4배 향상시킵니다.
* **컨텍스트 확장**: 메모리 효율성을 극대화하여 기존에는 불가능했던 수십만 토큰 이상의 긴 문맥 처리를 가능하게 합니다.
3. **버전 진화**:
* **FlashAttention-2**: 연산 순서 최적화와 작업 분할(Work Partitioning)을 통해 병렬성을 더욱 높여, FP16 기준 이론적 최대 성능의 70% 이상을 달성했습니다.
## ⚖️ Trade-offs & Caveats
* **연산량 자체의 한계**: 메모리 대역폭 문제는 해결하지만, 시퀀스 길이에 따른 연산량 증가($O(n^2)$) 자체를 선형으로 바꾸는 것은 아닙니다. 따라서 백만 토큰 이상의 초장기 시퀀스에서는 여전히 상당한 계산 비용이 발생합니다.
* **분산 처리 시의 상충**: Ring Attention과 같은 컨텍스트 병렬성 기술과 결합할 때, 세분화된 FlashAttention 처리가 통신 오버헤드로 인해 효율성 저하(Efficiency Penalties)를 초래할 수 있습니다. 이를 해결하기 위해 USP(Unified Sequence Parallelism)와 같은 하이브리드 접근법이 필요합니다.
## 🔗 지식 연결 (Graph)
* **상위 개념**: [[Attention Mechanisms|Attention Mechanisms]], [[LLM Inference Optimization|LLM Inference Optimization]]
* **하위/연관 기술**: [[KV Cache|KV Cache]], [[Ring Attention|Ring Attention]], [[Sparse Attention|Sparse Attention]], [[PagedAttention|PagedAttention]]
* **프로젝트 적용**: 초대형 컨텍스트 지원 RAG 엔진, 에이전트 자율 분석 루프
---
*Last updated: 2026-05-04*