Искусственный интеллект научили предсказывать поведение собаки

Исследователи из Вашингтонского университета и Института Аллена в области искусственного интеллекта создали нейросеть, которая может прогнозировать действия собаки на несколько секунд вперед. Препринт соответствующей работы опубликован на arXiv.org, кратко о ней рассказывает N+1.

Для создания алгоритма исследователи закрепили камеру на голове собаки, которая снимала видео от первого лица с частотой пять кадров в секунду. На лапах, хвосте и теле животного при этом были закреплены датчики, записывающие ее движения. В общей сложности были получены 24,5 тысячи кадров и соответствующих им данных о движениях собаки.

Нейросеть, которую создали исследователи, работает по тому же принципу, что и уже имеющиеся нейросети, которые создают следующие изображения после анализа уже имеющихся. Для этого они использовали три модели – одна из них предугадывала будущие движения собаки на основе предыдущих, другая – классифицировала объекты в кадре, третья – определяла последовательность движений, с помощью которых собака может переместиться из одного кадра в другой.

Алгоритм был обучен на большей части имеющейся базы, после чего он был протестирован на некотором количестве новых кадров. Результаты теста показали, что нейросеть способна планировать движения примерно так же, как это делает обычная собака. В частности, алгоритму удалось точно предсказать реакцию животного на брошенный человеком мячик.

Исследователи подчеркивают, что действие алгоритма в будущем можно улучшить, увеличив прогнозируемое время (сейчас оно составляет пять кадров), протестировав на большем количестве данных. Помимо этого, он может быть адаптирован и для применения на других животных.