PGA-KD: Principal Geometry Alignment and Semantic Consistency for VLM2Vec Distillation

The-Anh Tran, Thanh Xuan Nguyen, Duc Anh Nguyen, Dinh Viet Sang, Linh Ngo Van, Thien Huu Nguyen


Abstract
Adapting large vision–language models for embedding tasks (VLM2Vec) is standard practice, but distilling them into compact architectures remains challenging due to capacity mismatches in tokenization, dimensionality, and multimodal representations. To address this, we propose a robust distillation framework based on two complementary mechanisms: Principal Geometry Alignment (PGA) to stabilize training by aligning principal geometric components, and a Mutual Information Maximization objective to enforce global semantic consistency across intra- and inter-modal pairs. By jointly constraining geometric structure and semantic dependency, our method effectively mitigates model capacity discrepancies. Experimental results on the Massive Multimodal Embedding Benchmark (MMEB) across all four meta-tasks—under both In-Distribution and Out-of-Distribution settings—demonstrate that our approach consistently outperforms state-of-the-art distillation methods. Our source code is available at https://github.com/theanh-ktmt/PGA-KD.
Anthology ID:
2026.emnlp-main.257
Volume:
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
EMNLP
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
5827–5847
Language:
URL:
https://aclanthology.org/2026.emnlp-main.257/
DOI:
Bibkey:
Cite (ACL):
The-Anh Tran, Thanh Xuan Nguyen, Duc Anh Nguyen, Dinh Viet Sang, Linh Ngo Van, and Thien Huu Nguyen. 2026. PGA-KD: Principal Geometry Alignment and Semantic Consistency for VLM2Vec Distillation. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 5827–5847, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
PGA-KD: Principal Geometry Alignment and Semantic Consistency for VLM2Vec Distillation (Tran et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.257.pdf