无需源数据和标签,用极少参数高效适配大模型到新领域。
Unsupervised Parameter Efficient Source-free Post-pretraining
- 自监督训练+中心向量正则化,防止遗忘并节省一半计算。
- 仅优化极少量参数,8个目标领域表现稳定提升。
- 适合资源有限但需快速迁移的视觉模型部署场景。
随着视觉大模型参数量突破十亿级,将其适配至目标分布面临巨大算力与成本挑战。为此,我们提出UpStep——一种无监督、参数高效且无需源域数据的后预训练方法。首先设计自监督训练方案,在无源数据条件下对未标注目标域进行模型适配;为缓解灾难性遗忘问题,提出中心向量正则化(CVR),在50%训练迭代中跳过反向传播,显著降低计算开销;最后通过低秩适应方法实现参数高效更新,仅需优化极小比例参数。我们在多种基于ImageNet训练的骨干网络上验证该方法,成功将模型适配至八个不同目标领域,充分证明其泛化能力与适用性。
原文摘要 · Abstract (English)
Following the success in NLP, the best vision models are now in the billion parameter ranges. Adapting these large models to a target distribution has become computationally and economically prohibitive. Addressing this challenge, we introduce UpStep, an Unsupervised Parameter-efficient Source-free post-pretraining approach, designed to efficiently adapt a base model from a source domain to a target domain: i) we design a self-supervised training scheme to adapt a pretrained model on an unlabeled target domain in a setting where source domain data is unavailable. Such source-free setting comes with the risk of catastrophic forgetting, hence, ii) we propose center vector regularization (CVR), a set of auxiliary operations that minimize catastrophic forgetting and additionally reduces the computational cost by skipping backpropagation in 50\% of the training iterations. Finally iii) we perform this adaptation process in a parameter-efficient way by adapting the pretrained model through low-rank adaptation methods, resulting in a fraction of parameters to optimize. We utilize various general backbone architectures, both supervised and unsupervised, trained on Imagenet as our base model and adapt them to a diverse set of eight target domains demonstrating the adaptability and generalizability of our proposed approach.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。