SelanjutnyaPanduan berikutnya
Paralelisme Tensor untuk Model Besar
Teknis
PANDUAN Teknis
Jika suatu model terlalu besar untuk ditampung pada satu GPU, paralelisme model dan pipeline akan membagi model itu sendiri ke seluruh perangkat.
Dengarkan panduan iniDiceritakan oleh AI · ~2 min
Inilah yang memungkinkan pelatihan model bahasa raksasa dengan ratusan miliar parameter secara fisik.
Paralelisme model mempartisi satu model ke beberapa GPU sehingga tidak ada satu perangkat pun yang perlu memikul semua beban tersebut. Ada dua rasa utama. Paralelisme tensor (intra-layer) membagi matematika di dalam sebuah lapisan, seperti memotong perkalian matriks besar di seluruh GPU yang masing-masing menghitung bagian dari output. Paralelisme pipeline (antar-lapisan) menetapkan lapisan berurutan yang berbeda ke GPU yang berbeda, sehingga blok lapisan 1 berada di GPU 0, blok 2 di GPU 1, dan seterusnya, dengan aktivasi diteruskan seperti jalur perakitan. Tantangan dalam pipeline naif adalah 'gelembung': ketika GPU 0 bekerja pada batch pertama, GPU hilir tidak digunakan. Pipelining membagi setiap batch menjadi batch mikro sehingga semua tahapan tetap sibuk, sehingga meningkatkan pemanfaatan secara signifikan.
Melatih model bergaya GPT dengan NVIDIA Megatron-LM, yang membagi perhatian setiap lapisan transformator dan matriks feed-forward di seluruh GPU melalui paralelisme tensor.
Menggunakan GPipe untuk menempatkan berbagai lapisan visi raksasa atau model bahasa pada akselerator terpisah sementara batching mikro membuat mereka sibuk.
Mesin pipeline DeepSpeed mempartisi model multi-ratus miliar parameter menjadi beberapa tahap di banyak node.
Menggabungkan paralelisme tensor di dalam satu server 8-GPU dengan paralelisme pipeline yang mencakup beberapa server untuk melatih model yang terlalu besar untuk satu mesin.
Kerangka kerja semakin mengotomatiskan masalah sulit dalam memutuskan cara mempartisi model di seluruh perangkat, menggunakan pembuatan profil dan pencarian untuk menyeimbangkan komputasi dan komunikasi. Harapkan integrasi yang lebih erat antara tensor, pipeline, dan paralelisme data (paralelisme 3D), penjadwalan micro-batch yang lebih cerdas untuk hampir menghilangkan gelembung pipeline, dan perangkat keras dengan interkoneksi yang lebih cepat sehingga pemisahan satu lapisan di seluruh chip menjadi lebih murah dan rutin untuk model yang semakin besar.
Buletin gratis
Tiga cerita AI terverifikasi setiap pagi hari kerja, ditulis dalam bahasa Inggris sederhana. Gratis selamanya, tanpa iklan.
Satu email setiap hari kerja. Berhenti berlangganan dalam satu klik. Kami tidak pernah menjual atau membagikan alamat Anda.
Uji diri Anda sendiri
Umpan balik instan untuk setiap jawaban, dan sertifikat yang dapat dibagikan dengan ID yang dapat diverifikasi setelah Anda lulus kursus.
Mendukung pendidikan AI gratis. AI Understanding adalah organisasi nirlaba 501(c)(3) — tanpa iklan, tanpa paywall, selamanya. Berikan sumbangan
Jika suatu model terlalu besar untuk ditampung pada satu GPU, paralelisme model dan pipeline akan membagi model itu sendiri ke seluruh perangkat. Inilah yang memungkinkan pelatihan model bahasa raksasa dengan ratusan miliar parameter secara fisik.
Paralelisme model dan pipeline mempartisi model itu sendiri di seluruh perangkat, memungkinkan pelatihan jaringan yang jauh lebih besar daripada yang dapat ditampung oleh GPU mana pun.
Paralelisme tensor membagi komputasi dalam satu lapisan, misalnya memisahkan matriks berbobot besar sehingga setiap GPU menghitung sebagian dari output.
Di awal langkah pipeline, tahapan hilir belum memiliki masukan dan tidak digunakan, sehingga membuang-buang waktu GPU; celah menganggur ini disebut gelembung.
Membagi batch menjadi batch mikro memungkinkan beberapa batch mikro menempati tahapan yang berbeda secara bersamaan, membuat semua GPU tetap sibuk dan mengurangi waktu idle.
Paralelisme tensor sering berkomunikasi untuk menggabungkan kembali hasil matriks parsial, sehingga ditempatkan di tempat bandwidth interkoneksi tertinggi, di dalam node melalui NVLink.
Teruslah belajar
Panduan lainnya dipilih untuk topik ini
SelanjutnyaPanduan berikutnya
Paralelisme Tensor untuk Model Besar
Teknis