基于Diffusion Model的水声数据增强方法  

An Underwater Acoustic Data Enhancement Method Based on Diffusion Model

作　　者：景语 李红烨 程彬彬 郭乔鹤 宋雨荷 JING Yu;LI Hongye;CHENG Binbin;GUO Qiaohe;SONG Yuhe(No.710 R&D Institute,CSSC,Yichang 443003,China;Qingjiang Innovation Center,CSSC,Wuhan 430076,China;China Ship Research Academy,Bejing 100101,China)

机构地区：[1]中国船舶集团有限公司第七一〇研究所,湖北宜昌443003 [2]清江创新中心,湖北武汉430076 [3]中国舰船研究院,北京100101

出　　处：《数字海洋与水下攻防》2024年第6期639-647,共9页Digital Ocean & Underwater Warfare

摘　　要：深度学习需要大量数据对网络模型进行训练才能达到良好的识别效果,而水声数据的数量较少且难以获取。为了解决这一问题,采用了一种基于改进的扩散模型的水声数据增强方法,改进了噪声预测网络并增加条件生成模块。使用海上试验采集的水声数据进行训练,构建数据生成模型,生成高质量的数据样本,扩充训练数据集。实验结果表明,使用AlexNet作为分类模型进行训练时,该方法相较于传统数据增强方法能够将识别错误率降低约30%,验证了方法的有效性。Deep learning requires a large amount of data to train network models in order to achieve good recognition performance,while underwater acoustic data is scarce and difficult to obtain.To solve this problem,a diffusion model-based underwater acoustic data augmentation method is adopted,which uses underwater acoustic data collected from offshore experiments for training,constructs a data generation model,generates high-quality data samples,and expands the training dataset.The experimental results show that when using AlexNet as a classification model for training,this method can reduce the recognition error rate by about 30%compared with traditional data augmentation methods,which verifies the effectiveness of the proposed method.

关 键 词：水声目标识别 扩散模型 深度学习 数据增强 

分 类 号：TP391[自动化与计算机技术—计算机应用技术]