Настройка нейронной сети с помощью keras python

Построение нейронной сети является актуальной темой в самых разных задачах - от классификации продуктов на категории до распознавания лиц на видео. Однако, для получения качественного результата необходимо грамотно настроить ее параметры.

Рассмотрим процесс построения нейронной сети и регулирования ее гиперпараметров с помощью keras python.

Анализ будет проведен на примере открытого датасета из kaggle “Otto group product classification challenge” (Otto Group Product Classification Challenge | Kaggle). Количество строк в данном датасете составляет примерно 62 тысячи. Каждая строка соответствует одному продукту. Необходимо классифицировать продукты компании по 9-ти категориям, основываясь на 93 характеристиках. Каждая категория – это тип продукта, например, мода, электроника. Классы несбалансированы, что можно увидеть на графике.

Описательные статистики для переменных приведены на графике.

Шаг за шагом будут разобраны логические блоки для построения модели: входные данные, слои, количество нейронов, функция активации, процедура обучения.

Импортируем необходимые библиотеки:

На вход нейронной сети подаются тензоры. Тензоры – это общая форма векторов в виде n-мерной матрицы. Если данные табличные (как в данном случае), то тензор будет двумерный, где каждый столбец этой матрицы, это одно наблюдение из таблицы (одна строка табличных данных).

Общая структура сети разрабатывается с помощь объекта модели keras Sequential(), который создает последовательную модель с пошаговым добавлениям слоев в нее. Dense слой является самым необходимым и базовым. Он отвечает за соединение нейронов из предыдущего и следующего слоя. Например, если первый слой имеет 5 нейронов, а второй 3, то общее количество соединений между слоями будет 15. Dense слой отвечает за эти соединения, и у него есть настраиваемые гиперпараметры: количество нейронов, тип активации, инициализация типа ядра. Dropout Layer помогает избавиться от переобучения модели. Таким образом, некоторые нейроны становятся равными 0, и это сокращает вычисления в процессе обучения. Базовая модель представлена ниже:

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

Для данной задачи использовались функции активации ReLu и Softmax. Функция ReLu выглядит следующим образом:

Функция Softmax выглядит следующим образом:

Функция потерь – это показатель, который помогает понять, движется ли сеть в правильном направлении. Так как задача решается для многоклассовой классификации, то использовалась функция потерь sparse categorical crossentropy:

Посмотрим на качество модели с параметрами, заданными ранее. На графике изображена функция потерь и точность нашего классификатора. Видно, что наилучший результат наблюдается около 13-ой эпохи, однако наблюдается переобучение модели из-за большой флуктуации функции потерь.

Наиболее важной частью обучения модели является оптимизатор. Ранее был упомянут термин обратного распространения ошибки – это как раз и есть процесс оптимизации. Структура, которая ранее была описана, инициализируется случайными весами между нейронами слоев. Процесс обучения нужен для того, чтобы обновить эти случайные веса для получения высокой предсказательной способности. Сеть берет одну обучающую выборку и использует ее значения в качестве входных данных слоя, далее происходит активация этих данных, и на вход следующему слою подаются уже новые взвешенные данные после активации. Аналогичный процесс происходит и на последующих слоях. Результат последнего слоя будет прогнозом для обучающей выборки. Далее применяется функция потерь, которая показывает насколько сильно ошибается модель. Чтобы эту ошибку уменьшить, применяется функция оптимизатора, которая использует производные для ответа на вопрос: насколько сильно изменится функция потерь при небольшом изменении весов между нейронами.

Вычисление одной обучающей выборки из входного слоя называется проходом. Обычно обучение происходит батчами из-за ограничений системы. Батч – это набор обучающих выборок из входных данных. Сеть обновляет свои веса после обработки всех выборок из батча. Это называется итерацией (то есть успешная обработка всех обучающих выборок из батча с последующим обновлением весов в сети). А вычисление всех обучающих выборок, которые были во входных данных, с периодическими обновлениями весов, называется эпохой. На каждой итерации сеть использует функцию оптимизации, чтобы внести изменения в веса между нейронами. Шаг за шагом, с помощью нескольких итераций, а затем нескольких эпох, сеть обновляет свои веса и учится делать корректный прогноз. Существует много разных функций оптимизации. В нашем случае была использована Adagrad.

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

На графике видно, что проблемы переобучения больше нет, так как функция потерь стала более гладкой.

Увеличение количества эпох.

Заметно, что качество улучшилось.

Увеличие количества нейронов.

Качество модели стало еще лучше.

Был взят небольшой объем данных, чтобы показать прогноз модели. Из результатов видно, что ошибка модели невелика.

Как было ранее показано, самыми часто встречающимися классами являются номер 2 и 6. Поэтому чаще всего модель не ошибается именно в этих классах.

Таким образом, в данной статье были рассмотрены строительные блоки для нейронной сети на keras. Модель была избавлена от переобучения путем увеличения dropout, после чего более высокие результаты были достигнуты за счет увеличения количества нейронов и эпох. Основываясь на результатах обучения было получено, что общая точность предсказания модели по всем классам продуктов составляет чуть больше 0,81, что характеризует ее качество. Таким образом, данную модель можно использовать для решения задачи многоклассовой классификации. Однако структура является универсальной, и ее можно использовать при решении задач с применением нейронных сетей.