arXiv:2412.04077cs.CV2024-12CVPR被引 26

通过分解奇异值,只微调关键部分实现跨域泛化

SoMA: Singular Value Decomposed Minor Components Adaptation for Domain Generalizable Representation Learning

  • 基于奇异值分解,仅调整次要成分以保留通用特征
  • 在多个跨域任务上达到当前最优性能,且无额外计算开销
  • 适合需要高效微调的通用模型部署场景

领域泛化(DG)旨在利用一个或多个源域训练模型,以确保在未见目标域上的鲁棒表现。近年来,基础模型的参数高效微调(PEFT)在该问题中展现出良好效果。然而,现有方法仍难以平衡预训练模型的通用性与任务特异性特征的学习。为此,我们通过奇异值分解分析预训练权重,发现通用组件主要集中在主奇异分量中。据此提出奇异值分解次要成分适配(SoMA),仅选择性地微调次要奇异分量,其余部分保持冻结。同时冻结领域通用模块,并采用退火权重衰减策略,有效平衡了泛化能力与判别力。SoMA在涵盖领域泛化语义分割与目标检测的多个基准上取得当前最优结果。此外,该方法不引入推理额外开销或正则化损失,兼容任意主干网络和分类头,具有高度通用性,可轻松集成至多种任务。

原文摘要 · Abstract (English)

Domain generalization (DG) aims to adapt a model using one or multiple source domains to ensure robust performance in unseen target domains. Recently, Parameter-Efficient Fine-Tuning (PEFT) of foundation models has shown promising results in the context of DG problem. Nevertheless, existing PEFT methods still struggle to strike a balance between preserving generalizable components of the pre-trained model and learning task-specific features. To gain insights into the distribution of generalizable components, we begin by analyzing the pre-trained weights through the lens of singular value decomposition. Building on these insights, we introduce Singular Value Decomposed Minor Components Adaptation (SoMA), an approach that selectively tunes minor singular components while keeping the residual parts frozen. SoMA effectively retains the generalization ability of the pre-trained model while efficiently acquiring task-specific skills. Moreover, we freeze domain-generalizable blocks and employ an annealing weight decay strategy, thereby achieving an optimal balance in the delicate trade-off between generalizability and discriminability. SoMA attains state-of-the-art results on multiple benchmarks that span both domain generalized semantic segmentation to domain generalized object detection. In addition, our methods introduce no additional inference overhead or regularization loss, maintain compatibility with any backbone or head, and are designed to be versatile, allowing easy integration into a wide range of tasks.

领域泛化参数高效奇异值分解模型压缩

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。