Команда Лаборатории ИИ и Института искусственного интеллекта AIRI представила лучшее решение самой сложной задачи — Matching на Data Fusion Contest 2022.
Наши специалисты не остановились на базовом подходе (бустинг), а подошли к решению задачи с помощью нейронных сетей, подтвердили свою гипотезу и в итоге заняли 1-е место.
О задаче
По условиям задачи Matching нужно было осуществить сопоставление массивов синтетических данных и определить одних и тех же клиентов по их поведению в разных сферах.
Для решения задачи участникам предоставили:
данные транзакций клиентов по банковским картам;
данные о посещении web-страниц клиентов интернет-провайдера;
разметку соответствия между клиентами из этих организаций.
Датасет сформировали так, чтобы данные были обезличены, а информация о пользовательском поведении была сохранена.
В итоге для каждого клиента банка нужно было предоставить 100 наиболее вероятных клиентов провайдера. Точнее, код, который вычисляет ответ при вводе новых данных из тестового датасета, не использовавшихся при подготовке модели. При этом код должен корректно работать на любом компьютере, а на вычисление тратить не больше часа.
Решение
Финальное решение задачи — ансамбль из 11 сетей. Решение основано на библиотеке Сбера для работы с событийными данными PyTorch-Lifestream. Код можно посмотреть по ссылке и использовать для решения подобных задач.
Подробнее о том, как нейросеть обогнала бустинг, читайте здесь.
