Shortcut Learning in Generalist Robot Policies: The Role of Dataset Diversity and Fragmentation

## Контекст Область исследования связана с развитием общих (generalist) политик действий (policies) для роботов, которые могут выполнять широкий спектр задач в разных средах. Актуальной проблемой в этой области является ограниченность моделей в области общей целесообразности (generalization capability) — возникающая из-за того, что модели, при недостаточном разнообразии в обучающих данных, начинают опираться на специфические, но не значимые для решения задачи фичи (shortcut learning). Это проблема становится особенно яркой при работе с большими датасетами, которые состоят из независимых поддатасетов, собранных в различных условиях. Целью работы является не только идентификация факторов, которые способствуют этой проблеме, но и поиск способов ее устранения. ## Метод Для научного исследования использовались два основных подхода. Во-первых, был проведен теоретический анализ, позволяющий выявить основные причины shortcut learning — недостаточную разнообразность в поддатасетах и значимые различия в их распределениях (dataset fragmentation). Во-вторых, проводились эмпирические эксперименты, в ходе которых были протестированы различные модели на разных поддатасетах, а также на варьированных вариантах данных с помощью техник разметки и аугментации. Таким образом, было установлено, как разнообразие и структура данных влияют на общую способность модели к общей целесообразности. ## Результаты В ходе экспериментов был выявлен, что роботские модели, обученные на недостаточно разнообразных или сильно разбитых поддатасетах, используют в своих решениях сильно зависящие от контекста, но незначимые для задачи фичи. Были проведены ряд экспериментов с аугментацией данных, которые показали, что стратегия добавления новых, разнообразных обучающих примеров может существенно улучшить общую способность модели. Также были проанализированы результаты работы моделей на различных симуляционных и реальных задачах, что подтвердило тот факт, что улучшение разнообразия данных и их структуры позволяет значительно снизить уровень shortcut learning. ## Значимость Результаты работы имеют практическое значение для развития общих моделей действий для роботов. Они позволяют улучшить общую способность моделей, уменьшая риск зависимости от незначимых для задачи фич. Это влечет за собой рост эффективности в различных приложениях, включая управление роботами в различных ситуациях, включая нестандартные. Данные подходы также могут быть применены в учебных датасетах для роботов, чтобы снизить риск ошибок в ситуациях, где новые данные недоступны или дорогостоящи. ## Выводы Основным достижением является выявление того, что ограниченность раз