预训练嵌入质量决定图像少样本分类性能上限
PictSure: Pretraining Embeddings Matters for In-Context Learning Image Classifiers
- 用轻量融合变换器实现无需微调的图像分类
- 预训练数据质量影响显著,融合层训练数据多样性作用有限
- 适合关注少样本学习与端到端图像推理的开发者
在数据稀缺领域构建图像分类模型仍具挑战,因大规模标注数据难以获取。上下文学习(ICL)为少样本图像分类(FSIC)提供了新思路,但现有研究低估了编码器预训练与融合层训练数据的重要性。我们提出PictSure,一种仅依赖视觉的ICL模型系列,验证了轻量融合变压器架构的可行性,以及更广泛图像领域中优质嵌入表示的必要性。在域内与域外评估中,预训练带来的表示质量与下游ICL性能强相关。关键发现是:无论融合层训练数据从ImageNet单一来源扩展至多领域混合,性能提升均有限,表明只要嵌入结构良好,融合层即可有效适配。结果表明,视觉ICL的瓶颈在于表示质量,而非融合模块的训练多样性。为促进复现与应用,我们开源全部模型权重,并提供MCP服务器,支持大模型代理系统直接调用PictSure完成少样本图像分类,无需额外集成开销。代码见https://github.com/PictSure,模型见https://huggingface.co/pictsure。
原文摘要 · Abstract (English)
Building image classification models remains cumbersome in data-scarce domains, where collecting large labeled datasets is impractical. In-context learning (ICL) is a promising paradigm for few-shot image classification (FSIC), but prior work has underexplored the relative importance of encoder pretraining versus fusion-layer training data. We present PictSure, a vision-only ICL family of models that demonstrates the potential of easy-to-use fusion transformer architectures, as well as the need for better embedding representations across a wider range of image domains. In both in-domain and out-of-domain evaluations, we find that representation quality induced by pretraining strongly correlates with downstream ICL performance. Crucially, varying the training dataset for the fusion transformer, from ImageNet alone to diverse multi-domain mixtures, provides limited additional performance gains under the evaluated settings, demonstrating that the fusion layer appears capable of adapting effectively once embeddings are sufficiently structured. These results show that the bottleneck in visual ICL is representation quality, not fusion-module training diversity. To facilitate adoption and reproducibility, we release all model weights as open-source artifacts and provide an MCP server that exposes PictSure as a callable tool for LLM-based agentic systems, enabling few-shot image classification to be invoked directly within AI pipelines without integration overhead. Code can be found at https://github.com/PictSure and models at https://huggingface.co/pictsure.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。