about
Masked Autoencoding for Scalable and Generalizable Decision Making (arxiv.org)
1 point by tim_sw on Apr 28, 2023 | hide | past | pdf | discuss on HN

In plain words: An agent is pretrained by hiding random parts of state-and-action sequences and filling in the missing pieces, forcing it to learn how situations change. It then handles new tasks with no extra training and matches earlier one-step-at-a-time pretraining after light fine-tuning.

Abstract

We are interested in learning scalable agents for reinforcement learning that can learn from large-scale, diverse sequential data similar to current large vision and language models. To this end, this paper presents masked decision prediction (MaskDP), a simple and scalable self-supervised pretraining method for reinforcement learning (RL) and behavioral cloning (BC). In our MaskDP approach, we employ a masked autoencoder (MAE) to state-action trajectories, wherein we randomly mask state and action tokens and reconstruct the missing data. By doing so, the model is required to infer masked-out states and actions and extract information about dynamics. We find that masking different proportions of the input sequence significantly helps with learning a better model that generalizes well to multiple downstream tasks. In our empirical study, we find that a MaskDP model gains the capability of zero-shot transfer to new BC tasks, such as single and multiple goal reaching, and it can zero-shot infer skills from a few example transitions. In addition, MaskDP transfers well to offline RL and shows promising scaling behavior w.r.t. to model size. It is amenable to data-efficient finetuning, achieving competitive results with prior methods based on autoregressive pretraining.

Fangchen Liu, Hao Liu, Aditya Grover, Pieter Abbeel
arXiv:2211.12740 · cs.LG, cs.AI · submitted Nov 23, 2022 · updated May 27, 2023
abstract · pdf · html

add comment on HN