Карточка патентного документа
(11) TJ 1601 (13) PP Действует

СПОСОБ ГЕНЕРАЦИИ РЕАЛИСТИЧНЫХ ТАБЛИЧНЫХ ДАННЫХ С ВЫБРОСАМИ НА ОСНОВЕ ГЕНЕРАТИВНО-СОСТЯЗАТЕЛЬНОЙ СЕТИ zGAN

(51) МПК G06N 3/04 (45) Опубликован 30.05.2025
Сведения о документе ▾
(11) TJ Номер документа
1601
(13) Код типа документа
PP
(51) МПК
G06N 3/04
(45) Дата публикации
30.05.2025
Статус
Действует
Заявка ▾
(21) Рег. номер заявки
2502087
(22) Дата подачи заявки
14.04.2025
Лица ▾
(71) Заявитель(и)
Компания «ЗЕХН.АЙ» ЛТД («ZEHN.AI» Ltd). (TJ)
(72) Автор(ы)
Азизджон Азими (TJ)
(73) Патентообладатель(и)
Компания «ЗЕХН.АЙ» ЛТД («ZEHN.AI» Ltd). (TJ)

(57)Реферат

(56)     Изобретение относится к области искусственного интеллекта, в частности к способу генерации реали­стичных табличных данных с выбросами с использо­ванием генеративно-состязательной сети zGAN, включающей генератор, дискриминатор и модуль генерации ковариаций (covGEN). Генератор инициа­лизируют случайным шумом, обучают дискримина­тор различать синтетические и реальные данные, а генератор — создавать реалистичные данные с при­менением метода обратного распространения ошибки. Для генерации выбросов вычисляют ковариационные матрицы на основе реальных данных через модуль covGEN и интегрируют их в синтетические данные в соответствии с заданными распределениями (нор­мальное, Лапласовское, Вейбулловское, Гумбелев- ское, Леви). Приватность данных обеспечивается фильтром правдоподобия с использованием хеширо­вания.

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

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

Формула изобретения

Способ генерация реалистических табличных данных с выбросами, включающий использование архитектуры, позволяющей производить условную генерацию для фокусирования модели на специфи­ческих подмножествах данных, сохраняя редкие ка­тегории и сложные зависимости между признаками, инициализирование генератора случайным шумом для формирования синтетических данных, обучение генератора на повышение его способности формиро­вать реалистические синтетические данные с приме­нением метода обратного распространения ошибки, обучение дискриминатора на различении синтетиче­ских данных, поступающих от генератора и реаль­ных данных с целью повышения реалистичности синтетических данных, отличающийся тем, что для генерация синтетичных данных применяют архитек­тура генеративно-состязательной сети (zGAN), со­держащая генератор, дискриминатор, модуль генера­ции ковариацией (covGEN), хешированный фильтр правоподобия, условный вариационный автоэнкодер (cVAE), а также блоки позволяющие генерировать синтетические целевые переменные, причем, приме­нение архитектуры (zGAN) обеспечивает генерацию синтетических табличных данных с целенаправленно включенными выбросами, распределительными в соответствии нормальным, Лапласовским, Вейбул- ловским, Гумбелевским и Леви распределениями, используют условным вариационным автоэнкодер (cVAE) для генерации синтетических ковариацион­ных матриц с целью последующей генерации выбро­сов, вычисляют ковариационные матрицы на основе реальных данным с использованием модуля covGEN для создания выбросов, обеспечивают выбор между применением синтетических ковариационных мат­риц, получаемых от модуля cVAE и ковариационных матриц, полученных на основе реальных данных от модуля covGEN для последующей целенаправленной генерации синтетических выбросов, применяют фильтр правдоподобия для повышения приватности сгенерированных данных путем использования хеш кодов с исключением сохранения клиентских дан­ных, а также проводят вычисление показателей каче­ства, таких как AUC, для оценки стабильности и предсказательной способности модели на основе сгенерированных данных и применяют синтетиче­ских данных для обучения моделей машинного обу­чения и анализа влияния выбросов на показатель качества классификационной модели.

Описание

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

