SeterusnyaPanduan seterusnya
Perhatian Linear RWKV
AI bahasa
PANDUAN Teknikal
Perhatian linear menggantikan perhatian softmax kuadratik dalam Transformers dengan helah matematik yang berskala linear dengan panjang jujukan.
Dengar panduan iniAI-diceritakan · ~2 min
Performer is a landmark method that approximates softmax using random feature kernels, making very long sequences computationally affordable.
Perhatian Transformer Standard mengira skor antara setiap pasangan token, kos masa dan ingatan yang berkembang dengan kuasa dua panjang jujukan (O(n^2)). Perhatian linear menulis semula pengiraan supaya kos berkembang hanya secara linear (O(n)). Idea utama: perhatian softmax ialah softmax(QK^T)V, tetapi jika anda menggantikan softmax dengan peta ciri kernel phi, anda mendapat phi(Q)(phi(K)^T V). Oleh kerana pendaraban matriks adalah bersekutu, anda mengira phi(K)^T V dahulu (matriks d-demi-d yang kecil), mengelakkan matriks skor n-demi-n gergasi sepenuhnya. Pelaku, daripada Google pada tahun 2020, menjadikan ini anggaran yang tepat bagi softmax sebenar menggunakan FAVOR+ (Perhatian Pantas Melalui ciri Rawak Ortogonal positif), melukis unjuran rawak yang memastikan anggaran kernel tidak berat sebelah dan stabil.
Memproses urutan genomik atau protein yang panjang di mana perhatian kuadratik penuh akan meletihkan memori GPU
Ringkasan peringkat dokumen ke atas laporan yang sangat panjang tanpa potongan, menggunakan tulang belakang gaya Performer
Audio bentuk panjang atau pemodelan siri masa yang cekap dengan jujukan menjangkau puluhan ribu langkah
Mengurangkan kos inferens dalam model sembang konteks panjang dengan menggantikan beberapa lapisan softmax dengan varian perhatian linear
Perhatian linear tulen sering menjejaki softmax pada kualiti, jadi medan itu menumpu pada hibrid: model ruang keadaan (Mamba), perhatian linear berpagar dan seni bina yang menggabungkan beberapa lapisan perhatian penuh dengan banyak lapisan linear. Apabila tetingkap konteks mendorong ke arah berjuta-juta token, mekanisme linear dan sub-kuadrat semakin menarik untuk kos, dan perhatian linear gaya berulang sedang dikaji semula untuk inferens penstriman yang cekap dan model pada peranti.
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 linear menggantikan perhatian softmax kuadratik dalam Transformers dengan helah matematik yang berskala linear dengan panjang jujukan. Performer ialah kaedah mercu tanda yang menghampiri softmax menggunakan kernel ciri rawak, menjadikan jujukan yang sangat panjang mampu milik secara pengiraan.
Perhatian Softmax membandingkan setiap pasangan token, menghasilkan matriks skor n-by-n, jadi kos meningkat sebagai O(n^2).
Oleh kerana pendaraban matriks adalah bersekutu, anda boleh mengira phi(K)^T V dahulu, matriks d-demi-d yang kecil, bukannya phi(Q)phi(K)^T.
FAVOR+ menggunakan ciri rawak ortogonal positif untuk menganggarkan kernel softmax eksponen tanpa membentuk matriks perhatian penuh.
Ciri positif mengekalkan anggaran kernel bukan negatif, mengelakkan ketidakstabilan dan nilai negatif yang melanda peta ciri sin/cos terdahulu.
Dengan menyusun semula pengiraan dan tidak sekali-kali membina matriks n-demi-n, skala kos secara linear dengan panjang jujukan.
Teruskan belajar
Lebih banyak panduan dipilih untuk topik ini
SeterusnyaPanduan seterusnya
Perhatian Linear RWKV
AI bahasa