Computer Science & Technology

Forest Fire Recognition by Improved EfficientNet-E Model Based on ECA Attention Mechanism

Expand
  • School of Economics and Management,Beijing Forestry University,Beijing 100083,China
周浪(1992-),男,博士生,主要从事人工智能、大数据、智能优化算法、系统优化等研究。E-mail:zhoulang@bjfu.edu.cn

Received date: 2023-02-14

  Online published: 2023-06-20

Supported by

the National Natural Science Foundation of China(71901027);the Humanities and Social Sciences Foundation of the Ministry of Education of China(21YJA630012)

Abstract

The occurrence of fires has brought huge losses to society. The task of forest fire prevention and control is becoming increasingly urgent, and global warming has made this problem more complicated. Deep learning plays an important role in all walks of life. A large number of models are constantly designed and proposed, and there are various ways to improve the models. Therefore, this article proposed the EfficientNet-E model, which uses the more advanced ECA module (Efficient Channel Attention module) to replace the SE module in the EfficientNet. It improves the performance of the model by enhancing the performance of the attention mechanism. Compared with the SE module, the ECA module better retains the information during transmission, allowing the data features to be more fully retained during the transmission process, thus enabling the model to be optimized. To verify the performance of the EfficientNet-E model and the advantages of EfficientNet’s design idea in forest fire identification compared with traditional models, this article selected representatives of classic models, ResNet and DenseNet, as comparison references, and conducted related experiments in combination with EfficientNet and EfficientNet-E.The experiment selected 3 303 forest fire, non-fire and smoke pictures.The results of multiple rounds of tests show that EfficientNet-E is better than the conventional classic deep learning model in identifying forest fire data, and compared with the original EfficientNet’s average accuracy of 89.28%, EfficientNet-E’s average accuracy (90.04%) is obviously improved. The standard deviation is smaller and the training stability is better, which confirms the improved EfficientNet-E’s better performance.

Cite this article

ZHOU Lang, FAN Kun, QU Hua, et al . Forest Fire Recognition by Improved EfficientNet-E Model Based on ECA Attention Mechanism[J]. Journal of South China University of Technology(Natural Science), 2024 , 52(2) : 42 -49 . DOI: 10.12141/j.issn.1000-565X.230046

References

1 赵鹏武,武峻毅,张恒 .基于聚类分析法的我国森林火险等级区划研究[J].林业工程学报20216(3):142-148.
  ZHAO Pengwu, WU Junyi, ZHANG Heng .Study on forest fire risk classification in China using cluster analysis[J].Journal of Forestry Engineering20216(3):142-148.
2 祖鑫萍,李丹 .基于无人机图像和改进YOLOv3-SPP算法的森林火灾烟雾识别方法[J].林业工程学报20227(5):142-149.
  ZU Xinping, LI Dan .Research on forest fire smoke identification method based on UAV images and improved YOLOv3-SPP algorithm[J].Journal of Forestry Engineering20227(5):142-149.
3 杜嘉欣,常青,刘鑫 .面向森林火灾烟雾识别的深度信念卷积网络[J].现代电子技术202043(13):44-48.
  DU Jiaxin, CHANG Qing, LIU Xin .DBN-CNN for forest fire smoke recognition[J].Modern Electronics Technique202043(13):44-48.
4 李珍辉,鲁静文,陈镜伊,等 .基于InceptionV3卷积神经网络森林火灾检测方法[J].湖南工程学院学报(自然科学版)202131(4):44-49.
  LI Zhen-hui, LU Jing-wen, CHEN Jing-yi,et al .Forest fire detection method based on inception V3 convolutional neural network[J].Journal of Hunan University of Engineering (Natural Science Edition)202131(4):44-49.
5 李梁,董旭彬,赵清华 .改进Mask R-CNN在航拍灾害检测的应用研究[J].计算机工程与应用201955(21):167-176.
  LI Liang, DONG Xubin, ZHAO Qinghua .Application research of improved Mask R-CNN in aerial disaster detection[J].Computer Engineering and Applications201955(21):167-176.
