本发明提供一种用于交通状态估计的联邦学习算法,涉及交通状态监管领域。该用于交通状态估计的联邦学习算法,包括以下步骤:S1、构建LSTM;S2、发送给各个RSU;S3、模型训练;S4、利用A3C算法强化学习算法进行参数上传以及下载策略的分配;S5、各个RSU模型根据参数上传以及下载策略,选择是否将模型参数发送给服务器或者从服务器下载最新的模型参数以继续训练;S6、参数聚合并计算成本;S7、重复步骤S2‑S6;S8、得到预测结果以及成本。通过长短时记忆(LSTM)模型提取交通状态的时间相关性,并估计交通流量及道路车速两种交通状态指标,解决了基于联邦学习的交通状态估计研究网络资源分配不合理、客户端参与联邦学习不均衡的问题。
声明:
“用于交通状态估计的联邦学习算法” 该技术专利(论文)所有权利归属于技术(论文)所有人。仅供学习研究,如用于商业用途,请联系该技术所有人。
我是此专利(论文)的发明人(作者)