arXiv:2411.01099cs.CV2024-11ICLR被引 3

构建少类别图像分类基准,解决模型在小类场景下的评估难题

Few-Class Arena: A Benchmark for Efficient Selection of Vision Models and Dataset Difficulty Measurement

  • 提出统一基准FCA,专注2-10类的高效模型评估
  • 发现小类场景下模型性能与大类数据集表现差异显著
  • 提供类别相似度度量,助力模型与数据集快速筛选

我们提出Few-Class Arena(FCA),一个聚焦于少类别图像分类任务的统一基准。现有大量类别的基准数据集(80-1000类)广泛用于视觉模型架构演化评估,但真实应用中常仅涉及2-10个类别,导致基于多类数据训练的模型难以预测其在少类场景的表现。当前缺乏对少类别范式系统的评估工具。我们系统评估了在ImageNet子集(2-1000类)上训练的ResNet家族,并在十个数据集上测试了多种卷积神经网络与Transformer架构。为辅助数据集难度预判与模型高效选择,引入基于类别相似度的难度度量。FCA不仅提供新类别相似性方法、轻量模型设计支持,还推动新的缩放定律研究。该基准易于使用且可扩展,已开源至https://github.com/bryanbocao/fca。

原文摘要 · Abstract (English)

We propose Few-Class Arena (FCA), as a unified benchmark with focus on testing efficient image classification models for few classes. A wide variety of benchmark datasets with many classes (80-1000) have been created to assist Computer Vision architectural evolution. An increasing number of vision models are evaluated with these many-class datasets. However, real-world applications often involve substantially fewer classes of interest (2-10). This gap between many and few classes makes it difficult to predict performance of the few-class applications using models trained on the available many-class datasets. To date, little has been offered to evaluate models in this Few-Class Regime. We conduct a systematic evaluation of the ResNet family trained on ImageNet subsets from 2 to 1000 classes, and test a wide spectrum of Convolutional Neural Networks and Transformer architectures over ten datasets by using our newly proposed FCA tool. Furthermore, to aid an up-front assessment of dataset difficulty and a more efficient selection of models, we incorporate a difficulty measure as a function of class similarity. FCA offers a new tool for efficient machine learning in the Few-Class Regime, with goals ranging from a new efficient class similarity proposal, to lightweight model architecture design, to a new scaling law. FCA is user-friendly and can be easily extended to new models and datasets, facilitating future research work. Our benchmark is available at https://github.com/bryanbocao/fca.

少类别模型评估基准测试

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。