Comparative and Interpretative Analysis of CNN and Transformer Models in Predicting Wildfire Spread Using Remote Sensing Data
Not on the map. Why?
Abstract (arXiv)
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.
Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Builds on 15 of its 63 references that are in the map. First posted on arXiv 18 Mar 2025.
Builds on
15 of 63 references.
References · 15
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
- 2016Grad-CAM: Visual Explanations from Deep Networks via Gradient-based Localizationcited by 6,733
- 2017A Unified Approach to Interpreting Model Predictionscited by 6,555
- 2017Axiomatic Attribution for Deep Networkscited by 3,384
- 2021Swin-Unet: Unet-like Pure Transformer for Medical Image Segmentationcited by 1,222
- 2019Selective Kernel Networkscited by 403
- 2020A review of machine learning applications in wildfire science and managementcited by 65
- 2020Towards Interpretable Semantic Segmentation via Gradient-weighted Class Activation Mappingcited by 49
- 2023Slide-Transformer: Hierarchical Vision Transformer with Local Self-Attentioncited by 38
- 2023Finding the right XAI method -- A Guide for the Evaluation and Ranking of Explainable AI Methods in Climate Sciencecited by 27
Built on by
Citing papers per year: 2025: 2; 2026: 2.
4 papers.
- 2025From Black Box to Insight: Explainable AI for Extreme Event Preparednesscited by 3
- 2025Wildfire spread forecasting with Deep Learningcited by 1
- 2026Digital Twin and Agentic AI for Wild Fire Disaster Management: Intelligent Virtual Situation Roomcited by 0
- 2026Interpretable Patch-Based Deep Learning for Wildfire Spread Prediction from Ensemble Simulationscited by 0
- 2026Interpretable Patch-Based Deep Learning for Wildfire Spread Prediction from Ensemble Simulations16 Sep 2026
- 2026Digital Twin and Agentic AI for Wild Fire Disaster Management: Intelligent Virtual Situation Room9 Feb 2026
- 2025From Black Box to Insight: Explainable AI for Extreme Event Preparedness17 Nov 2025
- 2025Wildfire spread forecasting with Deep Learning23 May 2025