金字塔卷积融合YOLOv4的乳腺肿块检测算法

Breast Mass Detection Algorithm Based on Pyramid Convolution Fusion with YOLOv4

摘要: 针对乳腺肿块与腺体对比度低而导致检测精度低，假阳性率较高的问题，提出了一种基于改进的YOLOv4网络的乳腺肿块检测方法。首先引入金字塔卷积，通过不同大小和深度的卷积核，对输入的特征进行提取；其次，将原特征提取网络中的普通卷积替换为深度可分离卷积，减少网络训练参数，提升网络训练速度。实验结果表明，改进的YOLOv4算法在测试集上的敏感性达到81.49%，较原网络高了2.81%，AP值达到86.85%，比原网络的AP值高了4.27%；平均每幅图假阳性个数为0.418个，相较原网络降低了0.028个，该算法的检测性能较YOLOv4有明显提升。

Abstract: Aiming at the problems of low detection accuracy and high false positive rate caused by low contrast between breast mass and gland, a breast mass detection method based on improved YOLOv4 is proposed. Firstly, Pyramid Convolution is introduced to extract the input features through convolution kernels of different sizes and depths; Secondly, the ordinary convolution in the original feature extraction network is replaced by Depthwise Separable Convolution to reduce the network training parameters and improve the network training speed. The experimental results show that the sensitivity of the improved YOLOv4 algorithm on the test set is 81.49%, which is 2.81% higher than that of the original network, and the AP value is 86.85%, which is 4.27% higher than that of the original network; The average number of false positives per image is 0.418, which is 0.028 lower than the original network. The detection performance of the algorithm is significantly improved compared with YOLOv4.