一种基于扩散模型的雷电波生成方法和系统

本发明涉及雷电数据处理技术领域，尤其是一种基于扩散模型的雷电波生成方法和系统。本发明采用扩散模型生成雷电波数据，逆向扩散过程中采用的去噪神经网络在UNET模型结构上改进得到，改进方式为：在每一次下采样之后和上采样之前均加入小波注意力单元。本发明通过改进去噪神经网络，结合注意力技术小波变换，同时捕捉多尺度变换，提高模型的细节捕捉能力，特别是捕捉高频成分，帮助模型更好地保留雷电波的关键特征，从而提高扩散模型应用在雷电波生成任务上的品质。本发明克服了上述现有技术中扩散模型直接应用于雷电波的缺陷。