Известен способ CopulaGAN [1], который ком­бинирует в своей архитектуре возможности класси­ческих копул и принципы игрового обучения харак­терные для GAN в целом [2]. По назначению Copu- laGAN является аналогом zGAN.

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

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

CopulaGAN активно используется для задач синтеза табличных данных в финансовой и медицин­ской сферах, где важно сохранять взаимосвязи меж­ду признаками для анализа и моделирования. Не­смотря на перечисленные достоинства, модель усту­пает другим архитектурам GAN, таким как CTGAN и zGAN, в точности сохранения корреляций, возмож­ности целенаправленной генерации данных с выбро­сами, а также воспроизведении сложных нелинейных зависимостей.

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

Известен способ TVAE [1], который по назна­чению является аналогом zGAN. Данная генератив­ная модель также предназначена для генерации таб­личных данных, однако основана на принципе дей­ствия вариационного автоэнкодера, без использова­ния принципа действия характерного для GAN.

TVAE (Tabular Variational AutoEncoder) — это генеративная модель, предназначенная для создания синтетических табличных данных. Основой модели является вариационный автоэнкодер (VAE), который обучается формировать представление данных в ла­тентном пространстве и восстанавливать их с мини­мальной потерей информации. Такой подход позво­ляет способу TVAE моделировать сложные вероят­ностные зависимости в данных, делая её особенно эффективной для табличных наборов с разными ти­пами признаков.

В отличие от генеративно-состязательных се­тей, TVAE строится на основе принципа максимиза­ции доказательного нижнего порога (ELBO), что де­лает процесс её обучения более стабильным. Модель способна учитывать категориальные и числовые при­знаки, а также справляться с отсутствующими дан­ными путём их вероятностного восстановления. Ла­тентное пространство позволяет упростить структуру данных, делая процесс генерации более эффектив­ным.

Способ TVAE имеет свои ограничения. Из-за вероятностного характера генерации модель может хуже сохранять точные взаимосвязи между призна­ками, особенно в данных с высокой корреляцией между признаками или наличием сложных зависимо­стей. Кроме того, качество генерируемых данных может снижаться при наличии большого числа кате­гориальных признаков или при дисбалансе классов в исходных данных.

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

Наиболее близким к заявленному техническо­му решению по технической сущности и достигае­мому техническому результату является CTGAN [3], который является прототипом zGAN. Концептуально архитектура CTGAN приближена к части архитекту­ры zGAN отвечающей за обучение и генерацию син­тетических табличных данных.

CTGAN (Conditional Tabular Generative Adversarial Network) — это       генеративно­

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

Способ CTGAN включает специализированные компоненты для обработки табличных данных, такие как Gumbel-Softmax для категориальных признаков и методы перестройки дискретных данных для точного моделирования их распределений. Инициализируе­мый случайным шумом генератор прототипа создаёт синтетические данные, а дискриминатор обучается отличать их от реальных. Условная генерация позво-

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

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

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

zGAN существенно отличается как от прототи­па, так и от аналогов наличием возможности целена­правленной генерации синтетических выбросов в соответствии с теорией экстремальных значений [4]. Такая возможность позволяет моделировать редкие экономические события, обнаруживать и обрабаты­вать выбросы, что способствует повышению показа­телей качества моделей машинного обучения в раз­личных прикладных областях [5, 6, 7].

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

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

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

Модуль covGEN позволяет генерировать син­тетические выбросы в выбранных признаках с ис­пользованием задаваемых законов распределений. Фильтр правдоподобия, работающий с применением алгоритмов хеширования, позволяет сохранять при­ватность генерируемых данных благодаря отсут-

ствию необходимости сохранения клиентских дан­ных.

Оценка качества синтетических данных осу­ществляется по показателю AUC на примере реше­ния задачи бинарной классификации.

Краткое описание чертежей

Фиг. 1 иллюстрирует структуру zGAN

Фиг. 2 иллюстрирует структуру описывающую последовательность действий в поставленных экспе­риментах

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

Краткое описание таблиц

Таблица 1 содержит показатели приватных наборов данных

Таблица 2 содержит медианные значения AUC полученные в эксперименте «out-of-sample»

