利用跨域几何一致性校准视觉模型特征分布偏差
Calibrating Biased Distribution in VFM-derived Latent Space via Cross-Domain Geometric Consistency
- 基于跨域几何一致性,构建特征分布校准框架
- 联邦学习与长尾识别中性能显著提升
- 适合处理数据异构与样本不均衡场景
尽管深度学习发展迅速,但训练样本分布与真实分布之间仍存在显著差距,原因包括采样偏差、噪声等。在基础模型时代,我们发现使用现成的视觉基础模型(如CLIP、DINOv2)提取特征时,所得特征分布的几何形状在不同领域和数据集间具有显著可迁移性。为验证其实际价值,我们将该几何知识引导的分布校准框架应用于两个典型且具挑战性的场景:联邦学习与长尾识别。在联邦学习中,我们在隐私约束下获取全局几何形状,并据此生成客户端新样本,以弥合局部与全局观测之间的差距;在长尾学习中,利用样本丰富类别的几何知识,恢复样本稀缺尾部类别的真实分布。大量实验表明,所提出的几何知识引导分布校准方法能有效克服由数据异构性和样本不平衡导致的信息缺失,在多个基准上均取得性能提升。
原文摘要 · Abstract (English)
Despite the fast progress of deep learning, one standing challenge is the gap of the observed training samples and the underlying true distribution. There are multiple reasons for the causing of this gap e.g. sampling bias, noise etc. In the era of foundation models, we show that when leveraging the off-the-shelf (vision) foundation models (e.g., CLIP, DINOv2) for feature extraction, the geometric shapes of the resulting feature distributions exhibit remarkable transferability across domains and datasets. To verify its practical usefulness, we embody our geometric knowledge-guided distribution calibration framework in two popular and challenging settings: federated learning and long-tailed recognition. In the federated setting, we devise a technique of acquiring the global geometric shape under privacy constraints, then leverage this knowledge to generate new samples for clients, in the aim of bridging the gap between local and global observations. In long-tailed learning, it utilizes the geometric knowledge transferred from sample-rich categories to recover the true distribution for sample-scarce tail classes. Comprehensive experiments show that our proposed geometric knowledge-guided distribution calibration effectively overcomes information deficits caused by data heterogeneity and sample imbalance, with boosted performance across benchmarks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。