Geometry-Aware Adaptation for Pretrained Models
LOKI adapts pretrained machine learning models to reliably predict new classes when labels are sparse by exploiting the geometric relationships between existing class embeddings. It replaces standard prediction rules with a Fréchet mean estimator, allowing models to generalize better than traditional methods without needing extra training data for every new class.