![面向无人机群水库水质检测的分布式模型训练方法及系统](/CN/2023/1/147/images/202310735109.jpg)
基本信息:
- 专利标题: 面向无人机群水库水质检测的分布式模型训练方法及系统
- 申请号:CN202310735109.9 申请日:2023-06-20
- 公开(公告)号:CN116610958B 公开(公告)日:2024-07-26
- 发明人: 毛莺池 , 王佳俊 , 曾伟 , 李金洁 , 熊定松 , 张本腾 , 王龙宝 , 吴启华 , 秦鑫 , 安可君 , 戴继聪 , 张巍明 , 王四霞
- 申请人: 河海大学 , 华能澜沧江水电股份有限公司
- 申请人地址: 江苏省南京市江宁区佛城西路8号
- 专利权人: 河海大学,华能澜沧江水电股份有限公司
- 当前专利权人: 河海大学,华能澜沧江水电股份有限公司
- 当前专利权人地址: 江苏省南京市江宁区佛城西路8号
- 代理机构: 南京乐羽知行专利代理事务所(普通合伙)
- 代理人: 李玉平
- 主分类号: G06F18/214
- IPC分类号: G06F18/214 ; G06F18/24
摘要:
本发明公开一种面向无人机群水库水质检测的分布式模型训练方法及系统,在每一轮训练过程中,无人机从参数服务器接收到全局模型,对接收到的全局模型进行迭代训练,将交叉熵损失函数替换为焦点损失函数,平衡各类别样本在总损失值中的占比,降低本地模型对多数类样本的偏好性;在本地训练结束之后,无人机将模型梯度和损失值上传到参数服务器,参数服务器从接收到的模型梯度中选择主导梯度,并根据主导梯度对所有本地梯度进行修正,来缓解本地模型梯度间的冲突,最终提高水质检测模型的精度。本发明降低了中央服务器的负载,避免了模型训练时敏感数据泄露的风险,解决本地梯度在全局聚合阶段产生梯度冲突的问题,有助于更准确地识别水质问题。
公开/授权文献:
- CN116610958A 面向无人机群水库水质检测的分布式模型训练方法及系统 公开/授权日:2023-08-18
IPC结构图谱:
G | 物理 |
--G06 | 计算;推算;计数 |
----G06F | 电数字数据处理 |
------G06F18/00 | 模式识别 |
--------G06F18/10 | .预处理;数据清理 |
----------G06F18/21 | ..识别系统或技术的设计或设置;特征空间中的特征提取;盲源分离 |
------------G06F18/214 | ...生成训练模式;自引导方法,如捕获,促进 |