Исследователи Yandex Research разработали нейросетевую архитектуру TabM, которая эффективно обрабатывает табличные данные и строит высокоточные прогнозы. Решение уже доказало свою эффективность в бизнесе, медицине и науке.
Yandex Research представила новую нейросетевую архитектуру TabM для работы с табличными данными, которая быстро обрабатывает большие массивы информации и строит точные прогнозы. Эта технология востребована в бизнесе, науке и медицине — от оптимизации поставок до прогнозирования заболеваний. Разработка уже доказала свою эффективность на платформе Kaggle, где ее использовали для предсказания выживаемости пациентов после трансплантации костного мозга. Благодаря TabM победители соревнований заработали в сумме 60 тысяч долларов.
TabM — это ансамбль моделей, где каждая проводит свой анализ, а итоговый прогноз усредняется. Это обеспечивает оптимальный баланс между точностью и затратами вычислительных ресурсов. Тесты на 46 наборах данных показали, что TabM превосходит аналоги по точности (среднее место 1,7 против 2,9 у ближайшего конкурента) и стабильности работы.
Новая архитектура успешно конкурирует с классическими методами вроде CatBoost, XGBoost и LightGBM, которые раньше считались лучшими для табличных данных. TabM уже доступна разработчикам на GitHub, открывая новые возможности для анализа и прогнозирования в разных сферах.