核脉冲峰值序列轻量化神经网络核素识别模型及其FPGA加速方法

Lightweight neural network model for nuclide recognition based on nuclear pulse peak sequence and its FPGA acceleration method

  • 摘要: 放射性核素已在核医疗、核安保及无损检测等领域中广泛应用,而对其准确识别是放射性核素定性检测的基础。在便携式核素识别仪中,基于传统能谱分析方法存在延迟高、识别率低等不足。提出一种基于核脉冲峰值序列的核素识别轻量化神经网络模型及其FPGA硬件加速方法,通过引入深度可分离卷积和倒残差模块,并使用全局平均池化替代传统全连接层,构建了一种轻量化、高效的神经网络模型。针对网络训练数据集,通过蒙特卡罗工具包Geant4构建NaI(Tl)探测器模型,获取模拟能谱,再由核脉冲信号模拟仿真器根据能谱产生核脉冲信号序列,构建了16种核脉冲信号数据。最后,将训练好的模型通过量化、融合与并行计算等优化方法部署到PYNQ-Z2异构芯片,实现加速。实验结果表明,模型识别精度可达98.3%,相较传统卷积神经网络模型提高了13.2%,参数量仅为2 128。FPGA优化加速后单次识别耗时0.273 ms,功耗为1.94 W。

     

    Abstract: Radionuclides have been widely used in the fields of nuclear medicine, nuclear security and non-destructive testing, and their accurate identification is the basis of qualitative detection of radionuclides. In the portable nuclide recognition instrument, the traditional energy spectrum analysis method has the shortcomings of high delay and low recognition rate. This paper proposes a lightweight neural network model for nuclide recognition based on kernel pulse peak sequence and its FPGA hardware acceleration method. A lightweight and efficient neural network model is constructed by introducing depth-separable convolution and reciprocal residual modules, and using global average pooling to replace the traditional fully connected layer. For the network training data set, NaI (Tl) detector model was constructed through Monte Carlo toolkit Geant4 to obtain the analog energy spectrum, and then a simulator generated nuclear pulse signal sequences according to the energy spectrum, and 16 kinds of nuclear pulse signal data were constructed. Finally, the trained model is deployed to PYNQ-Z2 heterogeneous chip through optimization methods such as quantization, fusion and parallel computing to achieve acceleration. Experimental results show that the recognition accuracy of the proposed model can reach 98.3%, which is 13.2% higher than that of the traditional convolutional neural network model, and the number of parameters is only 2 128. After FPGA optimization and acceleration, the single recognition time is 0.273 ms, and the power consumption is 1.94 W.

     

/

返回文章
返回