Visual Encoders for Data-Efficient Imitation Learning in Modern Video Games

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Schäfer, Lukas, Jones, Logan, Kanervisto, Anssi, Cao, Yuhan, Rashid, Tabish, Georgescu, Raluca, Bignell, Dave, Sen, Siddhartha, Gavito, Andrea Treviño, Devlin, Sam
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908427778785280
author Schäfer, Lukas
Jones, Logan
Kanervisto, Anssi
Cao, Yuhan
Rashid, Tabish
Georgescu, Raluca
Bignell, Dave
Sen, Siddhartha
Gavito, Andrea Treviño
Devlin, Sam
author_facet Schäfer, Lukas
Jones, Logan
Kanervisto, Anssi
Cao, Yuhan
Rashid, Tabish
Georgescu, Raluca
Bignell, Dave
Sen, Siddhartha
Gavito, Andrea Treviño
Devlin, Sam
contents Video games have served as useful benchmarks for the decision-making community, but going beyond Atari games towards modern games has been prohibitively expensive for the vast majority of the research community. Prior work in modern video games typically relied on game-specific integration to obtain game features and enable online training, or on existing large datasets. An alternative approach is to train agents using imitation learning to play video games purely from images. However, this setting poses a fundamental question: which visual encoders obtain representations that retain information critical for decision making? To answer this question, we conduct a systematic study of imitation learning with publicly available pre-trained visual encoders compared to the typical task-specific end-to-end training approach in Minecraft, Counter-Strike: Global Offensive, and Minecraft Dungeons. Our results show that end-to-end training can be effective with comparably low-resolution images and only minutes of demonstrations, but significant improvements can be gained by utilising pre-trained encoders such as DINOv2 depending on the game. In addition to enabling effective decision making, we show that pre-trained encoders can make decision-making research in video games more accessible by significantly reducing the cost of training.
format Preprint
id arxiv_https___arxiv_org_abs_2312_02312
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Visual Encoders for Data-Efficient Imitation Learning in Modern Video Games
Schäfer, Lukas
Jones, Logan
Kanervisto, Anssi
Cao, Yuhan
Rashid, Tabish
Georgescu, Raluca
Bignell, Dave
Sen, Siddhartha
Gavito, Andrea Treviño
Devlin, Sam
Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
Video games have served as useful benchmarks for the decision-making community, but going beyond Atari games towards modern games has been prohibitively expensive for the vast majority of the research community. Prior work in modern video games typically relied on game-specific integration to obtain game features and enable online training, or on existing large datasets. An alternative approach is to train agents using imitation learning to play video games purely from images. However, this setting poses a fundamental question: which visual encoders obtain representations that retain information critical for decision making? To answer this question, we conduct a systematic study of imitation learning with publicly available pre-trained visual encoders compared to the typical task-specific end-to-end training approach in Minecraft, Counter-Strike: Global Offensive, and Minecraft Dungeons. Our results show that end-to-end training can be effective with comparably low-resolution images and only minutes of demonstrations, but significant improvements can be gained by utilising pre-trained encoders such as DINOv2 depending on the game. In addition to enabling effective decision making, we show that pre-trained encoders can make decision-making research in video games more accessible by significantly reducing the cost of training.
title Visual Encoders for Data-Efficient Imitation Learning in Modern Video Games
topic Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2312.02312