首页|期刊导航|中南大学学报(自然科学版)|自适应损失权重的多任务大地电磁深度学习反演

自适应损失权重的多任务大地电磁深度学习反演OA

Multi-task deep learning inversion for magnetotellurics using adaptive loss weighting

中文摘要英文摘要

数据驱动的深度学习在解决大地电磁反问题中具有良好的效果.为进一步提高深度学习在大地电磁的反演回归问题的泛化能力,本文对自适应损失权重策略的深度学习反演方法进行研究,通过构建一个"W"型的Transformer与卷积神经网络(CNN)混合编码的深度学习模型MSEAdpNet执行多任务学习,并利用自适应损失权重反馈引导模型的训练.该模型由1个共享的编码器和2个独立的解码器组成,其中,一个解码器进行反演,另一个解码器进行类物理驱动,训练时以含噪声数据作为输入,通过开发自适应误差权重函数,在提升任务总体性能的情况下,确保了反演解码器的权重占主导.研究结果表明:自适应损失权重策略的类物理驱动多任务大地电磁深度学习反演能保证验证误差有效下降,并且在理论模型反演和实测数据反演中均能取得良好的效果.

Data-driven deep learning has great potential in solving the inverse problems of magnetotellurics.To enhance the generalization ability of deep learning in the inversion regression problems of magnetotellurics,a deep learning inversion method was investigated based on an adaptive loss weighting strategy.A"W"-shaped deep learning model MSEAdpNet was constructed,which combined a Transformer and convolutional neural network(CNN)for multi-task learning,guiding the model training through adaptive loss weighting feedback.The model took noisy data as input and consisted of a shared encoder and two independent decoders.One decoder estimated the model error for the inversion task,while the other estimated the response data error for the auxiliary task.By using adaptive error weighting,the model ensured the priority of the inversion task while improved overall task performance.The results demonstrate that the multi-task deep learning approach with an adaptive loss weighting strategy can achieve good inversion results,effectively reduce validation errors,and it is validated with real measurement results.

樊娟;席振铢;王鹤;刘英锋;朱开鹏

西安科技大学 地质与环境学院,陕西 西安,710054||中煤科工西安研究院 (集团)有限公司,陕西 西安,710077||陕西省煤矿水害防治技术重点实验室,陕西 西安,710077中南大学 地球科学与信息物理学院,湖南 长沙,410083中南大学 地球科学与信息物理学院,湖南 长沙,410083中煤科工西安研究院 (集团)有限公司,陕西 西安,710077||陕西省煤矿水害防治技术重点实验室,陕西 西安,710077中煤科工西安研究院 (集团)有限公司,陕西 西安,710077||陕西省煤矿水害防治技术重点实验室,陕西 西安,710077

天文与地球科学

多任务深度学习大地电磁反演自适应损失权重神经网络

multi-task deep learningmagnetotellurics inversionadaptive loss weightingneural network

《中南大学学报(自然科学版)》 2026 (4)

1636-1648,13

黔科合重大专项([2024]029)国家重点研发计划项目(2024YFC2909203,2022YFC2903404)(Project([2024]029)supported by Guizhou Department of Science and TechnologyProjects(2024YFC2909203,2022YFC2903404)supported by the National Key Research and Development Program of China)

10.11817/j.issn.1672-7207.2026.04.015

评论