ICML2026

UniFast-HGR: Scalable and Efficient Maximal Correlation for Multimodal Models

Hongkang Zhang, Shao-Lun Huang, Yanlong Wang, Ercan KURUOGLU

摘要

This paper presents UniFast-HGR, a scalable surrogate for Hirschfeld--Gebelein--Rényi (HGR) maximal correlation in high-dimensional multimodal learning. The method replaces explicit covariance whitening with centered and 2\ell_2-normalized cosine alignment, rewrites the covariance interaction in local-batch Gram space, and removes invariant diagonal self-correlation through Trivial Spectrum Suppression (TSS). This reformulation preserves the standardized dependence-maximization target of HGR-style learning while replacing the finite-sample whitening estimator with a differentiable local-batch objective whose dominant structural cost is O(m2K)O(m^2K) for local batch size mm and feature dimension KK. OptFast-HGR further reduces the practical memory burden by estimating the off-diagonal structural term through stochastic projection. Experiments across retrieval, image classification, remote sensing segmentation, and multimodal emotion recognition show consistent gains over covariance-based HGR/CCA variants and contrastive or neural MI-estimator objectives on strong multimodal backbones, while microbenchmarks confirm stable behavior at extreme feature dimensions.