一种基于联邦学习的多用户协同训练人流统计方法及系统
摘要:
本发明提供一种基于联邦学习的多用户协同训练人流统计方法,旨在利用联邦学习框架搭建一个新颖的人群计数模型,达到让多用户多设备同时训练的目的。各个客户端利用图像数据集对图像分类网络进行本地训练以获取本地模型;在各经过至少一次本地训练后,中心服务器从客户端获取本地模型的权值及附加层参数并进行聚合处理;中心服务器利用聚合处理后的权值及附加层参数更新全局模型,并将聚合处理后的权值参数及附加层参数返回给各个客户端;各个客户端利用中心服务器返回的权值以及ground truth值进行贝叶斯估计,计算loss值,并利用返回的权值参数及附加层参数更新本地模型;重复执行直至所有客户端的loss值均收敛,则完成人流统计全局模型和本地模型的训练。
0/0