Equivariant Visual-Tactile Diffusion Policy for Contact-Rich Manipulation
Organizations: Department of Computer Science and Engineering, The Chinese University of Hong Kong · School of Computer Science and Engineering, Southeast University
Abstract
Imitation learning for contact-rich manipulation requires high-quality expert data that is expensive to obtain. This makes learning a sample-efficient policy a key issue. To address this, we propose VISTA, a workspace-level equivariant visuotactile diffusion policy for data-efficient contact-rich imitation learning. VISTA projects visual and tactile observations into spherical tokens, injects tactile contact cues into visual spherical directions through permutation-equivariant spherical fusion, and rotates the fused harmonic representation using the end-effector orientation. The resulting representation conditions an equivariant diffusion policy to predict spatially consistent actions. Extensive experiments in both simulation and real-world robotic settings show that VISTA substantially improves data efficiency over strong visuotactile imitation learning baselines. Project website: https://vista-paper.github.io/
Figures & tables
| Method | Grasp Classify | Insert HDMI | Insert Hole | Insert Tube | Lift Bottle | Lift Can | Pull Out Key | Put Bottle in Shelf | Mean |
| ManiFeel | 100.0 | 12.5 | 32.5 | 45.0 | 55.0 | 10.0 | 95.0 | 40.0 | 48.8 |
| VITAL | 100.0 | 7.5 | 27.5 | 55.0 | 20.0 | 17.5 | 65.0 | 57.5 | 43.8 |
| UniVTAC | 60.0 | 5.0 | 17.5 | 32.5 | 0.0 | 0.0 | 37.5 | 17.5 | 21.3 |
| EquiDiff + Tac | 95.0 | 10.0 | 42.5 | 92.5 | 57.5 | 37.5 | 97.5 | 87.5 | 65.0 |
| VISTA | 100.0 | 40.0 | 55.0 | 95.0 | 80.0 | 92.5 | 97.5 | 87.5 | 80.9 |
| Method | Erase | Insert USB | Insert pencil | Mean |
| EquiDiff + Tac | 1/10 | 3/10 | 0/10 | 1.33/10 |
| VISTA | 9/10 | 5/10 | 8/10 | 7.33/10 |
Appendix figures & tables8 assets
Supplementary material from the paper’s appendix.
Appendix
| Stage | Shape |
| Wrist RGB input | |
| Tactile RGB input | |
| Flattened image inputs | |
| Visual feature map | |
| Tactile feature map | |
| Visual sphere tokens |
| Variant | Tactile representation | Fusion stage | Fusion operator |
| ISP | none | – | – |
| VISTA-global- | invariant global vector | harmonic coefficients | addition |
| Early Fusion | 2D tactile feature map | before spherical lifting | conv on concatenated maps |
| ISP+Concat-Tac | spherical tokens | on the sphere, before harmonics | channel-wise concat + linear |
| VISTA | spherical tokens | on the sphere, before harmonics | cross-attention (visual , tactile ) |
| Task | VISTA group | Main randomized factors |
| Grasp classify | object class, grasp height, gripper closing pose | |
| Insert HDMI | slot , in-hand pitch/yaw, staging pose noise | |
| Insert hole | slot , slot yaw, target pitch, grasp height | |
| Insert tube | slot , grasp height, staging pose noise | |
| Lift bottle | bottle , bottle yaw, grasp pitch | |
| Lift can | can size, can , grasp pitch |
| Method | Group | Latency (ms) |
| ISP | 61.25 | |
| ISP | 65.66 | |
| VISTA | 63.74 | |
| VISTA | 71.54 | |
| EquiDiff + Tac | 79.88 |
| Method | Grasp Classify | Insert HDMI | Insert Hole | Insert Tube | Lift Bottle | Lift Can | Pull Out Key | Put Bottle in Shelf | Mean |
| ManiFeel | 100.0 | 12.5 | 32.5 | 45.0 | 55.0 | 10.0 | 95.0 | 40.0 | 48.8 |
| VITAL | 100.0 | 7.5 | 27.5 | 55.0 | 20.0 | 17.5 | 65.0 | 57.5 | 43.8 |
| UniVTAC | 60.0 | 5.0 | 17.5 | 32.5 | 0.0 | 0.0 | 37.5 | 17.5 | 21.3 |
| EquiDiff + Tac | 95.0 | 10.0 | 42.5 | 92.5 | 57.5 | 37.5 | 97.5 | 87.5 | 65.0 |
| VISTA | 100.0 | 40.0 | 55.0 | 95.0 | 80.0 | 92.5 | 97.5 | 87.5 | 80.9 |
| Method | Grasp Classify | Insert HDMI | Insert Hole | Insert Tube | Lift Bottle | Lift Can | Pull Out Key | Put Bottle in Shelf | Mean |
| ISP | 87.5 | 32.5 | 55.0 | 92.5 | 15.0 | 92.5 | 92.5 | 77.5 | 68.1 |
| VISTA | 100.0 | 40.0 | 55.0 | 95.0 | 80.0 | 92.5 | 97.5 | 87.5 | 80.9 |
| Method | Grasp Classify | Insert HDMI | Insert Hole | Insert Tube | Lift Bottle | Lift Can | Pull Out Key | Put Bottle in Shelf | Mean |
| ISP | 87.5 | 32.5 | 55.0 | 92.5 | 15.0 | 92.5 | 92.5 | 77.5 | 68.1 |
| VISTA-global- | 100.0 | 17.5 | 52.5 | 90.0 | 77.5 | 77.5 | 57.5 | 82.5 | 69.4 |
| Early Fusion | 100.0 | 42.5 | 37.5 | 90.0 | 75.0 | 77.5 | 82.5 | 77.5 | 72.8 |
| ISP+Concat-Tac | 100.0 | 37.5 | 67.5 | 85.0 | 67.5 | 92.5 | 77.5 | 65.0 | 74.1 |
| VISTA | 100.0 | 40.0 | 55.0 | 95.0 | 80.0 | 92.5 | 97.5 | 87.5 | 80.9 |
| Method | #Demo | Grasp Classify | Insert HDMI | Insert Hole | Insert Tube | Lift Bottle | Lift Can | Pull Out Key | Put Bottle in Shelf | Mean |
| ManiFeel | 50 | 100.0 | 12.5 | 32.5 | 45.0 | 55.0 | 10.0 | 95.0 | 40.0 | 48.8 |
| ManiFeel | 100 | 100.0 | 25.0 | 60.0 | 92.5 | 70.0 | 77.5 | 95.0 | 67.5 | 73.4 |
| VITAL | 50 | 100.0 | 7.5 | 27.5 | 55.0 | 20.0 | 17.5 | 65.0 | 57.5 | 43.8 |
| VITAL | 100 | 100.0 | 10.0 | 20.0 | 72.5 | 50.0 | 45.0 | 72.5 | 62.5 | 54.1 |
| UniVTAC | 50 | 60.0 | 5.0 | 17.5 | 32.5 | 0.0 | 0.0 | 37.5 | 17.5 | 21.3 |
| UniVTAC | 100 | 60.0 | 0.0 | 15.0 | 42.5 | 0.0 | 10.0 | 65.0 | 60.0 | 31.6 |