Исследовательская лаборатория Yandex Research анонсировала нейросетевую архитектуру TabM – она предназначена для обработки данных из таблиц. Инструмент показал большую эффективность при решении прикладных задач в бизнесе, медицинских и научных исследованиях.
В основе TabM лежит ансамблевый подход: прогноз формируется путем объединения результатов от нескольких подмоделей – это позволяет добиться высокой точности при умеренных вычислительных затратах. Архитектура конкурирует с признанными лидерами в области – бустингами CatBoost, XGBoost, Light GBM.
В результате тестов на 46 наборах данных TabM стабильно занимает первые места по качеству предсказаний и опережает конкурентов.
Один из примеров применения – предсказание уровня выживаемости пациентов после трансплантации стволовых клеток на Кaggle – это платформа для международных соревнований по анализу данных и машинному обучению от Google. Команды, использовавшие новую технологию, стали призерами и в сумме заработали шестьдесят тысяч долларов.
Исходный код архитектуры опубликован на GitHub, научная статья выложена на arXiv.
Комментарии