AVD-YOLOv5: a new lightweight network architecture for high-speed aortic valve detection from a new and large echocardiography dataset

Abstract

Abstract Heart disease detection is currently gaining widespread attention as a means to enhance the accuracy of cardiologists’ diagnoses from cardiac images and reduce diagnosis time. Although high-resolution computed tomography (CT) images are typically favored for heart disease detection, the drawbacks of cost and radiation exposure to patients necessitate alternative approaches. In this context, utilizing ultrasound images becomes pivotal to mitigate radiation risks and maintain cost-effectiveness. In this paper, we propose a novel lightweight model, AVD-YOLOv5, designed for automated aortic valve detection on echocardiography images. This model incorporates several enhancements to the YOLOv5 architecture. Notably, the depth-wise separable convolution significantly contributes to the model’s lightweight design by reducing the number of parameters while maintaining precision. We have also created a new and larger dataset comprising 260 echocardiography images specifically for aortic valve detection. Experimental results indicate that the precision value of the modified ADV-YOLOv5 model stands at 94.3%, with a recall value of 86.8%. The model also demonstrates a notable 67% reduction in inference time compared to the original YOLOv5 model. Although there is a marginal reduction in precision by 0.94%, the model’s efficiency is significantly increased. The proposed system can be used by cardiologists for more efficient and reliable diagnosis. Graphical abstract