Таблица 3 содержит медианные значения AUC при «out-of-time» эксперименте с zGAN на 50% наборе данных

Таблица 4 содержит медианные значения AUC при «out-of-time» эксперименте с CTGAN на 50% наборе данных

Таблица 5 содержит медианные значения AUC для различных процентов выбросов в макроданных для набора данных A9 в «out-of-time» эксперименте

Генератор генеративно-состязательной сети zGAN инициализируется случайным шумом, форми­руя синтетические данные на выходе. Дискримина­тор получает на вход как сгенерированные синтети­ческие, так и реальные данные и выдает двоичный код, соответствующий классам «реальные» или «син­тетические» данные.

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

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

zGAN, обученная на реальных табличных дан­ных, генерирует синтетические данные, неотличимые от реальных, в которые интегрируются макровыбро­сы. Модуль, который интегрирует макровыбросы в синтетические данные, обозначается символом «+».

Макровыбросы синтезируются генератором условной ковариации (covGEN) путем выборки из многомерного распределения, выбираемого вручную, для чего требуется либо ковариация, полученная из реальных данных, либо синтетические ковариацион­ные матрицы, сгенерированные условным вариаци­онным автоэнкодером (cVAE) [8]. Для генерации вы­бросов в соответствии с теорией экстремальных зна­чений (ТЭЗ) в настоящее время доступны нормаль­ное, Лапласовское, Вейбулловское, Гумбелевское и Леви распределения.

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

Синтетические табличные данные могут быть необходимы для задач, связанных с дальнейшим ана­лизом отдельных признаков набора данных, в част­ности целевых переменных [9, 10, 11]. Для формиро­вания синтетических целевых переменных, интегри­рованных в синтетические данные с выбросами, ис­пользуется модуль Целевая Модель, основанный на модели классификации, например, такой как Catboost. Модель классификации обучается на вы­борке реальных данных для получения слабой объяс­нительной информации из столбцов набора данных, содержащих макро- и микроданные.

Способность zGAN генерировать реалистич­ные синтетические данные может быть проверена экспериментально и сравнена с метриками качества других GAN.

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

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

Задача классификации выполнялась в двух ва­риантах: с разделением данных на обучающую и те­стовую выборки по отдельным наблюдениям, то есть в режиме «out-of-sample» (OOS), и с разделением данных по времени, то есть в режиме «out-of-time» (OOT). Способность zGAN генерировать синтетиче­ские данные позволяющие сохранять центральную тенденцию и дисперсию AUC при решении класси­фикационной задачи с разными пропорциями синте­тических и реальных данных характеризуют устой­чивость модели. Устойчивость модели демонстриру­ется, в том числе воспроизведением классификаци­онных экспериментов на разных наборах данных, как в режиме OOS, так и в режиме OOT.

Для демонстрации способности GAN сохра­нять и воспроизводить корреляции признаков был проведен корреляционный анализ на исходных и синтетических наборах данных.

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

-    Парсинг данных;

-    Предобработка данных;

-    Отбор признаков;

-    Разделение данных на обучающую и тесто­вую выборки;

-    Вычисление AUC базовой модели;

-    Семплирование обучающих данных (опцио­нально);

-    Обучение GAN на реальных обучающих дан­ных;

-    Генерация синтетических данных;

-    Выборка 80% обучающих и тестовых данных (опционально);

-    Обучение классификатора на синтетических данных;

-    Расчёт AUC на тестовых данных.

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

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

Эксперименты «Out-of-Sample» В экспери­ментах OOS, разделение на обучающую и тестовую выборки осуществлялось с учётом необходимости получения одинаковых законов распределения в обе­их выборках.

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

Базовая классификационная модель обучалась на обучающей выборке, а AUC вычислялся на тесто­вой. Для получения разброса метрики применялся подход Монте-Карло, имплементируемый следую­щим образом: случайным образом формировалась

 

