AVA-YOLO: image-based multiscale feature fusion enhanced perception model for snow avalanche detection

Abstract

The global climate change has led to frequent occurrences of snow avalanche disasters. However, the significant variations in scale and shape during the avalanche process, and complex background imagery pose significant challenges to automated detection efforts. There is an urgent need to combine advanced deep learning technology to research automatic detection and recognition of avalanches in the field. In this paper, a novel deep learning model based on YOLOv8 improved multi-scale detection called AVA-YOLO is proposed to solve this problem. In AVA-YOLO, a key component, AKA (AKConv Combined Attention) module was designed and developed. This module combines the deformable convolutional properties of AKConv with the state-of-the-art self-attention module Exponential Moving Average, aiming to better perceive the feature map information of different shaped avalanches and to enhance the global relevance, thus improving the utilization of the information. Secondly, a new multi-scale sensing network structure was designed by increasing the number of detection heads to four and introducing the AKA module into the key positions of the network, while the association between model layers was newly designed to enhance the fusion of shallow and deep information to improve the detection accuracy. Experimental results demonstrated the effectiveness of AVA-YOLO, achieving 95.7% mAP50 and 75.6% mAP50:95 detection accuracies, as well as an F1 score of 0.92. Finally, a number of experiments were conducted to demonstrate the superior performance of the proposed model in comparison to other versions of YOLO, which will further exploit the potential of webcams as an underutilized technical capability in snow avalanche intelligence and portable monitoring.