arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.07421cs.LG
收藏

WILDS:野外分布偏移的基准

WILDS: A Benchmark of in-the-Wild Distribution Shifts

Microsoft(微软)
Pang Wei Koh, Shiori Sagawa, Henrik Marklund, Sang Michael Xie, Marvin Zhang, Akshay Balsubramani, Weihua Hu, Michihiro Yasunaga, Richard Lanas Phillips, Irena … 展开作者

Pang Wei Koh, Shiori Sagawa, Henrik Marklund, Sang Michael Xie, Marvin Zhang, Akshay Balsubramani, Weihua Hu, Michihiro Yasunaga, Richard Lanas Phillips, Irena Gao, Tony Lee, Etienne David, Ian Stavness, Wei Guo, Berton A. Earnshaw, Imran S. Haque, Sara Beery, Jure Leskovec, Anshul Kundaje, Emma Pierson, Sergey Levine, Chelsea Finn, Percy Liang

摘要:

分布偏移--训练分布与测试分布不同--会大大降低部署在野外的机器学习(ML)系统的准确性。尽管它们在现实世界的部署中无处不在,但这些分布偏移在当今ML社区广泛使用的数据集中表现不足。为了解决这一差距,我们提出了WILDS,这是一个由10个数据集组成的策划基准,反映了现实世界应用中自然出现的各种分布偏移,例如肿瘤识别的医院偏移;野生动物监测的相机陷阱;以及卫星成像和贫困地图的时间和位置。在每个数据集上,我们表明标准训练产生的分布外性能大大低于分布内性能。这种差距即使在通过现有方法训练的用于解决分布偏移的模型中仍然存在,这强调了对训练模型的新方法的需求,这些方法对实践中出现的分布偏移的类型更鲁棒。为了促进方法开发,我们提供了一个开源包,它可以自动加载数据集,包含默认的模型架构和超参数,并简化评估。代码和排行榜可在https://wilds.stanford.edu上获得。

英文摘要:

Distribution shifts -- where the training distribution differs from the test distribution -- can substantially degrade the accuracy of machine learning (ML) systems deployed in the wild. Despite their ubiquity in the real-world deployments, these distribution shifts are under-represented in the datasets widely used in the ML community today. To address this gap, we present WILDS, a curated benchmark of 10 datasets reflecting a diverse range of distribution shifts that naturally arise in real-world applications, such as shifts across hospitals for tumor identification; across camera traps for wildlife monitoring; and across time and location in satellite imaging and poverty mapping. On each dataset, we show that standard training yields substantially lower out-of-distribution than in-distribution performance. This gap remains even with models trained by existing methods for tackling distribution shifts, underscoring the need for new methods for training models that are more robust to the types of distribution shifts that arise in practice. To facilitate method development, we provide an open-source package that automates dataset loading, contains default model architectures and hyperparameters, and standardizes evaluations. Code and leaderboards are available at https://wilds.stanford.edu.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