Squint: Fast Visual Reinforcement Learning for Sim-to-Real Robotics
Organizations: Department of Computer Science and Engineering, University of California San Diego, La Jolla, CA 92093, USA
Abstract
Visual reinforcement learning is appealing for robotics but expensive. Off-policy methods are sample-efficient yet slow while on-policy methods parallelize well but waste samples. Recent work has shown that off-policy methods can train faster than on-policy methods in wall-clock time for state-based control. Extending this to vision remains challenging, where high-dimensional input images complicate training dynamics and introduce substantial storage and encoding overhead. To address these challenges, we introduce Squint, a visual Soft Actor Critic method that achieves faster wall-clock training than prior visual off-policy and on-policy methods. Squint achieves this via parallel simulation, a distributional critic, resolution squinting, layer normalization, a tuned update-to-data ratio, and an optimized implementation. We evaluate on the SO-101 Task Set, a new suite of eight manipulation tasks in ManiSkill3 with heavy domain randomization, and demonstrate sim-to-real transfer to a real SO-101 robot. We train policies for 15 minutes on a single RTX 3090 GPU, with most tasks converging in under 6 minutes.
Figures & tables
| Squint (Ours) | Visual DAgger | |
|---|---|---|
| Total | 73/80 | 53/80 |
| Average | 91.3% | 66.3% |
| Squint (Ours) | Squint - No Color Jitter | |
|---|---|---|
| Total | 73/80 | 58/80 |
| Average | 91.3% | 72.5% |
| Squint 16x16 (Ours) | Squint 32x32 | |
|---|---|---|
| Total | 73/80 | 73/80 |
| Average | 91.3% | 91.3 % |
Visual RL
Visual reinforcement learning trains agents to perceive images and learn actions through interaction, enabling control when states are not directly observed.
See all 59 papers