arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.04492cs.CV
收藏

用于跨域小样本学习的判别性样本引导与参数高效特征空间自适应

Discriminative Sample-Guided and Parameter-Efficient Feature Space Adaptation for Cross-Domain Few-Shot Learning

University of Melbourne(墨尔本大学);Peter MacCallum Cancer Centre(彼得·麦卡勒姆癌症中心)

Rashindrie Perera, Saman Halgamuge

AI总结:

针对跨域小样本分类,本文提出基于预训练特征线性变换的参数高效自适应,并以判别性样本感知损失替代最近质心分类器,在 Meta-Dataset 上以约3倍参数效率取得新SOTA。

摘要:

本文研究跨域小样本分类,这一任务提出了一个具有挑战性的问题:利用少量带标注样本,在此前未见过的域中学习新类别。现有方法虽然具有一定效果,但面临若干局限;我们通过两项重要改进来缓解这些问题。首先,我们提出一种轻量级、参数高效的自适应策略,以解决在小型数据集上微调大量参数所带来的过拟合问题。该策略对预训练特征采用线性变换,显著减少可训练参数数量。其次,我们用一种判别性样本感知损失函数替代传统的最近质心分类器,增强模型对训练集中类间与类内方差的敏感性,从而改善特征空间中的聚类效果。在 Meta-Dataset 基准上的实证评估表明,我们的方法不仅在此前见过和未见过的数据集上分别将准确率最高提升 7.7% 和 5.3%,而且在取得上述性能的同时,其参数效率至少比现有方法高约 3 倍,从而确立了跨域小样本学习的新最优水平。我们的代码可在 https://github.com/rashindrie/DIPA 获取。

英文摘要:

In this paper, we look at cross-domain few-shot classification which presents the challenging task of learning new classes in previously unseen domains with few labelled examples. Existing methods, though somewhat effective, encounter several limitations, which we alleviate through two significant improvements. First, we introduce a lightweight parameter-efficient adaptation strategy to address overfitting associated with fine-tuning a large number of parameters on small datasets. This strategy employs a linear transformation of pre-trained features, significantly reducing the trainable parameter count. Second, we replace the traditional nearest centroid classifier with a discriminative sample-aware loss function, enhancing the model's sensitivity to the inter- and intra-class variances within the training set for improved clustering in feature space. Empirical evaluations on the Meta-Dataset benchmark showcase that our approach not only improves accuracy up to 7.7\% and 5.3\% on previously seen and unseen datasets, respectively, but also achieves the above performance while being at least $\sim3\times$ more parameter-efficient than existing methods, establishing a new state-of-the-art in cross-domain few-shot learning. Our code is available at https://github.com/rashindrie/DIPA.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