一种基于梯度掩膜的智能波束选择性能提升方法

    公开(公告)号:CN115066019A

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

    申请号:CN202210656050.X

    申请日:2022-06-10

    Applicant: 东南大学

    Abstract: 本发明公开了一种基于梯度掩膜的智能波束选择性能提升方法,具体包含以下步骤:基站收集数据建立数据集;基站建立并初始化一个神经网络结构,以学习输入数据与最优波束索引之间的映射;每一个训练回合开始之前,利用数据集计算一个梯度掩膜;训练过程中,使用随机梯度下降算法更新神经网络的参数,每次执行更新之前,将梯度乘以事先计算的梯度掩膜,得到被遮挡的梯度,使用被遮挡的梯度代替原始梯度执行更新;重复上述两步,迭代训练直至神经网络收敛。本发明通过提出的基于梯度掩膜的正则化步骤,有效减轻神经网络训练过程中的过拟合现象,从而提高基于深度学习的波束选择方法的准确率。

    一种基于微调的个性化波束选择方法

    公开(公告)号:CN115037344A

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

    申请号:CN202210663686.7

    申请日:2022-06-10

    Applicant: 东南大学

    Abstract: 本发明公开了一种基于微调的个性化波束选择方法,具体包含以下步骤:用户收集数据建立各自本地数据集;用户上传各自的本地数据集至中央服务器,构建全局数据集;中央服务器使用全局数据离线训练全局深度学习模型,而后将全局深度学习模型发送给用户;用户冻结全局模型的权重参数,仅允许更新偏置参数。用户仅使用本地数据继续训练全局模型,以微调全局模型使之更适合本地数据,从而建立个性化的波束选择模型;预测阶段,用户使用个性化的模型预测最优波束索引。本发明解决了现有方法难以在统计特性不同的用户数据上同时表现出优秀性能的问题,提高了基于深度学习的波束选择模型在用户本地数据上的预测准确率。

Patent Agency Ranking