English

Human-oriented Representation Learning for Robotic Manipulation

Robotics 2023-10-05 v1 Artificial Intelligence Computer Vision and Pattern Recognition Machine Learning

Abstract

Humans inherently possess generalizable visual representations that empower them to efficiently explore and interact with the environments in manipulation tasks. We advocate that such a representation automatically arises from simultaneously learning about multiple simple perceptual skills that are critical for everyday scenarios (e.g., hand detection, state estimate, etc.) and is better suited for learning robot manipulation policies compared to current state-of-the-art visual representations purely based on self-supervised objectives. We formalize this idea through the lens of human-oriented multi-task fine-tuning on top of pre-trained visual encoders, where each task is a perceptual skill tied to human-environment interactions. We introduce Task Fusion Decoder as a plug-and-play embedding translator that utilizes the underlying relationships among these perceptual skills to guide the representation learning towards encoding meaningful structure for what's important for all perceptual skills, ultimately empowering learning of downstream robotic manipulation tasks. Extensive experiments across a range of robotic tasks and embodiments, in both simulations and real-world environments, show that our Task Fusion Decoder consistently improves the representation of three state-of-the-art visual encoders including R3M, MVP, and EgoVLP, for downstream manipulation policy-learning. Project page: https://sites.google.com/view/human-oriented-robot-learning

Keywords

Cite

@article{arxiv.2310.03023,
  title  = {Human-oriented Representation Learning for Robotic Manipulation},
  author = {Mingxiao Huo and Mingyu Ding and Chenfeng Xu and Thomas Tian and Xinghao Zhu and Yao Mu and Lingfeng Sun and Masayoshi Tomizuka and Wei Zhan},
  journal= {arXiv preprint arXiv:2310.03023},
  year   = {2023}
}