выборка, составляющая 80% от общего объема дан­ных, отдельно для обучающих и тестовых данных. На каждой из полученных выборок обучалась клас­сификационная модель, и вычислялась соответству­ющая метрика. Итоговая метрика, представленная в исследовании, — это медианное значение AUC, по­лученное за 51 эксперимент для каждой модели и набора данных отдельно. 51-е значение в результатах экспериментов — это значение AUC, полученное без 80% выборки, то есть оно было вычислено с исполь­зованием всех записей из обучающих и тестовых наборов данных.

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

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

Эксперименты «Out-of-Time» В эксперимен­тах OOT набор данных был разделен на обучающие и тестовые выборки в порядке возрастания дат. Дан­ные делились на обучающую и тестовую выборки отдельно в двух пропорциях: 50% и 80%.

В классификационном OOT эксперименте ис­пользовались для генерации синтетических данных zGAN и его прототип CTGAN, классификационная модель обучалась на реальном обучающем наборе данных, при этом метрика AUC вычислялась только на реальном тестовом наборе данных, аналогично процедуре производимой в OOS экспериментах.

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

В «Out-of-Time» экспериментах была также ис­следована стабильность и повышение AUC в зависи­мости от количества выбросов, сгенерированных из нормального распределения на уровне 3ст в макропе­ременных наборов данных.

Такие эксперименты проводились только для zGAN, поскольку ни аналоги, ни прототип не обла­дают возможностью целенаправленной генерации выбросов. В этих экспериментах каждый набор дан­ных разделялся на обучающую и тестовую выборки на основе временных меток. Часть набора данных до 2022 года использовалась как обучающая выборка, в то время как данные с 2022 года включительно со­ставляли тестовую выборку. Модель zGAN обуча­лась на тренировочном наборе, после чего для каж­дого процента выбросов в макропеременных, вклю­чая вариант без выбросов, генерировалось по 80 син­тетических наборов данных.

В задаче бинарной классификации модель классификации обучалась для каждого набора синте­тических данных. AUC оценивался с применением метода Монте-Карло, на случайной выборке из те­стового набора, которая составляла 80% от всего те­стового набора данных. Последний шаг, включал 81­ю итерацию вычисления AUC. Для обучения класси­фикационной модели на последней итерации, ис­пользовался объединённый набор данных, состоя­щий из всех 80 сгенерированных наборов для каждо­го процента выбросов в отдельности, при этом AUC вычислялся на всем тестовом наборе.

Для оценки влияния выбросов на AUC в каче­стве метрик оценки использовались медианное зна­чение AUC, размах и межквартильный размах (IQR).

Корреляционный анализ. Для оценки спо­собности GAN сохранять и воспроизводить корреля­ции признаков использовался коэффициент корреля­ции Пирсона. Как исходный набор данных, так и синтетические данные были предварительно обрабо­таны следующим образом: пропущенные значения в числовых столбцах заполнялись консистентными значениями, которые значительно отличались от среднего значения столбца и не встречались в других местах данных. В категориальных столбцах пропу­щенные значения были сгруппированы в отдельную категорию, а все записи в категориальных столбцах были                             закодированы        с              использованием

OrdinalEncoder. Столбцы с датами и временем были преобразованы в формат Unix-time. После этих пре­образований набор данных был нормализован с ис­пользованием StandardScaler.

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

Описание данных В экспериментах использо­вались как приватные наборы данных с доступом, разрешенным компанией zypl.ai, так и открытый набор данных Титаник. В интересах владельцев ис­ходных данных характеристики приватных наборов, включая их названия, скрыты и обозначены прону­мерованными латинскими буквами: A1, ..., A9. Все наборы данных демонстрируют дисбаланс классов в целевой переменной. Коэффициент kim, использую­щийся для описания дисбаланса классов, представля­ет собой отношение количества образцов в минори­тарном классе к количеству образцов в мажоритар­ном классе.

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

Отношение миноритарных классов к мажори-                 менного ряда в приватных наборах данных приведе-

тарным kim, общее количество столбцов и записей,                    ны в таблице 1.

количество категориальных столбцов и наличие вре-

Таблица 1

Показатели

Ai

A2

Аз

Ад

А5

А6

А7

А8

А9

kim

0.07

0.19

0.02

0.08

0.01

0.14

0.21

0.23

0.06

Количество столбцов

21

29

33

18

29

