Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Rank-Aware Hyperbolic Alignment for Vision-Language Dataset Distillation

About

Vision-language dataset distillation (VLDD) compresses a large image-text paired dataset into a small set of synthetic pairs that can efficiently train contrastive vision-language models under strict data and compute budgets. Most existing methods match expert trajectories or cross-modal statistics, yet still enforce full-dimensional alignment in a Euclidean embedding space. This is often overly restrictive due to rank-deficient image--text correlation, with shared semantics concentrated in a low-dimensional range and remaining variation spread across a weakly correlated residual subspace. LoRS relaxes alignment at the similarity level by low-rank factorization, but does not explicitly control dominant alignment capacity and structure in the representation space. We thus propose a rank-aware hyperbolic alignment (RAHA) that combines hierarchical geometry with explicit alignment-capacity control. RAHA lifts multimodal representations to hyperbolic space and optimizes distilled pairs with asymmetric objectives that enforce geodesic alignment in the shared range while regularizing the residual subspace to preserve modality-private diversity and improve transfer robustness. Experiments on benchmarks show that RAHA demonstrates competitive cross-modal retrieval and improved transfer indicators under fixed budgets.

Jongoh Jeong, Sun-Kyung Lee, Kuk-Jin Yoon• 2026

Related benchmarks

TaskDatasetResultRank
Text-to-Image RetrievalFlickr30k (test)--
528
Image-to-Text RetrievalFlickr30k (test)--
472
Image ClassificationStanford Cars (test)--
320
Image ClassificationCUB-200-2011 (test)
Top-1 Acc14.43
316
Image-Text RetrievalFlickr30k (test)--
45
Image-Text RetrievalCOCO (test)--
41
Image-Text RetrievalFlickr8k (test)
Image Retrieval (IR)33.6
31
Text-to-Image RetrievalCOCO (test)--
28
Image-Text RetrievalCC3M-595K-LLaVA 1
IR Score7.6
6
Image ClassificationCIFAR-100 (test)
Top-1 Accuracy15.6
5
Showing 10 of 10 rows

Other info

GitHub

Follow for update