为遥感机器学习不确定性评估提供三个新基准数据集。
How Certain are Uncertainty Estimates? Three Novel Earth Observation Datasets for Benchmarking Uncertainty Quantification in Machine Learning
- 构建三类遥感任务的不确定性基准数据集
- 引入真实不确定性参考值,支持方法对比
- 适合遥感AI研究者与模型评估人员使用
不确定性量化(UQ)对评估地球观测(EO)产品可靠性至关重要。然而,机器学习模型在EO中的广泛应用带来了额外复杂性,因为这些模型本身也存在不确定性。尽管已有多种UQ方法,但其在EO数据集上的表现仍缺乏系统评估。社区面临的核心挑战是缺乏不确定性的真实标签,即无法判断不确定性估计的可信程度。本文通过引入三个专为EO机器学习模型设计的基准数据集,填补这一空白。数据集涵盖回归、图像分割和场景分类三种常见任务类型,支持对不同UQ方法的透明比较。我们详细描述了每个数据集的来源、预处理流程及标签生成方式,特别聚焦于参考不确定性的计算过程。同时展示了多个机器学习模型在各数据集上的基线性能,凸显这些基准在模型开发与对比中的价值。整体而言,本研究为人工智能在地球观测领域的研究者与实践者提供了重要资源,有助于提升机器学习输出结果的质量评估精度。数据集与代码可通过 https://gitlab.lrz.de/ai4eo/WG_Uncertainty 获取。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。