Искусство аугментации: как улучшить модели компьютерного зрения без сбора новых данных

в 9:03, , рубрики: Albumentations, Augmentor, imgaug, аугментация данных, инструменты аугментации, искуственный интеллект, методы аугментации, обучение нейросетей, примеры аугментации
Искусство аугментации: как улучшить модели компьютерного зрения без сбора новых данных - 1

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

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

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

Влияние аугментации данных на точность моделей

Улучшение ключевых метрик точности

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

Пример аугментации. Источник

Пример аугментации. Источник

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

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

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

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

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

Пример различных аугментаций на одном изображении. Источник

Пример различных аугментаций на одном изображении. Источник

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

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

Примеры из реальных исследований

Чтобы не быть голословными, обратимся к последним работам по теме аугментации. 

Так, в работе Донга и его коллег «3D-VirtFusion: Синтетическая 3D аугментация данных с помощью генеративных моделей диффузии и управляемого редактирования» предлагается продвинутый метод автоматической генерации синтетических 3D-сцен с использованием моделей диффузии и контролируемого редактирования. Добавление этих синтетических данных к реальному датасету ScanNet позволило улучшить качество семантической сегментации 3D-сцен на впечатляющие 2.7% mIoU. Особенно полезной аугментация оказалась в случаях ограниченного размера исходного датасета. 

Результаты семантической сегментации на валидном наборе ScanNet v2. Источник

Результаты семантической сегментации на валидном наборе ScanNet v2. Источник

С другой стороны, простые базовые методы аугментации тоже не стоит недооценивать. В исследовании Мюллера «TrivialAugment: Аугментация данных без настройки, но на уровне современных методов» предлагается простой, но эффективный метод аугментации изображений TrivialAugment (TA). Несмотря на свою простоту, TA показал результаты, сопоставимые или даже превосходящие многие более сложные методы аугментации на нескольких популярных датасетах, таких как CIFAR-10, CIFAR-100, SVHN, ImageNet. При этом TA требует минимальных вычислительных затрат. Словом, иногда простые методы аугментации могут быть вполне конкурентоспособны на практике.

Интересный пример использования генеративных моделей для аугментации приводится в работе Лингенберга и его коллег «DIAGen: Семантически разнообразная аугментация изображений с помощью генеративных моделей для малообъемного обучения». Авторы предлагают метод DIAGen, который вносит шум в векторные представления классов, использует большие языковые модели для генерации разнообразных текстовых описаний изображений и применяет взвешивание сгенерированных примеров. Это позволяет добиться значительного разнообразия синтетических данных не только с точки зрения текстуры, но и более высокоуровневых аспектов, таких как ракурс, окружение, погодные условия. 

Качественное сравнение синтетических изображений, полученных с помощью базовой модели DA-Fusion (слева) и модели DIAGen (справа). Каждый ряд был создан на основе одного и того же направляющего изображения 

Качественное сравнение синтетических изображений, полученных с помощью базовой модели DA-Fusion (слева) и модели DIAGen (справа). Каждый ряд был создан на основе одного и того же направляющего изображения 

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

Основные методы аугментации

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

Геометрические преобразования

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

Применение различных геометрических преобразований к рентгеновскому изображению. Источник

Применение различных геометрических преобразований к рентгеновскому изображению. Источник

К базовым геометрическим трансформациям относятся повороты и отражения. Повороты выполняются на случайный угол из заданного диапазона, например, от -20° до +20°, что позволяет моделировать различные ракурсы объектов. Отражения применяются по горизонтали и/или вертикали, но нужно учитывать специфику данных: в некоторых задачах вертикальные отражения могут давать нереалистичные изображения.

Другая группа геометрических преобразований — это сдвиги и масштабирование. Сдвиги выполняются на случайное число пикселей по горизонтали и/или вертикали, меняя положение объектов в кадре. При масштабировании изображение увеличивается или уменьшается на определенный коэффициент, после чего приводится к исходному размеру путем обрезки или дополнения пикселями. Это имитирует изменения размера и расстояния до объектов.

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

Цветовые преобразования

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

Один из самых популярных методов — случайное изменение яркости, контраста и насыщенности (color jittering). Значения пикселей в каждом канале умножаются на случайный коэффициент, близкий к 1, что эмулирует эффекты изменения освещенности и настроек камеры.

Пример применения color jittering. Источник

Пример применения color jittering. Источник

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

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

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

Шумовая аугментация

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

Чаще всего применяются аддитивные шумы с разными распределениями — Гауссовым, Пуассона, Лапласа и т.д. Они моделируются добавлением к значениям пикселей случайных величин с заданным распределением. Дисперсия шума обычно подбирается экспериментально для конкретной задачи.

Пример изображений с рандомным количеством шума. Источник

Пример изображений с рандомным количеством шума. Источник

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

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

Размытие и фокусировка изображений

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

Пример использования размытия. Источник

Пример использования размытия. Источник

