一种基于局部自注意力机制的个性化联邦聚合算法

    公开(公告)号:CN117350373B

    公开(公告)日:2024-03-01

    申请号:CN202311617677.5

    申请日:2023-11-30

    Abstract: 本发明涉及分布式机器学习技术领域,公开了一种基于局部自注意力机制的个性化联邦聚合算法,包括如下步骤:中央服务器将预训练好的全局模型分发给多个客户端;多个客户端根据本地数据集分别对所得模型进行训练;多个客户端将本地训练得到的模型参数分别加密,上传至中央服务器;中央服务器对所有模型的梯度进行聚合,生成全局模型参数;中央服务器将聚合后的全局模型参数分别加密,传回至多个客户端。本发明为解决联邦学习中非独立同分布数据的泛化误差问题提供了有效的策略,可以在保证数据隐私的同时提高模型的全局性能。(56)对比文件Hongda Wu等.Fast-Convergent FederatedLearning with Adaptive Weighting《.2021 |IEEE INTERNATIONAL CONFERENCE ONCOMMUNICATIONS (ICC 2021)》.2021,1078-1088.

    一种基于目标检测的多尺度特征融合方法

    公开(公告)号:CN115082688A

    公开(公告)日:2022-09-20

    申请号:CN202210620848.9

    申请日:2022-06-02

    Abstract: 本发明公开了一种基于目标检测的多尺度特征融合方法,其通过网络收集计算机视觉图像样本建立多尺度目标检测数据集,并将数据集划分为训练集和测试集;利用一阶段代表作YOLOv5算法负责图像内目标物体的检测;通过骨干网络多阶段多层次的卷积操作提取出多尺度图像特征;将其中一个支路以传统特征融合方式与颈部网络相连,另一支路以捷径方式与相同采样倍率的颈部网络相连,最后一支路以捷径方式与相同采样倍率的预测结构相连;通过深度学习一个三支路骨干网络结构,并将骨干网络中不同尺度的特征图像通过三支路向后实现神经网络的前向、后向传递;本发明具有目标检测准确率高,容易应用于大规模数据集及多种网络模型结构,实现方式简单,因此具有广阔的应用前景和巨大的市场价值。

Patent Agency Ranking