CNIPA.AI
검색으로 돌아가기
기록

一种基于跨架构知识蒸馏的网络任务模型构建方法

발명유효
10청구항 · 4 독립항
§ Ⅰ

개요

발명자

雷凯; 蒋杰; 夏霖翰; 邹贤能; 马晨浩; 张世琨; 崔斌; 晏紫微; 郑海洋

IPC 분류

H04L 41/14 (2022.01)H04L 41/147 (2022.01)H04L 41/16 (2022.01)H04L 47/10 (2022.01)G06F 18/25 (2023.01)G06F 18/214 (2023.01)G06N 3/0464 (2023.01)G06N 3/09 (2023.01)

本申请涉及一种基于跨架构知识蒸馏的网络任务模型构建方法,包括利用多模态编码器对多模态的网络运行数据集进行编码和投影,得到统一令牌;将统一令牌输入至包括Transformer模块和第一任务特定网络头的教师模型中,通过教师模型中的Transformer模块和第一任务特定网络头输出得到教师模型网络状态预测结果;将统一令牌输入至包括Mamba模块和第二任务特定网络头的学生模型中,通过学生模型中的Mamba模块和第二任务特定网络头输出得到学生模型网络状态预测结果;基于教师模型网络状态预测结果和学生模型网络状态预测结果构建总损失函数,利用总损失函数对学生模型进行训练,得到训练好的网络任务模型。上述方法能够同时解决计算复杂度和存储冗余的技术问题。