Adaptive Hyperbolic Kernels: Modulated Embedding in de Branges-Rovnyak Spaces
Leping Si, Meimei Yang, Hui Xue, Shipeng Zhu, Pengfei Fang
TL;DR
This work tackles distortion and rigidity in hyperbolic kernel methods by introducing a curvature-aware de Branges-Rovnyak RKHS that is isometric to the Poincaré ball $\mathbb{D}^n(c)$ and supports an adjustable multiplier to select the appropriate RKHS for any curvature $-c$. It develops a family of adaptive hyperbolic kernels, including the novel adaptive hyperbolic radial kernel (AHRad), designed to modulate hyperbolic features through learnable coefficients $\alpha_l$ and a base cosine-similarity kernel in the de Branges-Rovnyak space. The authors provide rigorous PD guarantees via the multiplier space and demonstrate empirical gains across few-shot, zero-shot, and semantic textual similarity tasks in vision and language. The approach yields lower-distortion embeddings and task-aware representations, offering practical improvements for hierarchical data modeling in cross-domain ML applications.
Abstract
Hierarchical data pervades diverse machine learning applications, including natural language processing, computer vision, and social network analysis. Hyperbolic space, characterized by its negative curvature, has demonstrated strong potential in such tasks due to its capacity to embed hierarchical structures with minimal distortion. Previous evidence indicates that the hyperbolic representation capacity can be further enhanced through kernel methods. However, existing hyperbolic kernels still suffer from mild geometric distortion or lack adaptability. This paper addresses these issues by introducing a curvature-aware de Branges-Rovnyak space, a reproducing kernel Hilbert space (RKHS) that is isometric to a Poincare ball. We design an adjustable multiplier to select the appropriate RKHS corresponding to the hyperbolic space with any curvature adaptively. Building on this foundation, we further construct a family of adaptive hyperbolic kernels, including the novel adaptive hyperbolic radial kernel, whose learnable parameters modulate hyperbolic features in a task-aware manner. Extensive experiments on visual and language benchmarks demonstrate that our proposed kernels outperform existing hyperbolic kernels in modeling hierarchical dependencies.
