PictSure:通过视觉嵌入功能挑战 _Few-Shot _分类的新方法

概述

近年来,从少量样本中识别新类别的 "少镜头图像分类"(FSIC)在图像分类领域备受关注。

尤其是上下文学习(ICL)方法,这种方法在测试时只提供几张有标签的图像,无需训练即可进行推理,在灵活性和效率方面很有前景。然而,以往的研究并未充分研究图像特征(嵌入向量)的预处理和学习方法对 ICL 分类准确性的影响。

本研究提出了一种仅使用视觉信息的 ICL 模型 PictSure,并系统分析了嵌入模型的结构、预训练方法和学习时机等因素对 FSIC 性能的影响。结果表明,PictSure 的性能明显优于传统方法,尤其是在使用预训练嵌入模型时。

这表明,不依赖语义语言信息,只依赖视觉特征的质量,也能实现高精度分类。

建议的方法

所提出的 "PictSure "是一种仅使用图像和基于变换器架构的 ICL 模型,用于少镜头分类,可从输入支持集(图像及其标签)和查询图像的上下文中预测查询标签。转换器的输入是一串结合了图像嵌入和标签嵌入的标记,而专门设计的非对称注意力掩码提供了一种查询取决于支持的结构。

该模型的另一个特点是对视觉嵌入质量的依赖性很强。ResNet 和 Vision Transformer (ViT) 被用作嵌入模型,在使用 ImageNet 或其他模型进行预训练后,它们或被固定或被微调。

特别是在 ViT 中,通过引入三重损失(Triplet Loss)和通常的分类损失(Classification Loss),可以获得结构更合理的嵌入空间。这使得视觉上相似的图像彼此接近,有助于稳定 ICL 中的标签预测。

PictSure 不仅适用于自然图像,还适用于语言信息贫乏的领域,如医学和农业。

实验

为了测试 PictSure 的有效性,我们使用域内(接近训练数据)和域外(不同分布)数据集进行了少量分类任务。实验在 5 路 1 次和 5 路 5 次设置下进行,并将 PictSure 的性能与基准 KNN 和基于 CLIP 的 ICL 方法 CAML 进行了比较。

结果表明,使用 ResNet 和 ViT 以及预训练和冻结嵌入模型的 PictSure 明显优于其他方法,尤其是在医疗和专业图像领域(如脑肿瘤、器官CMNIST)。

相比之下,使用 CLIP 的 CAML 在自然图像中取得了较高的准确率,而在医学领域的准确率却有所下降。这表明 CLIP 基于语言的预训练并不适合识别视觉细节。

此外,PictSure 对过度学习表现出强大而稳定的学习行为,随着支持集(上下文长度)数量的增加,准确率也在提高,而在 8 个支持集之后就会出现性能领先的情况。

相关推荐
AI浩2 小时前
SemOD:基于语义增强的多天气条件目标检测网络
网络·人工智能·目标检测
老兵发新帖2 小时前
AI驱动架构设计开源项目分析:next-ai-drawio
人工智能·开源·draw.io
rundreamsFly2 小时前
【云馨AI】基于 AI 的 COSMIC智能文档工具 第二代功能点评估:从效率到精准度的全面升级
人工智能·cosmic编写·cosmic
北京耐用通信2 小时前
调试复杂、适配难?耐达讯自动化Ethercat转Devicenet让继电器通讯少走弯路
人工智能·物联网·网络协议·自动化·信息与通信
人工智能教学实践2 小时前
AI大模型时代,计算机视觉课程如何™教”与“学”?
人工智能·计算机视觉
智源研究院2 小时前
智源开源 Reason-RFT:用强化学习重塑视觉推理,突破 VLM 泛化瓶颈
人工智能
DFT计算杂谈2 小时前
ABINIT能带计算数据处理脚本
数据库·人工智能
BioRunYiXue2 小时前
双荧光素酶报告基因实验
java·运维·服务器·数据库·人工智能·数据挖掘·eclipse
数据皮皮侠2 小时前
政府创新采购数据库(2016-2024)
大数据·数据库·人工智能·制造·微信开放平台