Контурный анализ и сверточные нейронные сети: обучение на малых неоднородных выборках

Информация о материале

Опубликовано: 23.11.2018, 18:36

Достижения прошедших пяти лет в области искусственного интеллекта, обработки изображений и распознавания образов основаны на глубоком обучении и сверточных нейронных сетях (СНС) различных архитектур. Архитектуры нейронных сетей характеризуются достаточно высокой универсальностью и переносимостью между различными областями применения. Так архитектура U-net, первоначально разработанная для сегментации медицинских цитологических изображений, сейчас с успехом занимает лидирующие позиции в конкурсах по обработке ДЗЗ.

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

Но как решать задачу, если выборки недостаточно для обучения сети, изображения неравномерно распределены по классам (количество различается в десятки раз), репрезентативность изображений для классов различна (экземпляры одного класса различаются значительно, для второго практически одинаковы), а тестовые изображения значительно различаются с обучающими яркостью, контрастностью, пространственным разрешением, видом подстилающей поверхности?

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

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

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

Тестовая выборка для проверки результата обучения представлена 20 изображениями: B-1 – 6 изображений, для проверки качества обучения по малой разнородной выборке, С-130 – 14 изображений – для самой представительной выборки. Изображения взяты из другого источника, имеют меньшее разрешение (0,3-0,5 м/точку против 0,15 м/точку – для обучающих), отличные значения яркости, схему освещения.  Примеры представлены ниже:

Образцы обучающей выборки	Образцы тестовой выборки

Несмотря на одинаковую схему камуфляжа для объектов обеих выборок, визуальные различия весьма значительны.

Сравнение результатов метода на основе контурного анализа выполнено с ансамблем из 8 СНС, каждая из которых нацелена на определение объекта определенного типа. Архитектура сети является одним из лидеров конкурса по распознаванию кораблей на спутниковых снимках. В параметры сети внесены изменения, необходимые для используемых выборок.

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

Результаты работы сети на обучающей выборке представлены в следующей таблице.

Высокое количество ошибок для классов B-52 и С-5 объясняется их малыми размерами и незначительными отличиями отдельных экземпляров. По всей видимости, темно-серый стандартный камуфляж транспортной и бомбардировочной авиации затрудняет выделение признаков для распознавания самолетов данных типов, поскольку класс С-37 (также очень компактный) распознается успешно за счет белого цвета его экземпляров.

Сравнительные результаты контурного и нейросетевого решений приведены в следующей таблице.

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

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

КодЛикс разрабатывает программное обеспечение обработки изображений, систем технического зрения, искусственный интеллект и нейросети. Мы создаём программное обеспечение любой сложности под ключ и с гарантией.