ДалееСледующее руководство
Использование ИИ для решения физических задач
Приложения
Техническое РУКОВОДСТВО
Прямой оценщик (STE) — это простой прием для обучения сетей, которые содержат сложные, недифференцируемые шаги, такие как округление или определение порога.
Прослушайте это руководствоОзвучено ИИ · ~2 min
Он использует дискретное значение при прямом проходе, но при вычислении градиентов делает вид, что операция была идентичной.
Некоторые операции, такие как округление до целого числа, бинаризация весов до +1/-1 или выбор верхней категории с помощью argmax, имеют производную, которая почти везде равна нулю и не определена на переходах. Этот нулевой градиент перестает учиться холодному. Прямой оценщик обходит эту проблему, разделяя прямой и обратный проходы: вперед он применяет настоящую сложную операцию; наоборот, он просто копирует входящий градиент, как если бы операция была идентификатором (или плавным прокси). Оценка смещена, поскольку истинный градиент на самом деле равен нулю, однако на практике это приближение «представь, что все гладко» очень хорошо обучает бинаризованные и квантованные сети, поэтому STE является рабочей лошадкой эффективного глубокого обучения.
Обучение двоичных и низкобитовых квантованных нейронных сетей для эффективного вывода на телефонах и периферийных устройствах.
Обратное распространение через поиск в дискретной кодовой книге в VQ-VAE и нейронных токенизаторах аудио/изображений.
Обучение с учетом квантования, при котором веса или активации округляются до фиксированной точки во время прямого прохода.
Обучение жесткому вниманию или дискретному стробированию, когда на пути вычислений находится argmax или порог.
STE лежит в основе роста количества низкоразрядных и бинарных нейронных сетей, используемых для искусственного интеллекта на устройствах и с ограниченным энергопотреблением, и он играет центральную роль в обучении моделей векторного квантования, подобных тем, которые используются в современных токенизаторах изображений и аудио. Текущая работа направлена на получение более точных и менее смещенных оценок градиента и лучшего теоретического понимания того, почему такое грубое приближение работает. Поскольку спрос на крошечные, быстрые, квантованные модели на телефонах и периферийном оборудовании растет, можно ожидать, что трюки в стиле STE останутся основополагающими, несмотря на их известную предвзятость.
Бесплатная рассылка
Три проверенных AI-истории каждое утро буднего дня, написанные простым английским. Бесплатно навсегда, без рекламы.
Одно электронное письмо каждый будний день. Отписаться в один клик. Мы никогда не продаем и не передаем ваш адрес.
Проверьте себя
Мгновенная обратная связь по каждому ответу и общий сертификат с проверяемым идентификатором после прохождения курса.
Поддержите бесплатное образование в области искусственного интеллекта. 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.
Продолжайте учиться
Другие руководства, выбранные по этой теме
ДалееСледующее руководство
Использование ИИ для решения физических задач
Приложения