ATT Squeeze U-Net: A Lightweight Network for Forest Fire Detection and Recognition

Jianmei Zhang,Hongqing Zhu,Pengyu Wang,Xiaofeng Ling

DOI: https://doi.org/10.1109/access.2021.3050628

IF: 3.9

2021-01-01

IEEE Access

Abstract:Forest fire is becoming one of the most significant natural disasters at the expense of ecology and economy. In this article, we develop an effective SqueezeNet based asymmetric encoder-decoder U-shape architecture, Attention U-Net and SqueezeNet (ATT Squeeze U-Net), mainly functions as an extractor and a discriminator of forest fire. This model takes attention mechanism to highlight useful features and suppress irrelevant contents by embedding Attention Gate (AG) units in the skip connection of U-shape structure. In this way, salient features are emphasized so that the proposed method could be competent at forest fire segmentation tasks with a small number of parameters. Specifically, we first replace classical convolution layer by a depthwise one and engage a Channel Shuffle operation as a feature communicator in the Fire module of classical SqueezeNet. Then, this modified SqueezeNet is employed as a substitution of the encoder of Attention U-Net and a corresponding DeFire module designed is combined into the decoder as well. Finally, to classify true fire, we take use of a fragment of the encoder in ATT Squeeze U-Net. The experimental results of modified SqueezeNet integrated Attention U-Net show that a competitive accuracy at 0.93 and an average prediction time at 0.89 second per image are achieved for reliable real-time forest fire detection.

computer science, information systems,telecommunications,engineering, electrical & electronic

What problem does this paper attempt to address?

The paper aims to address the problem of early detection and identification of forest fires. Specifically, the goal of the research is to develop an efficient and lightweight neural network architecture to achieve effective segmentation and classification of forest fires.

To achieve this goal, the authors propose a novel network architecture called ATT Squeeze U-Net. This architecture combines the advantages of SqueezeNet and Attention U-Net, with the main contributions including:

1. **Integration of Forest Fire Segmentation and Identification**: The proposed ATT Squeeze U-Net can simultaneously perform segmentation of forest fire areas and identification of fires.

2. **Significant Reduction in the Number of Parameters**: By using the modified SqueezeNet as the encoder part of the Attention U-Net, the number of parameters required for the entire model is greatly reduced.

3. **Enhanced Feature Learning Capability**: The introduction of Depthwise Convolution and Channel Shuffle operations in the Fire module of SqueezeNet improves the feature learning capability while reducing computational complexity.

4. **Application of Attention Mechanism**: By embedding Attention Gate (AG) in the skip connections of the U-Net structure, the model can highlight useful features and suppress irrelevant information.

Experimental results show that the proposed ATT Squeeze U-Net demonstrates high accuracy (0.93) in the forest fire segmentation task on public datasets, and the average prediction time per image is only 0.89 seconds, meeting the requirements for real-time forest fire detection.

In summary, the main objective of this research is to solve the problem of early detection and identification of forest fires by proposing a lightweight and efficient neural network architecture, and the effectiveness and practicality of the proposed method are demonstrated through experiments.