3 个月前

少样本图像分类的通用表示变换层

少样本图像分类的通用表示变换层

摘要

少样本分类旨在仅凭借少量样本识别未见类别。本文研究多域少样本图像分类问题,其中未见类别及其样本来自多样化的数据源。该问题近年来受到广泛关注,并推动了如Meta-Dataset等基准数据集的建立。在多域设置中,一个关键挑战是如何有效融合来自不同训练域的特征表示。为此,我们提出一种通用表示Transformer(Universal Representation Transformer, URT)层,该层通过元学习机制,动态地重新加权并组合最适配的域特定特征表示,从而利用通用特征实现少样本分类。实验结果表明,URT在Meta-Dataset基准上取得了新的最先进性能,尤其在支持的数据源数量方面显著优于现有方法,展现出卓越的跨域泛化能力。我们进一步分析了URT的多种变体,并通过注意力得分热力图可视化,深入揭示了模型实现跨域泛化的工作机制。相关代码已开源,地址为:https://github.com/liulu112601/URT。

代码仓库

liulu112601/URT
官方
pytorch

基准测试

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
少样本图像分类的通用表示变换层 | 论文 | HyperAI超神经