融合DropBlock和注意力机制的视频异常检测算法OA
A Video Anomaly Detection Algorithm Combining DropBlock and Attention Mechanism
视频异常检测是计算机视觉领域的关键任务之一.传统的视频异常检测方法存在着在复杂场景下易受背景噪声干扰、难以有效捕捉局部细节,且易因过拟合训练数据而导致模型泛化能力差、鲁棒性不足的问题.针对这些挑战,该文提出了一种融合DropBlock与注意力机制的视频异常检测算法.该算法基于U-Net架构,在瓶颈层和跳跃连接中分别引入了SE模块(Squeeze-and-EXcitation Module)和空间注意力模块(Spatial Attention Module),SE模块通过通道注意力机制增强重要通道的特征表示,而空间注意力模块则通过动态调整空间权重,提升对关键区域的关注.在SE模块后融合了Transformer,增强模型对视频时空特征的建模能力.同时,通过在卷积层中引入DropBlock,有效缓解了卷积网络的过拟合问题,增强了模型的泛化能力.实验结果表明,该方法在UCSD-Ped2、CUHK Avenue和ShanghaiTech公开数据集上的AUC指标分别达到96.9%、86.2%和73.1%,验证了其有效性.
Video anomaly detection is one of the key tasks in the field of computer vision.Traditional video anomaly detection methods are susceptible to background noise interference,difficult to effectively capture local details,and prone to poor generalization ability and insufficient robustness due to overfitting training data.In order to solve these challenges,we propose a video anomaly detection algorithm that combines DropBlock and attention mechanism.Based on the U-Net architecture,the Squeeze-and-Excitation Module and the Spatial Attention Module are introduced into the bottleneck layer and the jump connection,respectively.The SE module enhances the feature representation of important channels through the channel attention mechanism,while the spatial attention module increases the focus on key regions by dynamically adjusting spatial weights.The Transformer is integrated after the SE module to enhance the model's ability to model the spatio-temporal features of videos.At the same time,by introducing DropBlock into the convolutional layer,the overfitting problem of the convolutional network is effectively alleviated and the generalization ability of the model is enhanced.The ex-perimental results show that the AUC indexes of the proposed method on UCSD-Ped2,CUHK Avenue and ShanghaiTech public datasets reach 96.9%,86.2%and 73.1%,respectively,which verifies its effectiveness.
施圣卿;杨大为
沈阳理工大学 信息科学与工程学院,辽宁 沈阳 110159沈阳理工大学 信息科学与工程学院,辽宁 沈阳 110159
信息技术与安全科学
深度学习DropBlock空间注意力模块SE模块Transformer
deep learningDropBlockspatial attention moduleSE moduleTransformer
《计算机技术与发展》 2026 (2)
54-61,8
辽宁省自然科学基金面上项目(2022-MS-276)
评论