@inproceedings{fu2021learning,
  title={Learning Task Informed Abstractions},
  author={Fu, Xiang and Yang, Ge and Agrawal, Pulkit and Jaakkola, Tommi},
  booktitle={International Conference on Machine Learning},
  pages={3480--3491},
  year={2021},
  organization={PMLR}
}