基于DDPM−MBN的井下人员步态识别方法

Gait recognition method for underground personnel based on DDPM-MBN

  • 摘要: 现有基于毫米波雷达的人员步态识别方法基于小规模数据集进行训练,导致模型泛化能力不足,且难以从复杂的井下环境中提取有效的全局特征与局部特征,造成识别精度较低。针对上述问题,提出了一种基于去噪扩散概率模型(DDPM)−多分支网络(MBN)的井下人员步态识别方法。采用DDPM对雷达回波转换得到的时频谱图进行去噪与数据增强,有效扩充了井下步态数据量,提升了数据质量;通过MBN的1个全局分支和2个局部分支分别提取步态的全局特征和不同粒度的局部特征,实现了对步态多尺度特征的充分提取,提升了对行走方向和行走速度的识别能力;联合使用Softmax损失与三元组损失,对粗粒度特征(未经降维的2 048维特征)与细粒度特征(经降维后的256维特征)进行协同优化,从而增强了模型的宏观分类能力与特征判别性。实验结果表明,在自建的步态数据集上,DDPM−MBN模型的Rank−1准确率和平均精度均值(mAP)相较于ResNet−50分别提升了8.05%,16.96%;与主流步态识别模型相比,DDPM−MBN模型在Rank−1准确率和mAP指标上均最优,分别为97.91%和95.48%。

     

    Abstract: Existing millimeter-wave radar-based gait recognition methods are typically trained on small-scale datasets, resulting in poor model generalization and limited ability to extract effective global and local features from complex underground environments, which leads to low recognition accuracy. To address these issues, a gait recognition method for underground personnel based on the Denoising Diffusion Probabilistic Model (DDPM) and Multi-Branch Network (MBN) was proposed. The DDPM was used to denoise and augment the time–frequency spectrograms converted from radar echoes, effectively expanding the quantity of underground gait data and improving data quality. The MBN, consisting of one global branch and two local branches, extracted global gait features and local features of different granularities, enabling sufficient multi-scale feature extraction and improving the recognition of walking direction and speed. The Softmax loss and triplet loss were jointly employed to optimize coarse-grained features (2 048-dimensional features before dimensionality reduction) and fine-grained features (256-dimensional features after dimensionality reduction) in a collaborative manner, thereby enhancing the model's overall classification ability and feature discriminability. Experimental results showed that, on the self-built gait dataset, the DDPM-MBN model achieved Rank-1 accuracy and mean Average Precision (mAP) improvements of 8.05% and 16.96%, respectively, compared with ResNet-50. Compared with mainstream gait recognition models, the DDPM-MBN model achieved the best performance, with Rank-1 accuracy and mAP reaching 97.91% and 95.48%, respectively.

     

/

返回文章
返回