Please wait a minute...
空天防御  2024, Vol. 7 Issue (3): 94-101    
0
  专业技术 本期目录 | 过刊浏览 | 高级检索 |
基于改进注意力机制的自适应航迹预测方法
黄权印, 蔡益朝, 李浩, 唐晓, 王辰洋
空军预警学院,湖北 武汉 430000
Adaptive Trajectory Prediction Method Based on Improved Attention Mechanism
HUANG Quanyin, CAI Yichao, LI Hao, TANG Xiao, WANG Chenyang
Air Force Early Warning Academy, Wuhan 430000, Hubei, China
全文: PDF(2018 KB)  
输出: BibTeX | EndNote (RIS)      
摘要 针对现有循环神经网络在解决目标航迹预测问题中容易存在训练过拟合、预测精度不高、泛化能力差以及适应性不强的问题,提出了一种基于改进注意力机制和门控循环单元( GRU)的目标航迹预测方法。该方法通过早停法来自动终止网络训练过程,防止训练过拟合;通过模型检查点函数保存网络训练中的最优网络参数;通过把注 意力机制引入GRU 网络中,对轨迹特征赋予不同的权重来聚焦重点航迹信息,提高网络的预测性能。最后,通过仿真实验证明,该方法能够有效提升循环神经网络的预测精度、泛化性及适应性。
服务
把本文推荐给朋友
加入引用管理器
E-mail Alert
RSS
作者相关文章
关键词 航迹预测注意力机制早停法循环神经网络门控循环单元    
Abstract:The existing recurrent neural networks are subject to training overfitting, low prediction accuracy, poor generalization ability, and weak adaptability in solving target trajectory prediction. A target trajectory prediction method using an improved attention mechanism and Gated Recurrent Unit (GRU) was proposed, which could automatically terminate the network training process through an early stopping method to prevent overfitting during training. It saved the optimal network parameters during network training through the model checkpoint function. By introducing an attention mechanism into the GRU network and assigning different weights to trajectory features to focus on key trajectory information, the predictive performance of the network was optimized Finally, simulation experiments results show that the proposed method effectively improves the prediction accuracy, generalization, and adaptability of recurrent neural networks.
Key wordstrajectory prediction    attention mechanism    early stop method    recurrent neural network    gated recurrent unit
收稿日期: 2024-04-17      出版日期: 2024-07-25
:  TN 953  
基金资助:国家自然科学基金青年(61502522);国家社科基金重点(2022-SKJJ-B-056);湖北省自科基金面上项目(2019CFC897)
作者简介: 黄权印(1999—),男,硕士研究生。
引用本文:   
黄权印, 蔡益朝, 李浩, 唐晓, 王辰洋. 基于改进注意力机制的自适应航迹预测方法[J]. 空天防御, 2024, 7(3): 94-101.
HUANG Quanyin, CAI Yichao, LI Hao, TANG Xiao, WANG Chenyang. Adaptive Trajectory Prediction Method Based on Improved Attention Mechanism. Air & Space Defense, 2024, 7(3): 94-101.
链接本文:  
https://www.qk.sjtu.edu.cn/ktfy/CN/      或      https://www.qk.sjtu.edu.cn/ktfy/CN/Y2024/V7/I3/94

参考文献
[1] 臧红岩, 谢晓龙, 徐亚周, 陶业, 高长生. 基于循环神经网络的高超声速机动目标状态估计算法[J]. 空天防御, 2024, 7(4): 88-98.
[2] 蔡云泽, 张彦军. 基于双通道特征增强集成注意力网络的红外弱小目标检测方法[J]. 空天防御, 2021, 4(4): 14-22.
沪ICP备15013849号-1
版权所有 © 2017《空天防御》编辑部
主管单位:中国航天科技集团有限公司 主办单位:上海机电工程研究所 上海交通大学出版社有限公司