Flood-ResNet50: Optimized Deep Learning Model for Efficient Flood Detection on Edge Device

• Main Authors: Khan, Md Azim, Ahmed, Nadeem, Padela, Joyce, Raza, Muhammad Shehrose, Gangopadhyay, Aryya, Wang, Jianwu, Foulds, James, Busart, Carl, Erbacher, Robert F.
• Format: Conference Proceeding
• Language: English
• Subjects: AlexNet; Biological system modeling; Computational modeling; Deep learning; Edge device; Image edge detection; Jetson nano; ResNet50; Transfer learning; Transformers; VGG; Vision transformer; Wildlife
• ISSN: 1946-0759
• DOI: 10.1109/ICMLA58977.2023.00077

Floods are highly destructive natural disasters that result in significant economic losses and endanger human and wildlife lives. Efficiently monitoring Flooded areas through the utilization of deep learning models can contribute to mitigating these risks. This study focuses on the deployment of deep learning models specifically designed for classifying flooded and non-flooded in UAV images. In consideration of computational costs, we propose modified version of ResNet50 called Flood-ResNet50. By incorporating additional layers and leveraging transfer learning techniques, Flood-ResNet50 achieves comparable performance to larger models like VGG16/19, AlexNet, DenseNet161, EfficientNetB7, Swin(small), and vision transformer. Experimental results demonstrate that the proposed modification of ResNet50, incorporating additional layers, achieves a classification accuracy of 96.43%, F1 score of 86.36%, Recall of 81.11%, Precision of 92.41 %, model size 98MB and FLOPs 4.3 billions for the FloodNet dataset. When deployed on edge devices such as the Jetson Nano, our model demonstrates faster inference speed (820 ms), higher throughput (39.02 fps), and lower average power consumption (6.9 W) compared to larger ResNet101 and ResNet152 models.