用自适应注意力适配器,让大模型在小数据科学领域也能高效迁移。
Unleashing Foundation Vision Models: Adaptive Transfer for Diverse Data-Limited Scientific Domains
- 通过聚类中心与注意力机制,动态优化预训练特征。
- 在10个科学领域10个数据受限数据集上达最优表现。
- 兼容多种模型架构,适合科研中少样本视觉任务。
在大数据时代,计算机视觉得益于LAION-2B、LAION-400M、ImageNet-21K、Kinetics等大规模数据集,主流模型如ViT和ConvNeXt系列已具备丰富知识。然而,众多专业化且数据有限的科学下游任务仍面临严峻挑战。本文提出一种新型聚类注意力适配器(CLAdapter),通过引入注意力机制与聚类中心,利用分布相关性和变换矩阵对特征进行个性化增强,使微调后的模型能针对不同特征集学习差异化表示,从而实现从丰富预训练特征到多样下游场景的有效适配。该方法采用统一接口设计,可无缝集成于CNN与Transformer等多种模型架构,支持2D与3D场景。在涵盖通用、多媒体、生物、医学、工业、农业、环境、地理、材料科学、分布外(OOD)及3D分析等10个领域的数据集上,CLAdapter均取得领先性能,充分验证了其通过自适应迁移释放基础视觉模型潜力的有效性。代码已开源:https://github.com/qklee-lz/CLAdapter。
原文摘要 · Abstract (English)
In the big data era, the computer vision field benefits from large-scale datasets such as LAION-2B, LAION-400M, and ImageNet-21K, Kinetics, on which popular models like the ViT and ConvNeXt series have been pre-trained, acquiring substantial knowledge. However, numerous downstream tasks in specialized and data-limited scientific domains continue to pose significant challenges. In this paper, we propose a novel Cluster Attention Adapter (CLAdapter), which refines and adapts the rich representations learned from large-scale data to various data-limited downstream tasks. Specifically, CLAdapter introduces attention mechanisms and cluster centers to personalize the enhancement of transformed features through distribution correlation and transformation matrices. This enables models fine-tuned with CLAdapter to learn distinct representations tailored to different feature sets, facilitating the models' adaptation from rich pre-trained features to various downstream scenarios effectively. In addition, CLAdapter's unified interface design allows for seamless integration with multiple model architectures, including CNNs and Transformers, in both 2D and 3D contexts. Through extensive experiments on 10 datasets spanning domains such as generic, multimedia, biological, medical, industrial, agricultural, environmental, geographical, materials science, out-of-distribution (OOD), and 3D analysis, CLAdapter achieves state-of-the-art performance across diverse data-limited scientific domains, demonstrating its effectiveness in unleashing the potential of foundation vision models via adaptive transfer. Code is available at https://github.com/qklee-lz/CLAdapter.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。