Google DeepMind и Принстон обучили ИИ-модель понимать мышление человека

Команда исследователей из Helmholtz AI (исследовательская инициатива в Германии), Google DeepMind, Принстонского университета и других организаций представила Centaur — первую универсальную ИИ-модель, способную предсказывать поведение человека в большом числе когнитивных задач.

Centaur основана на языковой модели Llama 3.1 (70B параметров), которую дополнительно обучили на массиве Psych-101: это более 10 миллионов решений, принятых 60 тысячами участников в 160 экспериментах. Эти данные включают реакции на задачи восприятия, памяти, логического мышления, морального выбора и принятия решений в условиях неопределённости. Каждый эксперимент был переведён в текстовую форму, чтобы система могла «прочитать» задачи как инструкции на естественном языке. Такой подход позволил ей научиться предсказывать действия даже в новых, незнакомых условиях.

Centaur показывает точные прогнозы поведения даже в новых для неё задачах. Например, в одном из тестов использовалась игра, аналогичная слот-машине: испытуемые выбирали между двумя автоматами с разной вероятностью выигрыша. Модель предсказывала выбор участников с высокой точностью. В более абстрактных задачах логического вывода и моральной дилеммы Centaur также обошла существующие когнитивные модели.

Иллюстрация: Sora

Особый интерес вызывает сходство модели с работой мозга. Исследователи сравнили «внутренние представления» Centaur с результатами функциональной МРТ (fMRI — метод регистрации активности мозга в реальном времени). Оказалось, что паттерны активации модели в ответ на задачи ближе к нейроактивности человека, чем у любой другой модели, включая GPT-4 и PaLM-2.

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

Платформа Psych-101, на которой обучалась модель, и сама Centaur выложены в открытый доступ. В будущем планируется расширение — в том числе на детскую психологию, психиатрию и другие сферы.