Learning Along the Arrow of Time: Hyperbolic Geometry for Backward-Compatible Representation Learning
Abstract
Backward compatible representation learning enables updated models to integrate seamlessly with existing ones, avoiding to reprocess stored data. Despite recent advances, existing compatibility approaches in Euclidean space neglect the uncertainty in the old embedding models and force the new model to replicate outdated representations regardless of their quality, and thereby hindering the learning process. In this paper, we switch perspectives to hyperbolic geometry, where we treat time as a natural axis for capturing a model’s confidence and evolution. By lifting embeddings into hyperbolic space and constraining updated embeddings to lie within the entailment cone of the old ones, we maintain generational consistency across models while accounting for uncertainties in the representations. To further enhance compatibility, we introduce a robust contrastive alignment loss that dynamically adjusts alignment weights based on the uncertainty of the old embeddings. Experiments validate the superiority of the proposed method in achieving compatibility, paving the way for more resilient and adaptable machine learning systems.
Cite
Text
Bui et al. "Learning Along the Arrow of Time: Hyperbolic Geometry for Backward-Compatible Representation Learning." Proceedings of the 42nd International Conference on Machine Learning, 2025.Markdown
[Bui et al. "Learning Along the Arrow of Time: Hyperbolic Geometry for Backward-Compatible Representation Learning." Proceedings of the 42nd International Conference on Machine Learning, 2025.](https://mlanthology.org/icml/2025/bui2025icml-learning/)BibTeX
@inproceedings{bui2025icml-learning,
title = {{Learning Along the Arrow of Time: Hyperbolic Geometry for Backward-Compatible Representation Learning}},
author = {Bui, Ngoc and Yang, Menglin and Chen, Runjin and Neves, Leonardo and Ju, Mingxuan and Ying, Zhitao and Shah, Neil and Zhao, Tong},
booktitle = {Proceedings of the 42nd International Conference on Machine Learning},
year = {2025},
pages = {5842-5855},
volume = {267},
url = {https://mlanthology.org/icml/2025/bui2025icml-learning/}
}