Поиск архитектуры нейронной сети

Определение:

Поиск архитектуры нейронной сети (англ. Neural Architecture Search, NAS) — процесс автоматизации проектирования архитектуры, то есть нахождение самой модели машинного обучения. Системе NAS предоставляется набор данных, тип задачи (классификация, регрессия и т.д.), на основании которого система определяет архитектуру нейронной сети. Эта архитектура будет работать лучше всех других архитектур для данной задачи при обучении предоставленным набором данных.

NAS можно рассматривать как часть AutoML (Автоматическое машинное обучение), он имеет значительное совпадение с оптимизацией гиперпараметров (Настройка гиперпараметров). NAS находит архитектуру из всех возможных архитектур, следуя стратегии поиска, которая максимизирует производительность.

Принцип работы

Методы для NAS классифицируются по трем измерениям: пространство поиска (англ. Search Space), стратегия поиска (англ. Search Strategy) и стратегия оценки эффективности (англ. Performance Estimation Strategy). Схематичный принцип работы NAS отображен на рисунке 1.

Рисунок 1 — Обобщающая иллюстрация методов NAS. Стратегия поиска выбирает архитектуру [math]A[/math] из предопределенного пространства поиска [math]A[/math]. Архитектура передается в стратегию оценки производительности, которая возвращает оценку эффективности [math]A[/math] в стратегию поиска. (Источник)

Пространство поиска (англ. Search Space)

Пространство поиска определяет, какую нейронную архитектуру в принципе может обнаружить подход NAS. Это может быть цепочечная архитектура (см. рисунок 2), в которой выход уровня [math](n-1)[/math] подается как вход уровня [math](n)[/math]. Или это может быть современная сложная архитектура с пропущенным соединением (англ. multi-branch network).

• 

  Рисунок 2 — Chain-like и multi-branch сеть; Различные типы слоев визуализируются разными цветами. Ребро от слоя [math]L_i[/math] до слоя [math]L_j[/math] обозначает, что [math]L_j[/math] получает выходные данные [math]L_i[/math] в качестве входных данных.
• 

  Рисунок 3 — Слева: сell architecture, две разные ячейки, например normal cell (вверху) и reduction cell (внизу); Cправа: ячейки помещены во внешнюю структуру ручной работы

Предварительные знания о типичных свойствах архитектур способны уменьшить размер пространства поиска и упростить поиск. Тем не менее, они также могут помешать человеку найти новые архитектурные строительные блоки, которые выходят за рамки современных человеческих знаний.

Наиболее часто используемые типы архитектур для NAS — это entire structures (Источник 4.1.1 стр.6), cell-based structures (Источник 4.1.2 стр.6), progressive structures (Источник 4.1.3 стр.7) и morphism-based structures (Источник 4.1.4 стр.8).

Иногда используют внешнюю архитектуру ручной работы (макроархитектуру) с повторяющимися мотивами или ячейками. В таких случаях внешняя структура является фиксированной, NAS ищет только cell-архитектуры. Этот тип поиска известен как микро-поиск или cell search (см. рисунок 3).

Стратегия поиска (англ. Search Strategy)

Стратегия поиска подробно описывает, как исследовать пространство поиска, которое часто экспоненциально велико или даже неограниченно. Она включает в себя классический компромисс между разведкой и эксплуатацией, поскольку, с одной стороны, желательно найти быстро работающие архитектуры, с другой стороны, следует избегать преждевременного схождения.

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

Сравнение методов стратегий поиска

Лучшие результаты на сегодняшний день показывает NAS с использованием стратегии байесовской оптимизации (согласно исследованию Bayesian Optimization with Neural Architectures for Neural Architecture Search, Colin White, Willie Neiswanger, Yash Savani, рисунок 4).

Рисунок 4 — Слева: результат экспериментов, минимизирующих функцию потерь и количества параметров модели; справа: сравнение основных используемых в NAS алгоритмов

Байесовская оптимизация (англ. Bayes Optimization, BO) использует алгоритм для построения вероятностной модели целевой функции, а затем использует эту модель, чтобы выбрать наиболее перспективные гиперпараметры и оценивает выбранные гиперпараметры на истинной целевой функции. Следовательно, BO может итеративно обновлять вероятностную модель, ведя учет оценок прошлых результатов.

BANANAS (англ. Bayesian optimization with neural architectures for NAS)

Рисунок 5 — Иллюстрация мета-нейронной сети в алгоритме BANANAS

Сложностью применения байесовской оптимизации в NAS является обязательное наличие функции расстояния между слоями нейросети. Чтобы обойти этот момент, был разработан алгоритм BANANAS (Bayesian optimization with neural architectures for NAS) — алгоритм, использующий специальную кодировку path encoding для кодирования входных архитектур и получающий на выходе вероятностные распределения (см. рисунок 5).

Алгоритм BANANAS:

1. Выбираются [math]t_0[/math] случайных архитектур из пространства поиска (Search Space)
2. Итерационно проводится обучение ансамбля мета-нейронный сетей на выбранных архитектурах. Каждая сеть ансамбля является сетью прямой связи с полностью связанными слоями, каждому слою дается случайная инициализация весов и случайный порядок обучающего набора. Используемая функция ошибки - вариация MAPE (англ. Mean Absolute Percentage Error).
   1. Далее формируется набор архитектур-кандидатов посредством случайных изменений лучших архитектур после обучения
   2. Для каждой архитектуры-кандидата определяется значение переданной на вход ITS acquisition function ([independent Thompson sampling])
   3. Для архитектуры-кандидата с минимальным значением acquisition function определяется значение целевой вероятностной функции

Стратегия оценки эффективности (англ. Performance Estimation Strategy)

Целью NAS обычно является поиск архитектуры, обеспечивающей высокую точность прогнозов. Performance Estimation относится к процессу оценки этой производительности. Самый простой вариант — выполнить стандартное обучение и проверку архитектуры данных, но это, к сожалению, вычислительно дорого и ограничивает количество архитектур, которые можно изучить. Поэтому многие недавние исследования направлены на разработку методов, способных снизить стоимость этих оценок производительности.

Способы снижения стоимости Performance Estimation и увеличения скорости:

• Сокращение качества оценки — более высокая скорость достигается сокращением датасета
• Экстраполяция прямой обучения — функция оценки может быть экстраполирована после всего нескольких обучающих итераций
• Наследование / Сетевые морфизмы — параметры модели не ищутся каждый раз, они наследуются по каким-либо правилам
• Модели One-Shot / Распределение веса — обучается только одна модель, далее ее веса/параметры используются остальными моделями

Источники информации

• AutoML: A Survey of the State-of-the-Art Xin He, Kaiyong Zhao, Xiaowen Chu
• Bayesian Optimization with Neural Architectures for Neural Architecture Search, Colin White, Willie Neiswanger, Yash Savani
• Medium Towards Data Science - Neural Architecture Search (NAS) - The Future Of Deep Learning
• Neural Architecture Search: A Survey