Geometry-Aware Adaptation for Pretrained Models

Neural Information Processing Systems 

Machine learning models--including prominent zero-shot models--are often trained on datasets whose labels are only a small proportion of a larger label space. Such spaces are commonly equipped with a metric that relates the labels via distances between them.