ViTaS: Visual Tactile Soft Fusion Contrastive Learning for Visuomotor Learning
About
Tactile information plays a crucial role in human manipulation tasks and has recently garnered increasing attention in robotic manipulation. However, existing approaches mostly focus on the alignment of visual and tactile features and the integration mechanism tends to be direct concatenation. Consequently, they struggle to effectively cope with occluded scenarios due to neglecting the inherent complementary nature of both modalities and the alignment may not be exploited enough, limiting the potential of their real-world deployment. In this paper, we present ViTaS, a simple yet effective framework that incorporates both visual and tactile information to guide the behavior of an agent. We introduce Soft Fusion Contrastive Learning, an advanced version of conventional contrastive learning method and a CVAE module to utilize the alignment and complementarity within visuo-tactile representations. We demonstrate the effectiveness of our method in 12 simulated and 3 real-world environments, and our experiments show that ViTaS significantly outperforms existing baselines. Project page: https://skyrainwind.github.io/ViTaS/index.html.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Insertion | Simulation | Insertion Success Rate98.2 | 14 | |
| Block Rotate | Simulation | Success Rate93.3 | 7 | |
| Block Spin | Simulation | Success Rate70.5 | 7 | |
| Door | Simulation | Success Rate1 | 7 | |
| Dual Arm Lift | Simulation | Success Rate100 | 7 | |
| Egg Rotate | Simulation | Success Rate85.7 | 7 | |
| Insertion | Simulation Noisy | Success Rate0.892 | 7 | |
| Lift | Simulation Capsule Shape | Success Rate99.6 | 7 | |
| Lift | Simulation Cylinder Shape | Success Rate97.8 | 7 | |
| Mobile Catch | Simulation | Success Rate64.9 | 7 |