FireViT: An Adaptive Lightweight Backbone Network for Fire Detection

• Published in: Forests 2023-11, Vol.14 (11), p.2158
• Main Authors: Shen, Pengfei, Sun, Ning, Hu, Kai, Ye, Xiaoling, Wang, Pingping, Xia, Qingfeng, Wei, Chen
• Format: Article
• Language: English
• Subjects: Algorithms; Artificial neural networks; Carbon monoxide; CNN and transformer; Computer networks; Datasets; Deep learning; Deformation; Fire detection; Fire detectors; Fire fighting; Floating point arithmetic; Forest & brush fires; Formability; Lightweight; Localization; Modules; Natural lighting; Neural networks; Smoke; Smoke detectors; Target detection; Technology application; Transformers; Wildfires
• ISSN: 1999-4907; 1999-4907
• DOI: 10.3390/f14112158

Fire incidents pose a significant threat to human life and property security. Accurate fire detection plays a crucial role in promptly responding to fire outbreaks and ensuring the smooth execution of subsequent firefighting efforts. Fixed-size convolutions struggle to capture the irregular variations in smoke and flames that occur during fire incidents. In this paper, we introduce FireViT, an adaptive lightweight backbone network that combines a convolutional neural network (CNN) and transformer for fire detection. The FireViT we propose is an improved backbone network based on MobileViT. We name the lightweight module that combines deformable convolution with a transformer as th DeformViT block and compare multiple builds of this module. We introduce deformable convolution in order to better adapt to the irregularly varying smoke and flame in fire scenarios. In addition, we introduce an improved adaptive GELU activation function, AdaptGELU, to further enhance the performance of the network model. FireViT is compared with mainstream lightweight backbone networks in fire detection experiments on our self-made labeled fire natural light dataset and fire infrared dataset, and the experimental results show the advantages of FireViT as a backbone network for fire detection. On the fire natural light dataset, FireViT outperforms the PP-LCNet lightweight network backbone for fire target detection, with a 1.85% increase in mean Average Precision (mAP) and a 0.9 M reduction in the number of parameters. Additionally, compared to the lightweight network backbone MobileViT-XS, which similarly combines a CNN and transformer, FireViT achieves a 1.2% higher mAP while reducing the Giga-Floating Point Operations (GFLOPs) by 1.3. FireViT additionally demonstrates strong detection performance on the fire infrared dataset.