The invention discloses a traffic flow prediction method for personalized federal learning based on a prior attention mechanism, and the method comprises the steps: 1, a central server initializes and deploys a client model, and a client carries out the encoder model training and the attention mechanism calculation of time correlation, and uploads a traffic flow transmission state; 2, the central server receives the traffic flow transmission state, performs prior attention mechanism calculation of spatial correlation, and issues the processed traffic flow transmission state; 3, the client receives the processed traffic flow transmission state, performs decoder model training, and predicts a real label; 4, the client uploads the trained encoder parameters and decoder parameters to a central server, and the central server performs attention mechanism aggregation on the parameters and issues the parameters; and 5, the local server receives the aggregated parameters and brings the aggregated parameters into the model for traffic flow prediction. According to the invention, the accuracy of traffic flow prediction can be improved, and the data privacy of each client can be guaranteed.

    本发明公开了一种基于先验注意力机制的个性化联邦学习的交通流量预测方法,包括:1中央服务器初始化并部署客户端模型,客户端进行编码器模型训练以及时间相关性的注意力机制计算,并上传交通流量传递状态;2中央服务器接收流量传递状态,并进行空间相关性的先验注意力机制计算,并下发处理后的交通流量传递状态;3客户端接收处理后的交通流量传递状态并进行解码器模型训练,预测真实标签;4客户端将训练好的编码器参数和解码器参数上传至中央服务器,中央服务器对参数进行注意力机制的聚合并下发参数;5地方服务器接收聚合后的参数带入模型用于交通流量预测。本发明不仅能提高交通流量预测的准确性,还能保障每个客户端的数据隐私。


    Access

    Download


    Export, share and cite



    Title :

    Personalized federated learning traffic flow prediction method based on prior attention mechanism


    Additional title:

    基于先验注意力机制的个性化联邦学习的交通流量预测方法


    Contributors:
    CHAI YIDONG (author) / HUANG AMEI (author) / LIU YI (author) / LIU LONGSHUN (author) / PAN YUE (author) / CHEN YUYING (author) / LIU HAOXIN (author) / ZHOU LUJUAN (author) / ZHOU ANQI (author) / YANG XINGGE (author)

    Publication date :

    2024-04-30


    Type of media :

    Patent


    Type of material :

    Electronic Resource


    Language :

    Chinese


    Classification :

    IPC:    G06Q Datenverarbeitungssysteme oder -verfahren, besonders angepasst an verwaltungstechnische, geschäftliche, finanzielle oder betriebswirtschaftliche Zwecke, sowie an geschäftsbezogene Überwachungs- oder Voraussagezwecke , DATA PROCESSING SYSTEMS OR METHODS, SPECIALLY ADAPTED FOR ADMINISTRATIVE, COMMERCIAL, FINANCIAL, MANAGERIAL, SUPERVISORY OR FORECASTING PURPOSES / G06N COMPUTER SYSTEMS BASED ON SPECIFIC COMPUTATIONAL MODELS , Rechnersysteme, basierend auf spezifischen Rechenmodellen / G08G Anlagen zur Steuerung, Regelung oder Überwachung des Verkehrs , TRAFFIC CONTROL SYSTEMS




    Traffic flow prediction method based on attention mechanism

    SUN LIJUN / LIU MINGZHI / LIU GUANFENG et al. | European Patent Office | 2022

    Free access

    Cluster type federated learning-based urban traffic flow prediction method

    ZHANG ZIJIAN / BAI SIMENG / WU YAN et al. | European Patent Office | 2024

    Free access

    Traffic flow prediction method based on self-attention mechanism and deep learning

    JIE YURUI / HAO CHUNLIN / ZHANG JIAN | European Patent Office | 2024

    Free access

    Traffic flow prediction method based on space-time attention mechanism

    WANG JIAYING / YANG HENG / SHAN JING et al. | European Patent Office | 2024

    Free access