An Effective Attention-based CNN Model for Fire Detection in Adverse Weather Conditions

被引:14
|
作者
Yar, Hikmat [1 ]
Ullah, Waseem [1 ]
Khan, Zulfiqar Ahmad [1 ]
Baik, Sung Wook [1 ]
机构
[1] Sejong Univ, IRIIT, Seoul 143747, South Korea
基金
新加坡国家研究基金会;
关键词
Attention mechanism; Convolutional neural network; Fire detection; Forest fire; Vehicle fire; Building fire; Uncertain environment; Remote sensing; CONVOLUTIONAL NEURAL-NETWORKS; FLAME DETECTION; SURVEILLANCE; CLASSIFICATION; COLOR;
D O I
10.1016/j.isprsjprs.2023.10.019
中图分类号
P9 [自然地理学];
学科分类号
0705 ; 070501 ;
摘要
Worldwide, fire disasters cause social, economic, and environmental damage, so their early detection and instant response is crucial for saving lives and property, which is possible via a soft computing approach. Researchers have therefore proposed different artificial intelligence-based approaches to detect fire at its early stages. However, several factors including domain shift problems, limited performance, and model complexity reduced their effectiveness in a real-world environment. To address these challenges, this study proposes an extensive data augmentation approach to tackle the domain shift problem by employing a mono-depth and atmospheric scattering model to generate and synthesise hazy, foggy, and night-time images. To further enhance the detection of small fire-affected areas, a refined DenseNet backbone is employed. This network is fine-tuned to extract intricate spatial details across different scales and establish meaningful connections between them. Subsequently, we introduce a soft attention mechanism into the backbone network. This entails utilising a 3D convolutional operation within the proposed attention framework to capture more extensive spatial features, which contributes to an enhanced discriminatory capability for detecting fire. Moreover, we have created a novel dataset explicitly designed for effective fire detection from a remote sensing perspective, known as the aerial drone and satellite fire (ADSF) dataset, encompasses images captured by drones and satellites. This broader range of image sources affords coverage across a wider expanse of geographical locations, surpassing the capabilities of CCTV cameras. Our proposed model achieves significant accuracy improvements, surpassing the state-of-the-art (SOTA) on both original and augmented datasets. For the Yar, FD, DFAN, and ADSF original datasets, we observed average accuracy increases of 1.25%, 1.95%, 2%, and 1.5%, respectively. In the comparisons with augmented versions of these datasets, our model consistently outperforms the SOTA, with accuracy improvements of 0.7, 1.97%, 1.42%, and 1.53%, respectively. Additionally, in our ablation studies, the proposed model excels in terms of both performance and resource efficiency. It achieves higher frame rates of 22.0, 72.5, and 7.0 FPS on CPU, GPU, and Raspberry Pi, respectively, with just 7.2 million learning parameters, making it well-suited for real-time deployment on edge devices. Finally, a Grad-CAM XAI method is utilised to visualise and highlight the most important regions in images emphasized by the model. The code and dataset are publicly available at https://gith ub.com/Hikmat-Yar/ISPRS-Fire-Detection.
引用
收藏
页码:335 / 346
页数:12
相关论文
共 50 条
  • [1] Pay "Attention" to Adverse Weather: Weather-aware Attention-based Object Detection
    Chaturvedi, Saket S.
    Zhang, Lan
    Yuan, Xiaoyong
    [J]. 2022 26TH INTERNATIONAL CONFERENCE ON PATTERN RECOGNITION (ICPR), 2022, : 4573 - 4579
  • [2] Intrusion Detection Using Attention-Based CNN-LSTM Model
    Al-Omar, Ban
    Trabelsi, Zouheir
    [J]. ARTIFICIAL INTELLIGENCE APPLICATIONS AND INNOVATIONS, AIAI 2023, PT I, 2023, 675 : 515 - 526
  • [3] A Large-Scale Database and a CNN Model for Attention-Based Glaucoma Detection
    Li, Liu
    Xu, Mai
    Liu, Hanruo
    Li, Yang
    Wang, Xiaofei
    Jiang, Lai
    Wang, Zulin
    Fan, Xiang
    Wang, Ningli
    [J]. IEEE TRANSACTIONS ON MEDICAL IMAGING, 2020, 39 (02) : 413 - 424
  • [4] An augmented attention-based lightweight CNN model for plant water stress detection
    Mohd Hider Kamarudin
    Zool Hilmi Ismail
    Noor Baity Saidi
    Kousuke Hanada
    [J]. Applied Intelligence, 2023, 53 : 20828 - 20843
  • [5] An augmented attention-based lightweight CNN model for plant water stress detection
    Kamarudin, Mohd Hider
    Ismail, Zool Hilmi
    Saidi, Noor Baity
    Hanada, Kousuke
    [J]. APPLIED INTELLIGENCE, 2023, 53 (18) : 20828 - 20843
  • [6] Attention-Based CNN Model for Burn Severity Assessment
    Rahman, Saeka
    Faezipour, Miad
    Ribeiro, Guilherme Aramizo
    Ridelman, Elika
    Klein, Justin D.
    Angst, Beth A.
    Shanti, Christina M.
    Rastgaar, Mo
    [J]. 2023 IEEE EMBS INTERNATIONAL CONFERENCE ON BIOMEDICAL AND HEALTH INFORMATICS, BHI, 2023,
  • [7] Efficient attention-based networks for fire and smoke detection
    Xiao, Bowei
    Yan, Chunman
    [J]. Journal of Electronic Imaging, 2024, 33 (05)
  • [8] KianNet: A Violence Detection Model Using an Attention-Based CNN-LSTM Structure
    Vosta, Soheil
    Yow, Kin-Choong
    [J]. IEEE ACCESS, 2024, 12 : 2198 - 2209
  • [9] An attention-based CNN-LSTM model for subjectivity detection in opinion-mining
    Sagnika, Santwana
    Mishra, Bhabani Shankar Prasad
    Meher, Saroj K.
    [J]. NEURAL COMPUTING & APPLICATIONS, 2021, 33 (24): : 17425 - 17438
  • [10] An attention-based CNN-BiLSTM model for depression detection on social media text
    Thekkekara, Joel Philip
    Yongchareon, Sira
    Liesaputra, Veronica
    [J]. EXPERT SYSTEMS WITH APPLICATIONS, 2024, 249