Исследовательская группа из Helmholtz AI, компании Google DeepMind, Принстонского университета и иных учреждений представила Centaur — первую универсальную AI-модель, способную прогнозировать человеческое поведение в широком спектре когнитивных задач.
Centaur разработана на базе языковой модели Llama 3.1 (70 миллиардов параметров), прошедшей дополнительное обучение на наборе Psych-101. В набор вошли более 10 миллионов решений, принятых 60 тысячами участников в 160 экспериментах. Эти данные охватывают задачи, связанные с восприятием, памятью, логикой, моральным выбором и принятием решений в условиях неопределённости. Все эксперименты были переведены в текстовые инструкции, что позволило системе «понять» их как естественный язык. Этот подход дал возможность модели прогнозировать действия даже в новых и нестандартных условиях.
Centaur демонстрирует точные предсказания поведения даже в новых задачах. Например, в одном из тестов использовалась игра, похожая на слот-машину, где участники выбирали между двумя автоматами с разными шансами на выигрыш. Модель с высокой точностью предсказывала выбор участников. В более сложных задачах, связанных с логическими выводами и моральными дилеммами, Centaur превосходила существующие когнитивные модели.
Особенно интересно сходство работы модели и человеческого мозга. Учёные сравнили «внутренние представления» Centaur с результатами функциональной МРТ (метода регистрации активности мозга в реальном времени). Оказалось, что активация модели на задачи более близка к нейроактивности человека, чем у других моделей, включая GPT-4 и PaLM-2.
Centaur также способствует созданию новых когнитивных теорий. В одном из исследований она обнаружила новую стратегию в задаче, где люди учитывали не только вероятность, но и предстоящее вознаграждение. Ранее такое поведение не описывалось в психологии, однако, после выявления моделью, оно подтвердилось в ходе дополнительных проверок на участниках.
Платформа Psych-101, на которой обучалась модель, и сама Centaur доступны в открытом доступе. В будущем планируется расширение охвата — на детскую психологию, психиатрию и другие области.
Источник: iXBT
