一种模型训练方法、装置及存储介质

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
一种模型训练方法、装置及存储介质
申请号:CN202411561816
申请日期:2024-11-04
公开号:CN119561721A
公开日期:2025-03-04
类型:发明专利
摘要
本申请提供一种模型训练方法、装置及存储介质,涉及人工智能技术领域,能够提升小型DGA域名检测模型的准确率,提升资源利用率。该方法包括:获取训练数据集;基于第一域名检测模型,确定训练数据集中的域名所属类型的第一概率分布,并基于第二域名检测模型,确定训练数据集中的域名所属类型的第二概率分布以及第三概率分布;第一域名检测模型占用的内存大于第二域名检测模型占用的内存;根据第一概率分布、第二概率分布、第三概率分布以及训练数据集中域名的真实分布,确定第二域名检测模型的损失值;根据损失值对第二域名检测模型的参数进行梯度调整,直至调整后的第二域名检测模型的域名检测的准确率高于预设阈值。
技术关键词
模型训练方法 DGA域名 BERT模型 模型训练装置 计算机程序指令 数据 计算机程序产品 内存 字符 处理器 可读存储介质 人工智能技术 参数 存储器 误差 服务器 关系 框架