15

30

68

25

Количество за­писей

347K

62K

1.76M

34K

759K

18K

3189

139K

2721

Количество кате­гориальных столбцов

6

10

16

18

13

6

15

10

9

Количество кате­горий

56

44

370

60

44

29

242

47

67

Наличие time­series

V

V

V

X

X

X

V

V

V

 

Эксперименты были проведены, в том числе, на открытом наборе данных - Титаник. Обучающая выборка оригинального набора данных Титаник была разделена на обучающую и тестовую выборки в со­отношении 0.33. Обучающие данные включали 596 записей, а тестовые данные — 295 записей.

Целевая переменная — «Survived», признаки, отобранные для дальнейших экспериментов: «Pclass», «Sex», «Age», «SibSp», «Parch», «Fare», «Cabin», «Embarked». Отношение миноритарного класса к мажоритарному kim составляет 0.62. Набор данных содержит всего 9 признаков, из которых 7 являются категориальными. В общей сложности в этих столбцах присутствуют 173 категории.

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

Описание моделей Эксперименты включали две GAN модели, а именно CTGAN (прототип) и CopulaGAN (аналог); одну альтернативную модель - TVAE (аналог). CTGAN использует условную веро­ятность для моделирования сложных зависимостей в табличных данных, что делает его эталоном для за­дач, включающих как числовые, так и категориаль­ные переменные. TVAE предлагает вероятностный подход, основанный на автоэнкодерах, что позволяет моделировать сложные распределения.

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

Результаты экспериментов
Классификационная задача

Out-of-Sample эксперименты В таблице 2 по­казаны медианные значения AUC полученные в ре­зультате экспериментов.

 

Таблица 2

Набор данных

zGAN

CTGAN

ТЛАЕ

CopulaGAN

Baseline

А1

0.7553

0.7496

0.7504

0.6681

0.7841

А2

0.7972

0.7633

0.7428

0.7959

0.8216

А3

0.7230

0.7035

0.6461

0.6219

0.7552

А4

0.7095

0.6350

0.6244

0.6663

0.7343

А5

0.8311

0.8284

0.6163

0.6763

0.8680

А6

0.8120

0.8025

0.7898

0.7942

0.8154

А7

0.7704

0.7013

0.7279

0.7093

0.8183

А8

0.7118

0.6992

0.7257

0.7010

0.7907

А9

0.6525

0.5705

0.6874

0.5955

0.7109

Титаник

0.8163

0.7923

0.7874

0.8154

0.8736

 

Результаты показывают, что в восьми из десяти экспериментов модель zGAN превосходит все анало­ги и прототип, за исключением базовой модели, обу­ченной на реальных данных. В среднем AUC модели zGAN на 0,03 пункта выше, чем у CTGAN, и на 0,05 пункта выше, чем у TVAE или CopulaGAN.

Out-of-Time эксперименты Результаты OOT экспериментов с моделью zGAN, в которых исполь­зовалась пропорциональная смесь реальных и синте­тических данных, представлены ниже в таблице 3.

 

           
    Подпись: Набор дан¬ных	100% Синте-тические	1:1	0.1:1	0.01:1	0.001:1	100% Ре-альные Данные
