Get our free extension to see links to code for papers anywhere online!

 Add to Chrome

 Add to Firefox

CatalyzeX Code Finder - Browser extension linking code for ML papers across the web! | Product Hunt Embed

Refactoring Policy for Compositional Generalizability using Self-Supervised Object Proposals

Oct 26, 2020
Tongzhou Mu, Jiayuan Gu, Zhiwei Jia, Hao Tang, Hao Su



We study how to learn a policy with compositional generalizability. We propose a two-stage framework, which refactorizes a high-reward teacher policy into a generalizable student policy with strong inductive bias. Particularly, we implement an object-centric GNN-based student policy, whose input objects are learned from images through self-supervised learning. Empirically, we evaluate our approach on four difficult tasks that require compositional generalizability, and achieve superior performance compared to baselines.

* 34th Conference on Neural Information Processing Systems (NeurIPS 2020), Vancouver, Canada 


Share this with someone who'll enjoy it:

   Access Paper Source



Share this with someone who'll enjoy it: