ДаліНаступний посібник
Using AI to Work Through Physics Problems
Додатки
Технічний КЕРІВНИЦТВО
Наскрізний оцінювач (STE) — це простий трюк для навчання мереж, які містять жорсткі, недиференційовані кроки, такі як округлення або порогове значення.
Послухайте цей посібникШІ-розказаний · ~2 min
Він використовує дискретне значення на прямому проході, але прикидається, що операція була ідентичністю під час обчислення градієнтів.
Деякі операції, такі як округлення до цілого числа, двійкове переведення ваг до +1/-1 або вибір найвищої категорії за допомогою argmax, мають похідну, яка дорівнює нулю майже скрізь і не визначена на стрибках. Цей нульовий градієнт зупиняє вивчення холоду. Наскрізний оцінювач обходить це, роз’єднуючи проходи вперед і назад: вперед, він застосовує справжню жорстку операцію; назад, він просто копіює вхідний градієнт наскрізь, ніби операція була ідентифікатором (або плавним проксі). Оцінка є упередженою, оскільки справжній градієнт насправді дорівнює нулю, але на практиці ця апроксимація «вдавайте, що це гладко» надзвичайно добре тренує двійкові та квантовані мережі, тому STE є робочою конячкою ефективного глибокого навчання.
Навчання двійкових і низькорозрядних квантованих нейронних мереж для ефективного висновку на телефонах і периферійних пристроях.
Зворотне розповсюдження через пошук дискретної кодової книги у VQ-VAE та нейронні токенізатори аудіо/зображення.
Навчання з урахуванням квантування, де ваги або активації округлюються до фіксованої коми під час проходу вперед.
Навчання посиленого уваги або дискретне стробування, коли на шляху обчислення є аргмакс або поріг.
STE лежить в основі сплеску низькорозрядних і двійкових нейронних мереж, які прагнуть для штучного інтелекту з обмеженим енергоспоживанням на пристрої, і є центральним для навчання моделей векторного квантування, подібних до тих, що використовуються в сучасних токенізаторах зображень і аудіо. Поточна робота шукає точніші, менш упереджені оцінки градієнта та краще теоретичне розуміння того, чому таке грубе наближення працює. Оскільки попит на крихітні, швидкі, квантовані моделі зростає на телефонах і периферійному обладнанні, очікуйте, що трюки в стилі STE залишаться основоположними, незважаючи на їх відому упередженість.
Безплатний розсилок
Три перевірені історії про ШІ щоробочого ранку, написані простою мовою. Безплатно назавжди, без реклами.
Один лист щотижня з понеділка по п'ятницю. Відписатися в один клік. Ми ніколи не продаємо та не передаємо вашу адресу.
Перевір себе
Миттєва зворотна связь на кожну відповідь та сертифікат із перевіреним ідентифікатором, яким можна поділитися, після проходження курсу.
Підтримайте безплатну освіту у сфері ШІ. AI Understanding — це некомерційна організація 501(c)(3) — без реклами, без платних стін, назавжди. Зробіть пожертву
Наскрізний оцінювач (STE) — це простий трюк для навчання мереж, які містять жорсткі, недиференційовані кроки, такі як округлення або порогове значення. Він використовує дискретне значення на прямому проході, але прикидається, що операція була ідентичністю під час обчислення градієнтів.
The guide describes a hard forward pass and a surrogate backward pass.
The guide says hard rounding has a zero or undefined derivative that can stop learning.
The guide explicitly notes that the approximation is biased.
The guide names quantization and rounding as common applications.
The guide gives this pattern to preserve the hard value while passing the identity gradient.
Продовжуйте вчитися
Інші посібники, вибрані для цієї теми
ДаліНаступний посібник
Using AI to Work Through Physics Problems
Додатки