Ai	0.6738 (0.6641 : 0.6812)	0.6684
(0.6335	:
0.6862)	0.6262
(0.6017	:
0.6415)	0.6165
(0.5979	:
0.6320)	0.6138
(0.6014	:
0.6305)	0.6100
(0.5996	:
0.6184)
A2	0.7862 (0.7786 : 0.7917)	0.8108
(0.8015	:
0.8156)	0.8157
(0.8087	:
0.8204)	0.8191
(0.8138	:
0.8225)	0.8227
(0.8178	:
0.8264)	0.8238
(0.8195	:
0.8279)
Аз	0.7146 (0.7101 : 0.7186)	0.7302
(0.7264	:
0.7339)	0.7433
(0.7389	:
0.7472)	0.7435
(0.7395	:
0.7462)	0.7444
(0.7393	:
0.7479)	0.7456
(0.7417	:
0.7488)
А7	0.8250 (0.8134 : 0.8517)	0.8219
(0.8050	:
0.8446)	0.8092
(0.7854	:
0.8366)	0.8091
(0.7812	:
0.8295)	0.8076
(0.7830	:
0.8241)	0.8091
(0.7847	:
0.8378)
А8	0.7686 (0.7628 : 0.7725)	0.7804
(0.7767	:
0.7834)	0.7853
(0.7787	:
0.7893)	0.7859
(0.7807	:
0.7900)	0.7865
(0.7824	:
0.7903)	0.7859
(0.7799	:
0.7895)
А9	0.7117 (0.6674 : 0.7747)	0.7233
(0.6660	:
0.7805)	0.6443
(0.5628	:
0.7069)	0.6627
(0.5789	:
0.7597)	0.7046
(0.6568	:
0.7567)	0.6862
(0.6129	:
0.7429)
 
  Подпись: Экспериментальные результаты показывают сравнимые показатели предсказательной точности классификационных моделей, обученных на синте-тических данных zGAN, комбинации данных zGAN и реальных данных, а также базовой модели, обучен-ной только на реальных данных. В случаях наборов данных A1, A7 и A9 медианное значение AUC суще-ственно возрастает при использовании синтетиче-   Подпись: ских данных или добавлении пропорционально большого объема синтетических данных к реальным данным. В случае набора данных A1 AUC увеличива-ется на 0,064 пункта.
Сравнительные	результаты	OOT-
экспериментов, полученные на основе наборов дан-ных, сгенерированных CTGAN, показаны ниже в таблице 4.
 
 
    Подпись: Набор Дан-ных	100% Синте-тические	1:1	0.1:1	0.01:1	0.001:1	100% Ре-альные Данные
А1	0.7077 (0.6959 : 0.7151)	0.7054
(0.6675	:
0.7144)	0.6295
(0.6137	:
0.6678)	0.6177
(0.6047	:
0.6338)	0.6103
(0.6018	:
0.6196)	0.6100
(0.5996	:
0.6184)
А2	0.7815 (0.5000 : 0.7933)	0.8063
(0.8016	:
0.8111)	0.8199
(0.8140	:
0.8245)	0.8235
(0.8184	:
0.8275)	0.8237
(0.8189	:
0.8287)	0.8238
(0.8195	:
0.8279)
А3	0.7025 (0.6884 : 0.7109)	0.7368
(0.7317	:
0.7396)	0.7452
(0.7425	:
0.7478)	0.7453
(0.7414	:
0.7488)	0.7456
(0.7427	:
0.7491)	0.7456
(0.7417	:
0.7488)
А7	0.7707 (0.7456 : 0.7961)	0.8117
(0.7951	:
0.8370)	0.8108
(0.7862	:
0.8406)	0.8090
(0.7830	:
0.8324)	0.8079
(0.7802	:
0.8363)	0.8091
(0.7847	:
0.8378)
А8	0.7481 (0.7409 : 0.7527)	0.7761
(0.7700	:
0.7812)	0.7814
(0.7772	:
0.7857)	0.7855
(0.7798	:
0.7892)	0.7862
(0.7822	:
0.7898)	0.7859
(0.7799	:
0.7895)
А9	0.6861 (0.6731 : 0.7050)	0.7072
(0.6725	:
0.7427)	0.7136
(0.7039	:
0.7214)	0.7094
(0.6897	:
0.7202)	0.7104
(0.7014	:
0.7283)	0.6862
(0.6129	:
0.7429)


Таблица 4

 

В OOT-эксперименте по смешиванию реаль­ных и синтетических данных с использованием CTGAN было продемонстрировано влияние смеши­вания на медианное значение AUC. В случае набора данных A2 AUC, рассчитанное на реальных данных, на 0,001 пункта превышает наивысшее значение AUC, полученное при смешивании данных, тогда как в случае A3 улучшения AUC при смешивании не бы­ло. В случаях A1, A7, A8 и A9 наблюдалось увеличе­ние AUC при смешивании, при этом наибольшее увеличение составило 0,0977 пункта для датасета A1.

