/Text-to-concept

Primary LanguageJupyter NotebookOtherNOASSERTION

Text-To-Concept (and Back) via Cross-Model Alignment

This repository provides the PyTorch implementation of the following paper:

Paper

Text-To-Concept (and Back) via Cross-Model Alignment

Mazda Moayeri*, Keivan Rezaei*, Maziar Sanjabi, and Soheil Feizi

International Conference on Machine Learning (ICML), 2023.
open access: arXiv

Please see guideline.md for technical details. two examples of using TextToConcept framework can be found in example_search_concept_logic notebook, example_zeroshot notebook, and example drift notebook.

Citation

If you find our work useful, please cite:

@article{moayeri2023text,
  title={Text-To-Concept (and Back) via Cross-Model Alignment},
  author={Moayeri, Mazda and Rezaei, Keivan and Sanjabi, Maziar and Feizi, Soheil},
  journal={arXiv preprint arXiv:2305.06386},
  year={2023}
}