SıradakiSonraki rehber
Dikkat Açma ve Kafa Budama
Teknik
Teknik KILAVUZ
FlashAttention is an IO-aware way to compute exact attention without materializing the full attention matrix in high-bandwidth memory.
Bu kılavuzu dinleyinAI tarafından anlatılan · ~2 min
It uses tiling and online softmax to reduce memory traffic; speed and memory gains depend on sequence length, hardware and implementation.
Standard attention compares every token to every other token, producing an N-by-N score matrix that grows quadratically with sequence length. Naively, that matrix is written to and read back from GPU high-bandwidth memory (HBM), and that shuttling — not the multiplications — is the real bottleneck. Flash Attention, introduced by Tri Dao and colleagues in 2022, reorganizes the computation so the matrix is never fully stored. It processes queries, keys, and values in small tiles that fit in fast on-chip SRAM, computes partial results, and stitches them together using an online running-softmax trick. The algorithm computes the same attention operation while reducing high-bandwidth-memory traffic and intermediate storage. It can be faster, especially on long sequences, but measured gains depend on GPU, sequence length, dtype and implementation.
Daha düşük bellek maliyetiyle daha uzun bağlam pencereleriyle Llama ve GPT sınıfı sistemler gibi büyük dil modellerinin eğitimi.
Uzun bir istemin ilk kez okunduğu ön doldurma aşamasını hızlandırarak sohbet asistanlarına daha hızlı hizmet vermek.
Tek bir GPU'da uzun sıralı dikkati mümkün kılarak kitapların veya kod tabanlarının tamamını alan belge analiz araçlarının etkinleştirilmesi.
Yüksek çözünürlüklü girişlerin çok uzun simge dizileri oluşturduğu görüntü ve ses Transformatörlerine güç verilmesi.
FlashAttention-2 and FlashAttention-3 research explores improved work partitioning, asynchronous execution, and hardware-specific low-precision paths. Results depend on the GPU, sequence length, dtype and implementation. These algorithms can reduce memory traffic for dense attention, but they do not remove the underlying query-key pair computations. Benchmark the exact training or serving workload before assuming a speedup.
Ücretsiz bülten
Hafta içi her sabah sade İngilizce yazılmış, doğrulanmış üç yapay zeka hikayesi. Sonsuza kadar ücretsiz, reklam yok.
Hafta içi her gün bir e-posta. Tek tıklamayla aboneliğinizi iptal edin. Adresinizi asla satmayız veya paylaşmayız.
Kendinizi test edin
Her yanıtla ilgili anında geri bildirim ve bir kursu geçtiğinizde doğrulanabilir bir kimliğe sahip paylaşılabilir bir sertifika.
Ücretsiz AI eğitimini destekleyin. AI Understanding, bir 501(c)(3) kâr amacı gütmeyen kuruluştur; hiçbir zaman reklam yoktur, ödeme duvarı yoktur. Bağış yap
FlashAttention is an IO-aware way to compute exact attention without materializing the full attention matrix in high-bandwidth memory. It uses tiling and online softmax to reduce memory traffic; speed and memory gains depend on sequence length, hardware and implementation.
Flash Attention, IO-farkındadır: hızlı çip üzerindeki SRAM ile yavaş, yüksek bant genişlikli bellek arasında gidip gelen verileri azaltır; bu, aritmetiğin kendisinden ziyade gerçek darboğazdır.
Hesaplamayı, her bloğun hızlı SRAM'a sığacağı şekilde döşer, matrisin tamamını HBM'de gerçekleştirmeden kısmi çıktıları hesaplar ve biriktirir.
Çevrimiçi softmax, döşemeleri aktarırken çalışan bir maksimumu ve çalışan paydayı korur, daha önceki kısmi çıktıları yeniden ölçeklendirerek son normalleştirmenin tam olmasını sağlar.
Saf dikkat matrisi bellekte N-kare olarak ölçeklenir, bu nedenle tam depolamadan kaçınmak, tam olarak diziler uzun olduğunda en büyük tasarrufu sağlar.
GÇ bilinçli, algoritmanın, aritmetik işlemler yerine HBM trafiğini en aza indirerek, bellek hiyerarşisinde veri taşıma maliyetine göre tasarlandığı anlamına gelir.
Öğrenmeye devam et
Bu konu için daha fazla rehber seçildi
SıradakiSonraki rehber
Dikkat Açma ve Kafa Budama
Teknik