Размытие изображения имитирует эффект расфокусировки камеры или смаз при движении объектов. Наиболее часто используются Гауссово размытие и размытие поля. Первое выполняет свертку изображения с маской Гауссова фильтра, а второе — с маской равномерного усреднения. Также применяются более сложные виды размытия, такие как motion blur (шевелёнка) и disk blur, имитирующие смаз от движения камеры или объекта и эффект боке соответственно.

Фокусировка, наоборот, повышает резкость изображения. Классический подход — использование нерезкого маскирования (unsharp masking), когда к изображению прибавляется его высокочастотная составляющая, выделенная с помощью фильтра Лапласа или Собеля. Это позволяет усилить границы и мелкие детали объектов.

Комбинируя размытие и фокусировку с различными параметрами, можно генерировать реалистичные вариации изображений с разной глубиной резкости. Это учит модель выделять объекты интереса независимо от фокуса камеры и не реагировать на размытый фон или передний план.

Микширование изображений

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

Два образца изображений и примеры послойного и патчевого смешивания, соответственно. Источник

Два образца изображений и примеры послойного и патчевого смешивания, соответственно. Источник

Один из простейших подходов — наложение одного изображения поверх другого с некоторой прозрачностью (alpha blending). Это позволяет комбинировать объекты из разных изображений и генерировать сэмплы с большим количеством объектов интереса.

Другой популярный метод — смешивание изображений (image blending), когда новое изображение получается путем поэлементного взвешенного суммирования двух или более исходных. Этот подход лежит в основе таких техник, как Mixup, CutMix и их многочисленных вариаций. В Mixup новое изображение является взвешенной суммой пары изображений, а соответствующая метка — взвешенной суммой их меток. CutMix работает аналогично, но смешивает не целые изображения, а их случайные фрагменты, что позволяет сохранить пространственную структуру объектов.

Методы для задач сегментации и обнаружения объектов

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

Различие между классификацией, обнаружением и сегментацией объектов. Источник

Различие между классификацией, обнаружением и сегментацией объектов. Источник

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

Специфичные для сегментации и детекции аугментации включают различные методы вырезания и вставки объектов. Простейший подход — случайное копирование объектов вместе с соответствующими фрагментами карты сегментации и вставка их на новое место в том же или другом изображении. Более сложные методы, такие как InstaBoost, адаптивно подбирают место и способ вставки объекта на основе карты вероятностей.

Другое направление — генерация синтетических данных на основе 3D-моделей. Имея трехмерные модели объектов и сцен, можно рендерить их с разных ракурсов, в разных условиях освещения и окружения, получая реалистичные изображения вместе с точной разметкой. Такой подход особенно эффективен для задач, где трудно получить реальные размеченные данные, например, для автономного вождения или роботизированных манипуляций.

Современные подходы на основе генеративных моделей

В последние годы генеративные модели, особенно генеративно-состязательные сети (GAN), произвели революцию в области синтеза изображений. Они позволяют создавать высококачественные реалистичные изображения, неотличимые от настоящих фотографий. Генеративные модели отлично подходят для задач аугментации, позволяя синтезировать примеры c более сложными и тонкими вариациями, чем у традиционных методов трансформации изображений.

Один из ключевых подходов — использование условных GAN (cGAN) для генерации изображений, удовлетворяющих заданным условиям, таким как метка класса, атрибуты объекта или семантическая карта. Примеры таких моделей — AC-GAN, BigBiGAN, StyleGAN. Они позволяют порождать разнообразные реалистичные изображения объектов заданного класса, с заданными свойствами или в заданном окружении. Использование таких синтетических данных для обучения значительно улучшает обобщающую способность и устойчивость моделей.

Система работы GAN. Источник

Система работы GAN. Источник

Другое направление — использование GAN для трансформации и редактирования существующих изображений. Модели, подобные CycleGAN и StarGAN, способны переносить стиль, изменять атрибуты объектов или превращать объекты одного класса в другой, сохраняя при этом реалистичность и целостность изображения. Такие трансформации расширяют разнообразие обучающих данных и позволяют моделировать сложные вариации внешнего вида объектов.

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

Аугментация в реальном времени

Традиционные методы аугментации данных предполагают создание дополнительных обучающих примеров в offline-режиме, до начала обучения модели. Однако в последнее время набирает популярность подход аугментации в реальном времени (online augmentation, real-time augmentation).

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

Визуализация процесса расширения данных в реальном времени во время обучения. Источник

Визуализация процесса расширения данных в реальном времени во время обучения. Источник

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

Важным направлением исследований является разработка адаптивных стратегий аугментации в реальном времени. Идея в том, чтобы динамически подстраивать вероятности и параметры различных аугментаций в зависимости от текущего состояния модели и сложности обучающих примеров. Примерами таких подходов можно назвать AutoAugment, RandAugment, Adversarial AutoAugment, которые используют обучение с подкреплением и состязательное обучение для автоматического поиска оптимальных стратегий аугментации, адаптированных к конкретной задаче и датасету.

Инструменты для аугментации данных

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

Обзор популярных библиотек и фреймворков

Наиболее универсальными и богатыми по возможностям являются специализированные библиотеки, такие как Albumentations, Imgaug и Augmentor