OOT-эксперименты по смешиванию реальных обучающих и синтетических данных показывают в большинстве случаев увеличение AUC, что также указывает на улучшение стабильности в задаче клас­сификации. Увеличение медианного AUC на синте­

 

тических данных или при смешивании синтетиче­ских данных с реальными данными наблюдается как в экспериментах с CTGAN, так и с zGAN, что указы­вает на то, что улучшение стабильности в задаче классификации происходит независимо от GAN, ис­пользуемой для генерации синтетических данных.

Результаты OOT-экспериментов с синтетиче­скими выбросами в макроданных синтетических наборов данных, сгенерированных zGAN, показаны в таблице 5, на примере набора данных A9, характери­зующего растущую экономику.

 
  Подпись: Процент выбросов	Медианное AUC (Размах)
100%	0.6862 (0.5706 : 0.7947)
50%	0.6867 (0.5669 : 0.7677)
10%	0.6962 (0.5736 : 0.7769)
7.7%	0.7061 (0.6357 : 0.7819)
7.4%	0.7122 (0.5985 : 0.8048)
7.1%	0.7077 (0.6039 : 0.7770)
7%	0.7095 (0.6170 : 0.7948)
6.9%	0.7074 (0.6163 : 0.7807)
6.6%	0.7079 (0.5985 : 0.7650)
6.3%	0.6966 (0.5838 : 0.7875)
6%	0.7096 (0.6371 : 0.7853)
5%	0.7147 (0.5858 : 0.7890)
3%	0.7116 (0.6130 : 0.7694)
1%	0.7032 (0.6027 : 0.7736)
Без выбросов	0.7020 (0.6154 : 0.8194)


Таблица 5

Эксперименты с синтетическими выбросами показывают увеличение медианного AUC в боль­шинстве случаев, без явной линейной зависимости от процента выбросов в макроданных. Когда количе­ство синтетических выбросов превышает 10% от об­щего числа записей в макроданных, медианный AUC снижается. Наибольшее увеличение медианного AUC наблюдается при 3%, 5% и 7% выбросов в столбцах макроданных. Значительное снижение AUC наблюдается, когда количество выбросов в синтети­ческих макроданных достигает 50 и более процентов. Максимальное и статистически значимое увеличение AUC наблюдается при 5% выбросов в синтетических макроданных.

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

Корреляционный анализ На Фиг. 3 представ­лены тепловые карты, иллюстрирующие различия между матрицами корреляции синтетических набо­ров данных, сгенерированных zGAN, прототипом и аналогами, по сравнению с оригинальным набором данных. Способ TVAE показывает умеренные разли­чия, что указывает на адекватное сохранение ориги­нальных корреляций. Способ CopulaGAN демон­стрирует более значительные изменения, что свиде­тельствует о менее надежном сохранении ориги­нальной структуры. Прототип - CTGAN и zGAN по­казывают наименьшие различия, при этом zGAN превосходит прототип, что указывает на наивысшую степень точности в поддержании оригинальной структуры корреляций. Таким образом, CTGAN и zGAN являются наиболее эффективными в генера­ции синтетических данных, которые близки к ориги­нальному набору данных, за которыми следует TVAE, который предлагает адекватный баланс меж­ду точностью и качеством сохранения корреляций, а CopulaGAN является наименее эффективным.

Заключение Эксперименты при out-of-sample разделении данных, демонстрируют сравнение каче­ства классификации на синтетических данных, сге­нерированных современными открытыми GAN и генеративными моделями, основанными на класси­ческих принципах, в том числе прототипом и анало­гами, демонстрируют преимущества zGAN по срав­нению с исследованными моделями. В среднем, AUC zGAN превышает таковой у прототипа на 0.03 пунк­та, а также на 0.05 пункта у аналогов.

Применение zGAN для генерации синтетиче­ских табличных данных позволяет модели классифи­кации изучать основные закономерности в данных без взаимодействия с реальными данными клиентов. Эксперименты при out-of-time разделении данных показывают, что обучение классификационной моде­ли на синтетических наборах данных улучшает ее способность к обобщению, положительно влияя на качество классификационной задачи при работе с будущими, ранее не виденными данными. В экспе­рименте по обогащению и смешиванию реальных данных с синтетическими данными для набора дан­ных A1 было достигнуто увеличение медианного AUC на 0.0638 пункта при использовании только синтетических данных для обучения, и увеличение AUC на 0.0584 пункта было получено при смешива­нии реальных данных для обучения с синтетически­ми данными в соотношении 1:1.

