arXiv:2504.16277cs.LGcs.AI2025-04被引 8

针对特定部署场景,从数据集中挑选最优子集以提升模型表现。

DataS^3: Dataset Subset Selection for Specialization

  • 根据部署数据分布,从通用训练集筛选专用子集。
  • 专家手动筛选的子集性能比全量训练高51.3%。
  • 适用于医疗影像、野外监测等真实场景的模型优化。

在诸多实际机器学习应用中(如X光片骨折检测、相机陷阱物种识别),模型需在特定部署环境(如某医院、某国家公园)上表现良好,而非泛化于整个领域。然而,这些部署常面临数据分布不均衡、独特的问题。训练数据与部署数据分布差异会导致性能下降,因此需从现有训练数据中选择适配部署的子集。本文提出数据集子集选择专化问题(DS3):给定来自通用分布的训练集和来自目标部署分布的查询集(可能无标签),目标是选出能最大化部署性能的训练子集。我们构建了DataS^3——首个专为DS3问题设计的数据集与基准测试平台,涵盖多个真实应用领域,每个领域包含多个待专化的部署场景。通过全面评估来自核心集、数据过滤与数据编排等多个算法家族的方法,发现通用分布方法在部署任务上表现持续不佳。此外,我们验证了人工精心筛选的专家子集可超越全量训练,最高实现51.3%的准确率提升。该基准强调了定制化数据编排对提升部署性能与训练效率的关键作用,尤其在公共数据集日益普及、模型广泛落地的背景下,其重要性将愈发凸显。

原文摘要 · Abstract (English)

In many real-world machine learning (ML) applications (e.g. detecting broken bones in x-ray images, detecting species in camera traps), in practice models need to perform well on specific deployments (e.g. a specific hospital, a specific national park) rather than the domain broadly. However, deployments often have imbalanced, unique data distributions. Discrepancy between the training distribution and the deployment distribution can lead to suboptimal performance, highlighting the need to select deployment-specialized subsets from the available training data. We formalize dataset subset selection for specialization (DS3): given a training set drawn from a general distribution and a (potentially unlabeled) query set drawn from the desired deployment-specific distribution, the goal is to select a subset of the training data that optimizes deployment performance. We introduce DataS^3; the first dataset and benchmark designed specifically for the DS3 problem. DataS^3 encompasses diverse real-world application domains, each with a set of distinct deployments to specialize in. We conduct a comprehensive study evaluating algorithms from various families--including coresets, data filtering, and data curation--on DataS^3, and find that general-distribution methods consistently fail on deployment-specific tasks. Additionally, we demonstrate the existence of manually curated (deployment-specific) expert subsets that outperform training on all available data with accuracy gains up to 51.3 percent. Our benchmark highlights the critical role of tailored dataset curation in enhancing performance and training efficiency on deployment-specific distributions, which we posit will only become more important as global, public datasets become available across domains and ML models are deployed in the real world.

数据筛选模型部署性能提升

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