arXiv:2503.14150cs.CVeess.IV2025-03被引 15

对比四类模型预测野火蔓延,发现Transformer和UNet更准更可解释。

Comparative and Interpretative Analysis of CNN and Transformer Models in Predicting Wildfire Spread Using Remote Sensing Data

  • 用遥感数据比较Autoencoder、ResNet、UNet和Swin-UNet的预测能力。
  • Swin-UNet和UNet准确率更高,尤其在捕捉火势前兆特征上表现突出。
  • 结合XAI技术揭示关键影响因素,助力模型优化与实际应用决策。

面对全球野火威胁加剧,众多计算机视觉方法已应用于遥感数据分析。然而,因缺乏定量且可解释的对比分析,深度学习模型在野火预测中的选择仍不明确。本研究系统比较了四种主流架构:Autoencoder、ResNet、UNet及基于Transformer的Swin-UNet。利用包含近十年美国加州遥感数据的真实数据集,预测次日野火蔓延范围。定量分析表明,基于Transformer的Swin-UNet与UNet普遍优于Autoencoder和ResNet,主要得益于Transformer的注意力机制以及UNet和Swin-UNet中跳连结构对特征信息的高效保留,从而提升预测精度与可解释性。进一步采用XAI技术分析各模型,发现UNet与Swin-UNet能更聚焦于‘火势前驱图’、‘干旱’和‘植被’等关键特征,同时保持对其他区域的均衡关注,进而实现更优性能。研究为未来模型设计提供重要参考,并指导不同场景下的模型选型。

原文摘要 · Abstract (English)

Facing the escalating threat of global wildfires, numerous computer vision techniques using remote sensing data have been applied in this area. However, the selection of deep learning methods for wildfire prediction remains uncertain due to the lack of comparative analysis in a quantitative and explainable manner, crucial for improving prevention measures and refining models. This study aims to thoroughly compare the performance, efficiency, and explainability of four prevalent deep learning architectures: Autoencoder, ResNet, UNet, and Transformer-based Swin-UNet. Employing a real-world dataset that includes nearly a decade of remote sensing data from California, U.S., these models predict the spread of wildfires for the following day. Through detailed quantitative comparison analysis, we discovered that Transformer-based Swin-UNet and UNet generally outperform Autoencoder and ResNet, particularly due to the advanced attention mechanisms in Transformer-based Swin-UNet and the efficient use of skip connections in both UNet and Transformer-based Swin-UNet, which contribute to superior predictive accuracy and model interpretability. Then we applied XAI techniques on all four models, this not only enhances the clarity and trustworthiness of models but also promotes focused improvements in wildfire prediction capabilities. The XAI analysis reveals that UNet and Transformer-based Swin-UNet are able to focus on critical features such as 'Previous Fire Mask', 'Drought', and 'Vegetation' more effectively than the other two models, while also maintaining balanced attention to the remaining features, leading to their superior performance. The insights from our thorough comparative analysis offer substantial implications for future model design and also provide guidance for model selection in different scenarios.

野火预测Transformer可解释性遥感

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