arXiv:2409.01236cs.CV2024-09被引 15

为高光谱图像分类提供可信赖的预测置信度,提升关键场景下的模型安全性。

Spatial-Aware Conformal Prediction for Trustworthy Hyperspectral Image Classification

  • 引入空间感知的校准预测框架,融合像素间空间相关性增强置信度计算
  • 在95%置信水平下确保真实标签被包含,且预测集更紧凑高效
  • 理论证明有效,适合对可靠性要求高的遥感、环境监测等应用

高光谱图像(HSI)分类旨在为每个像素分配唯一标签以识别不同地物类别。尽管深度分类器在此领域已实现高预测准确率,但缺乏对预测置信度的严格量化。量化模型预测的确定性对预测模型的安全使用至关重要,而这一局限性限制了其在错误代价较高的关键场景中的应用。为支持HSI分类器的安全部署,我们首先提供了理论证明,确立了新兴不确定性量化技术——校准预测在HSI分类中的有效性。随后提出一种校准方法,可为任意训练好的HSI分类器生成可信的预测集合,确保这些集合以用户指定的概率(如95%)包含真实标签。在此基础上,我们提出空间感知校准预测(SACP),一种专为HSI数据设计的校准预测框架。该方法通过聚合具有高空间相关性的像素的非符合度得分,融入了HSI固有的空间信息,显著提升了预测集合的效率。理论与实证结果均验证了所提方法的有效性。源代码可在 https://github.com/J4ackLiu/SACP 获取。

原文摘要 · Abstract (English)

Hyperspectral image (HSI) classification involves assigning unique labels to each pixel to identify various land cover categories. While deep classifiers have achieved high predictive accuracy in this field, they lack the ability to rigorously quantify confidence in their predictions. Quantifying the certainty of model predictions is crucial for the safe usage of predictive models, and this limitation restricts their application in critical contexts where the cost of prediction errors is significant. To support the safe deployment of HSI classifiers, we first provide a theoretical proof establishing the validity of the emerging uncertainty quantification technique, conformal prediction, in the context of HSI classification. We then propose a conformal procedure that equips any trained HSI classifier with trustworthy prediction sets, ensuring that these sets include the true labels with a user-specified probability (e.g., 95\%). Building on this foundation, we introduce Spatial-Aware Conformal Prediction (\texttt{SACP}), a conformal prediction framework specifically designed for HSI data. This method integrates essential spatial information inherent in HSIs by aggregating the non-conformity scores of pixels with high spatial correlation, which effectively enhances the efficiency of prediction sets. Both theoretical and empirical results validate the effectiveness of our proposed approach. The source code is available at \url{https://github.com/J4ckLiu/SACP}.

高光谱图像校准预测不确定性量化空间建模

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