Albumentations предлагает более 70 различных трансформаций, оптимизированных для быстрой работы на CPU с данными в формате NumPy. Библиотека имеет унифицированный API, позволяющий легко комбинировать аугментации в произвольном порядке, а также встроенную поддержку вероятностей и величин трансформаций. 

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

Augmentor отличается простым высокоуровневым API для построения конвейеров аугментации данных и поддержкой многопоточной обработки, в том числе с использованием генераторов. Также активно появляются новые библиотеки, такие как Kornia, которая предоставляет функциональность для аугментации данных на GPU с использованием PyTorch.

Относительная производительность библиотек аугментации изображений. Источник

Относительная производительность библиотек аугментации изображений. Источник

Помимо специализированных библиотек, базовую функциональность по аугментации изображений предоставляют многие универсальные фреймворки и библиотеки компьютерного зрения и глубокого обучения. Так, широко известная библиотека OpenCV, являющаяся де-факто стандартом в области компьютерного зрения, содержит более 10 базовых техник аугментации. Фреймворки глубокого обучения, такие как Keras, TensorFlow и PyTorch, также имеют встроенные модули для аугментации данных. Несмотря на ограниченную функциональность по сравнению со специализированными библиотеками, их использование может быть оправдано простотой интеграции с конвейерами данных и моделями, реализуемыми средствами этих фреймворков.

Возможности OpenCV. Источник

Возможности OpenCV. Источник

Некоторые библиотеки и фреймворки компьютерного зрения предоставляют более специализированную функциональность. SimpleCV, ориентированная на задачи компьютерного зрения для встраиваемых систем и Raspberry Pi, содержит набор техник аугментации, оптимизированных для обработки видеопотоков и изображений с камер в режиме реального времени. Библиотека scikit-image для обработки научных изображений включает модули для геометрических и фотометрических преобразований, в том числе для моделирования искажений оптических систем.

Существуют также узкоспециализированные библиотеки для аугментации данных в отдельных прикладных областях. В некоторых исследованиях упоминается MRI Augmentation — набор техник для аугментации МРТ-изображений с учетом особенностей данной модальности, таких как анизотропность вокселей и специфика шумов. Automold, разработанная для задач автоматизированного визуального контроля качества, обеспечивает аугментацию изображений промышленных деталей и компонентов.

Сравнение возможностей и преимуществ различных инструментов

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

Первая характеристика — это спектр поддерживаемых техник аугментации. Специализированные библиотеки, такие как Albumentations и Imgaug, обычно предоставляют наиболее широкий набор трансформаций, включая как классические методы вроде геометрических и фотометрических преобразований, так и более продвинутые подходы на основе глубокого обучения и GANs. Универсальные библиотеки компьютерного зрения и фреймворки глубокого обучения, например OpenCV и Keras, зачастую ограничиваются базовым набором аугментаций. Однако последние, как правило, выигрывают в простоте использования за счет более высокоуровневого API и лучшей интеграции с другими компонентами соответствующих экосистем.

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

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

Четвертый фактор — кроссплатформенность и поддержка различных языков программирования. Большинство специализированных библиотек аугментации, равно как и фреймворков глубокого обучения, ориентированы на языки Python и R. Однако ряд библиотек, таких как OpenCV, Albumentations и Augmentor, предоставляют API и для других языков.

Практические аспекты

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

Оценка эффективности аугментации на этапе обучения модели

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

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

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

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

Кривая обучения до и после аугментации данных. Источник

Кривая обучения до и после аугментации данных. Источник

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

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

Необходимый объем данных для эффективной аугментации

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

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

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

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

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

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

Типичные ошибки при аугментации данных и способы их избежать

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

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

Случайные примеры дополненных изображений из MNIST. Столбцы слева направо: оригинал, масштаб, произвольно обрезанное, произвольно стертое, искаженное и размытое

Случайные примеры дополненных изображений из MNIST. Столбцы слева направо: оригинал, масштаб, произвольно обрезанное, произвольно стертое, искаженное и размытое

Другая типичная ошибка — использование слишком «агрессивных» аугментаций, приводящих к генерации нереалистичных или некорректных примеров. Так, в работе К. Чена и его коллег исследуется влияние различных параметров предвзятых полей на качество сегментации миокарда левого желудочка на МР-изображениях сердца. Обнаружено, что применение случайных деформаций с большой амплитудой приводит к появлению анатомически некорректных примеров и снижает точность сегментации. Аналогично, использование слишком сильных предвзятых полей ухудшает результаты, поскольку такие искажения нехарактерны для реальных МРТ-снимков. В таких областях лучше всегда иметь «при себе» эксперта в предметной области, который сможет проверить данные на достоверность.

Визуализация сгенерированных адверсариальных примеров и ошибочных предсказаний сети

Визуализация сгенерированных адверсариальных примеров и ошибочных предсказаний сети

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

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

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

Заключение

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

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

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

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

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

Будем рады услышать в комментариях про ваш опыт применения этой технологии!

Автор: breakmirrors

Источник

* - обязательные к заполнению поля


https://ajax.googleapis.com/ajax/libs/jquery/3.4.1/jquery.min.js