SeterusnyaPanduan seterusnya
Perhatian Pertanyaan Berkumpulan
AI bahasa
PANDUAN AI Bahasa
Perhatian Berbilang Pertanyaan (MQA) ialah sentuhan penjimatan memori pada perhatian pengubah yang berkongsi satu set kunci dan nilai merentasi semua kepala perhatian.
Dengar panduan iniAI-diceritakan · ~2 min
Ia mempercepatkan penjanaan teks secara mendadak dengan mengecilkan memori yang mesti dikocok oleh model.
Perhatian berbilang kepala standard memberikan setiap kepala pertanyaan, kunci dan unjuran nilainya sendiri. Semasa penjanaan, kunci dan nilai untuk semua token lalu mesti dicache dan dimuat semula pada setiap langkah — cache KV ini menjadi kesesakan utama, kerana membacanya dari ingatan adalah lebih perlahan daripada matematik itu sendiri. Perhatian Berbilang Pertanyaan, yang dicadangkan oleh Noam Shazeer pada 2019, menyimpan unjuran pertanyaan yang berasingan bagi setiap kepala tetapi meruntuhkan kunci dan nilai kepada satu kepala dikongsi. Ini mengecilkan cache KV dengan faktor yang sama dengan bilangan kepala, kadangkala 8x hingga 64x lebih kecil. Hasilnya ialah penyahkodan autoregresif yang lebih pantas dan jejak memori yang lebih ringan, dengan hanya penurunan kualiti yang sederhana. Jalan tengah, Perhatian Pertanyaan Berkumpulan, mengimbangi pertukaran.
Mempercepatkan penjanaan token demi token dalam pembantu sembang di mana cache KV, bukan pengiraan mentah, mengehadkan pemprosesan.
PaLM Google, yang menggunakan Perhatian Berbilang Pertanyaan untuk mendayakan inferens berskala besar yang cekap.
Melayan ramai pengguna serentak pada satu GPU dengan mengecilkan memori cache KV setiap permintaan.
Perhatian Pertanyaan Berkumpulan dalam Llama 2 70B dan Llama 3, keturunan langsung yang mengimbangi kelajuan MQA dengan kualiti perhatian penuh.
MQA menetapkan bahawa anda boleh memangkas kepala kunci/nilai berlebihan dengan sedikit bahaya, dan cerapan itu kini membentuk hampir setiap LLM inferens pantas. Medan ini sebahagian besarnya telah tertumpu pada Perhatian Pertanyaan Berkumpulan (GQA), digunakan dalam Llama 2/3 dan banyak lagi, yang menggunakan beberapa kumpulan KV dan bukannya satu untuk memulihkan kualiti sambil mengekalkan kebanyakan kelajuan. Kerja masa depan menggabungkan idea ini dengan pemampatan cache KV, pengkuantitian dan perhatian berbilang pendam untuk mendorong konteks yang lebih panjang dan penyajian yang lebih murah.
Newsletter percuma
Tiga cerita AI yang disahkan setiap pagi hari bekerja, ditulis dalam Bahasa Inggeris biasa. Percuma selamanya, tiada iklan.
Satu e-mel setiap hari bekerja. Berhenti melanggan dalam satu klik. Kami tidak pernah menjual atau berkongsi alamat anda.
Uji diri anda
Maklum balas segera pada setiap jawapan, dan sijil yang boleh dikongsi dengan ID yang boleh disahkan sebaik sahaja anda lulus kursus.
Sokong pendidikan AI percuma. AI Understanding ialah pertubuhan bukan keuntungan 501(c)(3) — tiada iklan, tiada bayaran dinding, selamanya. Buat derma
Perhatian Berbilang Pertanyaan (MQA) ialah sentuhan penjimatan memori pada perhatian pengubah yang berkongsi satu set kunci dan nilai merentasi semua kepala perhatian. Ia mempercepatkan penjanaan teks secara mendadak dengan mengecilkan memori yang mesti dikocok oleh model.
MQA menyimpan pertanyaan berasingan bagi setiap kepala tetapi berkongsi satu unjuran utama dan satu unjuran nilai merentas semua kepala.
Memuat semula cache KV yang besar setiap langkah adalah terikat lebar jalur; MQA mengecilkan cache itu untuk mempercepatkan penyahkodan.
Memandangkan kunci dan nilai runtuh daripada H kepala kepada satu, cache mengecut kira-kira kiraan kepala.
Berkongsi kunci dan nilai mengurangkan sedikit kapasiti perwakilan, menyebabkan penurunan kualiti kecil sebagai pertukaran untuk keuntungan kelajuan besar.
Perhatian Pertanyaan Berkumpulan (GQA) menggunakan beberapa kumpulan KV dan bukannya satu, mengimbangi kualiti dan kelajuan.
Teruskan belajar
Lebih banyak panduan dipilih untuk topik ini
SeterusnyaPanduan seterusnya
Perhatian Pertanyaan Berkumpulan
AI bahasa