用不确定性感知模型分类烟雾密度,提升火灾应急响应可靠性。
Uncertainty-Aware Wildfire Smoke Density Classification from Satellite Imagery via CBAM-Augmented EfficientNet with Evidential Deep Learning

- 结合CBAM注意力与证据深度学习,单次前向传播输出烟雾等级与置信度。
- 在1.6万张卫星图上达93.8%加权准确率,50%高置信预测准确率达96.7%。
- 可识别模糊边界区域,适合空气质量评估与灾害监测场景。
从卫星图像中快速准确评估野火烟雾严重程度对应急响应、空气质量建模和健康风险管控至关重要。现有深度学习方法将烟雾检测视为二分类任务,仅输出点估计且无置信度衡量。本文提出一种概率框架,将卫星图像块分为轻度、中度、重度三类,并在单次前向传播中分解输出认知不确定性(空缺度)与随机不确定性(不一致度)。模型采用预训练EfficientNet-B3作为主干网络,集成CBAM模块与证据深度学习头,通过狄利克雷浓度参数直接估计不确定性,无需蒙特卡洛采样。在来自野火检测数据集的16,298张真实卫星图像上评估,模型达到93.8%加权测试准确率(未加权为91.1%),ECE=0.0274。保留最确定的50%样本进行选择性预测时,准确率达96.7%。随着图像质量下降,不确定性单调上升,空缺度可作为有效图像质量指标。中度烟雾类别的认知不确定性最高(平均空缺度=0.187),验证模型正确识别了模糊的烟雾边界区域。CBAM空间注意力图聚焦于结构特征明显的场景区域,t-SNE显示轻度与重度烟雾聚类分离清晰。
原文摘要 · Abstract (English)
Rapid and accurate wildfire smoke severity assessment from satellite images is essential for emergency response, air quality modeling, and human health risk management. Existing deep learning approaches treat smoke detection as a binary task, producing point estimates without any measure of prediction confidence. We propose a probabilistic framework to categorize a satellite patch into Light, Moderate, and Heavy severity classes and to provide decomposed epistemic and aleatoric uncertainty in a single forward pass. Our architecture uses the backbone of a pre-trained EfficientNet-B3 and a CBAM module with an evidential deep learning head that predicts Dirichlet concentration parameters, directly estimating vacuity (epistemic) and dissonance (aleatoric) without Monte Carlo sampling. Evaluated on 16,298 real satellite patches derived from the Wildfire Detection dataset, our model achieves 93.8% weighted test accuracy (91.1% unweighted) with ECE=0.0274. Selective prediction retaining the most certain 50% of patches achieves 96.7% accuracy. As image quality degrades, uncertainty increases monotonically, and vacuity is a practical scan quality measure. The Moderate class represents transitional smoke conditions that exhibit the highest epistemic uncertainty (mean vacuity = 0.187), confirming the model correctly identifies ambiguous smoke boundary regions. CBAM spatial attention maps localize to structurally distinctive scene regions, and t-SNE demonstrates the clear cluster separation of Light and Heavy smoke.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。