Wildfire Detection Using Vision Transformer with the Wildfire Dataset

TL;DR

This paper explores the use of Vision Transformers trained on a large wildfire image dataset to improve early wildfire detection, addressing challenges like data quality and environmental factors.

Contribution

It introduces a ViT-based approach trained on a comprehensive wildfire dataset, demonstrating its potential for accurate real-time wildfire detection.

Findings

High accuracy achieved in wildfire classification

Effective preprocessing pipeline for high-resolution images

Potential for real-time early wildfire detection

Abstract

The critical need for sophisticated detection techniques has been highlighted by the rising frequency and intensity of wildfires in the US, especially in California. In 2023, wildfires caused 130 deaths nationwide, the highest since 1990. In January 2025, Los Angeles wildfires which included the Palisades and Eaton fires burnt approximately 40,000 acres and 12,000 buildings, and caused loss of human lives. The devastation underscores the urgent need for effective detection and prevention strategies. Deep learning models, such as Vision Transformers (ViTs), can enhance early detection by processing complex image data with high accuracy. However, wildfire detection faces challenges, including the availability of high-quality, real-time data. Wildfires often occur in remote areas with limited sensor coverage, and environmental factors like smoke and cloud cover can hinder detection. Additionally, training deep learning models is computationally expensive, and issues like false positives/negatives and scaling remain concerns. Integrating detection systems with real-time alert mechanisms also poses difficulties. In this work, we used the wildfire dataset consisting of 10.74 GB high-resolution images categorized into 'fire' and 'nofire' classes is used for training the ViT model. To prepare the data, images are resized to 224 x 224 pixels, converted into tensor format, and normalized using ImageNet statistics.