FireLite: Leveraging Transfer Learning for Efficient Fire Detection in Resource-Constrained Environments

Mahamudul Hasan,Md Maruf Al Hossain Prince,Mohammad Samar Ansari,Sabrina Jahan,Abu Saleh Musa Miah,Jungpil Shin
2024-09-30
Abstract:Fire hazards are extremely dangerous, particularly in sectors such as the transportation industry, where political unrest increases the likelihood of their occurrence. By employing IP cameras to facilitate the setup of fire detection systems on transport vehicles, losses from fire events may be prevented proactively. However, the development of lightweight fire detection models is required due to the computational constraints of the embedded systems within these cameras. We introduce FireLite, a low-parameter convolutional neural network (CNN) designed for quick fire detection in contexts with limited resources, in response to this difficulty. With an accuracy of 98.77\%, our model -- which has just 34,978 trainable parameters achieves remarkable performance numbers. It also shows a validation loss of 8.74 and peaks at 98.77 for precision, recall, and F1-score measures. Because of its precision and efficiency, FireLite is a promising solution for fire detection in resource-constrained environments.
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
The problem that this paper attempts to solve is to achieve efficient fire detection in resource - constrained environments (such as embedded systems and IP cameras). Specifically, fire is especially dangerous in the transportation industry, especially in politically unstable regions where the risk of fire is higher. Traditional fire detection methods rely on human intervention or simple alarm systems and usually cannot provide timely responses, resulting in serious damage or even loss of life. To meet this challenge, the author proposes a lightweight convolutional neural network (CNN) named "FireLite", which aims to detect fire quickly and accurately while adapting to the computational limitations of resource - constrained devices. The following are the specific problems and solutions mentioned in the paper: ### 1. **Problem Background** - **Fire Risk**: Especially in the transportation industry, fire can lead to serious accidents, affecting everything from passenger trains to cargo ships. - **Limitations of Traditional Methods**: Traditional fire detection methods rely on human intervention or simple alarm systems and cannot provide timely responses. - **Resource - Constrained Environments**: Devices such as embedded systems and IP cameras usually have limited memory and processing power, so lightweight models need to be developed to adapt to these limitations. ### 2. **Proposed Method** - **FireLite Model**: The author designed a lightweight CNN model with only 34,978 parameters, which can operate efficiently in resource - constrained environments. - **Transfer Learning**: By using the pre - trained MobileNet architecture and combining transfer learning techniques, FireLite can utilize high - level features in the ImageNet dataset, thereby improving detection performance. - **Model Optimization**: By fine - tuning some layers of MobileNet and introducing regularization techniques (such as batch normalization and dropout), FireLite reduces the risk of overfitting while maintaining high accuracy. ### 3. **Experimental Results** - **Performance Metrics**: FireLite has an accuracy rate of 99.18% on the FireNet dataset, a validation loss of 8.74, and performs well in terms of precision, recall, and F1 - score. - **Comparison with Existing Models**: Compared with other existing lightweight fire detection models (such as FireNet, FireNet - v2, FireNet - Tiny, etc.), FireLite can still maintain high performance with a significantly reduced number of parameters. ### 4. **Conclusion** - FireLite achieves fast and accurate fire detection in resource - constrained environments through its efficient architecture and transfer learning techniques. - The lightweight design of this model makes it suitable for devices such as embedded systems and IP cameras, providing a new solution for fire prevention in the transportation industry. In summary, this paper aims to solve the deficiencies of existing methods in resource - constrained environments by developing a lightweight and efficient fire detection model, thereby improving the timeliness and accuracy of fire detection.