Написать пост

Учёные из Google усовершенствовали ИИ для мобильных устройств

Аватар Артем Гаврилов

Нейросеть MnasNet работает эффективнее известных на данный момент конкурентов и подбирает архитектуру индивидуально для каждого мобильного устройства.

Специалисты Google описали работу системы MnasNet на основе автоматизированной нейронной сети в публикации. Нейросеть использует способ обучения с подкреплением для выбора подходящей мобильному устройству архитектуры.

Структура

Система MnasNet содержит:

  • объект управления на базе рекуррентной нейросети для обучения и подбора архитектуры модели;
  • тренера, выстраивающего модели и обучающий их точности;
  • машину вывода, которая вычисляет скорость модели на телефонах, используя TensorFlow.

Команда использовала многокритериальную оптимизацию с целью добиться высокой скорости и точности. Дополнительно учёные задействовали алгоритм обучения с подкреплением и функцией вознаграждения. Таким образом, MnasNet находит для каждой платформы оптимальность по Парето. Для каждого мобильного устройства создаются индивидуальные особенности, требующие определённой архитектуры.

Учёные из Google усовершенствовали ИИ для мобильных устройств 1

Для достижения оптимального баланса между гибкостью поиска и областью допустимых решений, применили иерархический подход. Он представляет свёрточную нейронную сеть в виде ряда блоков. Затем использует последовательный поиск для назначения слоёв архитектуры каждому блоку. Благодаря этому каждый слой использует различные операции и связи. При этом слои блоков принудительно копируются.

Учёные из Google усовершенствовали ИИ для мобильных устройств 2

Тестирование

Учёные включили скорость поиска архитектуры в функцию вознаграждения поискового алгоритма. По заявлению разработчиков, система подбирает модель в 1,5 раза быстрее MobileNetV2 и в 2,4 раза быстрее NASNet, при этом учитывает оптимальные показатели точности и скорости. В качестве доказательства результат продемонстрировали на примере работы с базой изображений ImageNet.

Учёные из Google усовершенствовали ИИ для мобильных устройств 3
Следите за новыми постами
Следите за новыми постами по любимым темам
774 открытий774 показов