6 周浪,樊坤,瞿华,等 .基于Sparse-DenseNet模型的森林火灾识别研究[J].北京林业大学学报202042(10):36-44.
  ZHOU Lang, FAN Kun, QU Hua,et al .Forest fire identification based on Sparse-DenseNet model[J].Journal of Beijing Forestry University202042(10):36-44.
7 李巨虎,范睿先,陈志泊 .基于颜色和纹理特征的森林火灾图像识别[J].华南理工大学学报(自然科学版)202048(1):70-83.
  LI Juhu, FAN Ruixian, CHEN Zhibo .Forest fire recognition based on color and texture features[J].Journal of South China University of Technology (Natural Science Edition)202048(1):70-83.
8 李惠鹏,李长勇,李贵宾,等 .基于深度学习的多品种鲜食葡萄采摘点定位[J].中国农机化学报202243(12):155-161.
  LI Huipeng, LI Changyong, LI Guibin,et al .Picking point positioning of multi-variety table grapes based on deep-learning[J].Journal of Chinese Agricultural Mechanization202243(12):155-161.
9 吴天赐,郁佳鑫,黄晓峰,等 .深度学习图片分类模型ResNet-18用于判定口腔鳞状细胞癌浸润方式的初步研究[J].口腔医学研究202339(10):917-922.
  WU Tianci, YU Jiaxin, HUANG Xiaofeng,et al .Preliminary study on deep learning picture classification model for identification and classification of invasion pattern of oral squamous cell carcinoma[J].Journal of Stomatology202339(10):917-922.
10 苏令涛,李瑞泽,张功磊,等 .基于深度学习的农作物病虫害识别研究[J].数学建模及其应用202211(4):1-12.
  SU Lingtao, LI Ruize, ZHANG Gonglei,et al .Research on deep learning based crop pest and disease identification[J].Mathematical Modeling and Its Application202211(4):1-12.
11 GOUR M, JAIN S, KUMAR T S .Residual learning based CNN for breast cancer histopathological image classification[J].International Journal of Imaging Systems and Technology202030(3):621-635.
12 WEINAN E, MA C, WU L .A comparative analysis of optimization and generalization properties of two-layer neural network and random feature models under gradient descent dynamics[J].Science China Mathematics202063(7):1235-1265.
13 TAN M, LE Q V .EfficientNet:rethinking model scaling for convolutional neural networks[C]∥Proceedings of the 36th International Conference on Machine Learning.Long Beach:ACM,2019:6105-6114.
14 尹梓睿,张索非,张磊,等 .适于行人重识别的二分支EfficientNet网络设计[J].信号处理202036(9):1481-1488.
  YIN Zirui, ZHANG Suofei, ZHANG Lei,et al .Design of a two-branch EfficientNet for person re-identification[J].Signal Processing202036(9):1481-1488.
15 叶冲,杨晶东 .基于CBAM-EfficientNet的垃圾图像分类算法研究[J].智能计算机与应用202111(5):218-222.
  YE Chong, YANG Jingdong .Algorithm research on garbage image classification based on CBAM-EfficientNet[J].Intelligent Computer and Application202111(5):218-222.
16 姜天宇,赵晓林,赵搏欣,等 .基于EfficientNet的木薯叶病变自动分类模型[J].计算机应用202242(S1):64-70.
  JIANG Tianyu, ZHAO Xiaolin, ZHAO Boxin,et al .Automatic classification model for cassava leaf disease based on EfficientNet[J].Computer Application202242(S1):64-70.
17 WANG Q, WU B, ZHU P,et al .Supplementary material for ECA-Net:efficient channel attention for deep convolutional neural networks[C]∥Proceedings of the 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition.Seattle,WA:IEEE,2020:13-19.
Outlines

/