Генерация синтетических выбросов в случае OOT экспериментов, где тренировочные данные бы­ли собраны до 2022 года, а тестовые данные — с 2022 по 2023 год, приводит к значительному улуч­шению качества решения задачи классификации, как показано на наборе данных A9, представляющем рас-

тущую экономику. В эксперименте по генерации синтетических выбросов в макроданных синтетиче­ского набора данных было достигнуто увеличение медианного AUC при умеренном количестве синте­тических выбросов до 10%, в то время как увеличе­ние выбросов выше 50% снижало качество. Наибольшее и наиболее значительное повышение AUC на 0.0127 пункта наблюдалось при 5% выбросов. Эти результаты подчеркивают важность тща­тельной балансировки количества выбросов, особен­но в прогнозах out-of-time, где выбросы часто обла­дают большей предсказательной способностью.

В экспериментах по корреляционному анализу zGAN демонстрирует способность сохранять корре­ляции реальных данных на уровне прототипа, при этом превосходит аналоги

(56) Документы, цитированные в отчёте

   1. Xu, L., Skoularidou, M., Cuesta-Infante, A., & Veeramachaneni, K. (2019). Modeling tabular data using conditional GAN. arXiv preprint arXiv:1907.00503. До-ступно по ссылке: https://arxiv.org/abs/1907.00503.
2.    Ian J. Goodfellow, Jean Pouget-Abadie, Mehdi Mirza, Bing Xu, David Warde-Farley, Sherjil Ozair, Aa¬ron Courville, and Yoshua Bengio. Generative adversarial networks, 2014.
3.    Laurens de Haan and Ana Ferreira. Extreme Value Theory. Springer New York, 2006.
4.    Zheng, H., Liu, Y., Ji, M., Wu, F., & Fang, L. (2015). Learning High-level Prior with Convolutional Neural Networks for Semantic Segmentation. arXiv pre¬print arXiv: 1511.06988. https://arxiv.org/abs/1511.06988
5.    Kristina Bluwstein, Marcus Buckmann, Andreas Joseph, Sujit Kapadia, and Ozgur Simsek. Credit growth, the yield curve and financial crisis prediction: Evidence from a machine learning approach. Journal of Internation¬al Economics, 145:103773, 2023.
6.    M. Ghil, P. Yiou, S. Hallegatte, B. D. Malamud, P. Naveau, A. Soloviev, P. Friederichs, V. Keilis-Borok, D. Kondrashov, V. Kossobokov, O. Mestre, C. Nicolis, H. W. Rust, P. Shebalin, M. Vrac, A. Witt, and I. Zaliapin. Extreme events: dynamics, statistics and prediction. Non¬linear Processes in Geophysics, 18(3):295-350, 2011.
7.    Sunday Adeola Ajagbe and Matthew O. Adigun. Deep learning techniques for detection and prediction of pandemic diseases: a systematic literature review. Multi-media Tools and Applications, 83:5893-5927, 1 2024.
8.    Gowthami Somepalli, Micah Goldblum, Avi Schwarzschild, C. Bayan Bruss, and Tom Goldstein.
2
SAINT: improved neural networks for tabular data via row attention and contrastive pre-training. CoRR, abs/2106.01342, 2021.
9.    Blaz Skrlj, Saso Dzeroski, Nada Lavrac, and Matej Petkovic. Feature importance estimation with self-attention networks. CoRR, abs/2002.04464, 2020.
10.    Penghui Zhao, Zhongjun Ding, Yang Li, Xiao- han Zhang, Yuanqi Zhao, Hongjun Wang, and Yang Yang. Sgad-gan: Simultaneous generation and anomaly detection for time-series sensor data with generative ad¬versarial networks. Mechanical Systems and Signal Processing, 210:111141, 3 2024

📎 Прикреплённые файлы