arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2412.06451cs.LGcs.AIeess.IV
收藏

不确定性估计的确定性有多高?用于基准测试机器学习中不确定性量化的三个新型地球观测数据集

How Certain are Uncertainty Estimates? Three Novel Earth Observation Datasets for Benchmarking Uncertainty Quantification in Machine Learning

Technical University of Munich(慕尼黑工业大学);National University of Sciences & Technology Pakistan(巴基斯坦国立科技大学);German Aerospace Center(德国航空航天中心);University of Bristol(布里斯托大学);Munich Center for Machine Learning(慕尼黑机器学习中心)

Yuanyuan Wang, Qian Song, Dawood Wasif, Muhammad Shahzad, Christoph Koller, Jonathan Bamber, Xiao Xiang Zhu

AI总结:

针对地球观测领域机器学习不确定性量化缺乏真值基准的问题,本文构建了覆盖回归、分割、分类三类任务的三个专用基准数据集,提供基线结果并开放资源,助力UQ方法的评估与模型质量提升。

摘要:

不确定性量化(UQ)对于评估地球观测(EO)产品的可靠性至关重要。然而,机器学习模型在EO中的广泛应用带来了额外的复杂性,因为这些模型本身就具有内在不确定性。尽管目前已有多种针对机器学习模型的UQ方法,但它们在EO数据集上的性能仍 largely 未得到评估。该领域的一个关键挑战是,除了图像/信号的标签外,缺乏不确定性的真值基准,即不确定性估计本身的确定性程度。本文通过引入三个专门为EO机器学习模型中的UQ设计的基准数据集填补了这一空白。这些数据集对应EO中的三类常见问题类型:回归、图像分割和场景分类,能够对EO机器学习模型的不同UQ方法进行透明的比较。我们描述了每个数据集的构建过程和特征,包括数据来源、预处理步骤和标签生成,其中特别关注参考不确定性的计算方式。我们还展示了多个机器学习模型在每个数据集上的基线性能,凸显了这些基准在模型开发和比较中的实用价值。总体而言,本文为从事EO人工智能领域的研究人员和从业者提供了宝贵资源,有助于推动更准确、更可靠的机器学习模型输出质量评估方法的发展。数据集和代码可通过https://gitlab.lrz.de/ai4eo/WG_Uncertainty获取。

英文摘要:

Uncertainty quantification (UQ) is essential for assessing the reliability of Earth observation (EO) products. However, the extensive use of machine learning models in EO introduces an additional layer of complexity, as those models themselves are inherently uncertain. While various UQ methods do exist for machine learning models, their performance on EO datasets remains largely unevaluated. A key challenge in the community is the absence of the ground truth for uncertainty, i.e. how certain the uncertainty estimates are, apart from the labels for the image/signal. This article fills this gap by introducing three benchmark datasets specifically designed for UQ in EO machine learning models. These datasets address three common problem types in EO: regression, image segmentation, and scene classification. They enable a transparent comparison of different UQ methods for EO machine learning models. We describe the creation and characteristics of each dataset, including data sources, preprocessing steps, and label generation, with a particular focus on calculating the reference uncertainty. We also showcase baseline performance of several machine learning models on each dataset, highlighting the utility of these benchmarks for model development and comparison. Overall, this article offers a valuable resource for researchers and practitioners working in artificial intelligence for EO, promoting a more accurate and reliable quality measure of the outputs of machine learning models. The dataset and code are accessible via https://gitlab.lrz.de/ai4eo/WG_Uncertainty.

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

↑