Pretrained Visual Representations in Reinforcement Learning
<|reference_start|>Pretrained Visual Representations in Reinforcement Learning: Visual reinforcement learning (RL) has made significant progress in recent years, but the choice of visual feature extractor remains a crucial design decision. This paper compares the performance of RL algorithms that train a convolutional neural network (CNN) from scratch with those that utilize pre-trained visual representations (PVRs). We evaluate the Dormant Ratio Minimization (DRM) algorithm, a state-of-the-art visual RL method, against three PVRs: ResNet18, DINOv2, and Visual Cortex (VC). We use the Metaworld Push-v2 and Drawer-Open-v2 tasks for our comparison. Our results show that the choice of training from scratch compared to using PVRs for maximising performance is task-dependent, but PVRs offer advantages in terms of reduced replay buffer size and faster training times. We also identify a strong correlation between the dormant ratio and model performance, highlighting the importance of exploration in visual RL. Our study provides insights into the trade-offs between training from scratch and using PVRs, informing the design of future visual RL algorithms.<|reference_end|>
@article{williams2024pretrained,
    title={Pretrained Visual Representations in Reinforcement Learning},
    author={Emlyn Williams, Athanasios Polydoros},
    journal={arXiv preprint arXiv:2407.17238},
    year={2024},
    archivePrefix={arXiv},
    eprint={2407.17238},
    primaryClass={cs.RO cs.LG}
}