公开(公告)号：CN118861558A

公开(公告)日：2024-10-29

申请号：CN202410884910.4

申请日：2024-07-03

申请人： 上海交通大学

发明人： 王中瑞 ,  高睿 ,  朱修奇 ,  赵嘉瑶

IPC分类号： G06F18/20 ,  G06F18/10 ,  G06F18/214 ,  G06F18/2433 ,  G06N3/0442 ,  G06N3/045 ,  G06N3/082 ,  G06N3/084 ,  G06F123/02

摘要： 本发明涉及一种基于LSTM网络的水面多目标运动状态估计轻量化方法，包括以下步骤：获取水面目标运动状态历史信息并进行预处理；利用预处理后的水面目标运动状态历史信息对Social‑LSTM网络进行预训练，得到水面多目标运动状态预训练模型；基于掩码模型对水面多目标运动状态预训练模型的权重矩阵进行非结构化剪枝，并对剪枝后的模型进行重训练，得到水面多目标运动状态重训练模型；基于水面多目标运动状态重训练模型进行多目标运动状态估计。与现有技术相比，本发明能够克服基于深度学习的水面多目标运动状态估计算法的硬件需求高、模型泛化性差的缺陷，大大提高针对水面多目标的运动状态估计的效率。