电子学报2024,Vol.52Issue(6) :1938-1944.DOI:10.12263/DZXB.20230230

基于特征蒸馏的变分编码器交通流预测模型

Traffic Flow Prediction Model Based on Spatio-Temporal Feature Distillation Variational Autoencoder

欧阳毅 汤文燕 黎晏伶
电子学报2024,Vol.52Issue(6) :1938-1944.DOI:10.12263/DZXB.20230230

基于特征蒸馏的变分编码器交通流预测模型

Traffic Flow Prediction Model Based on Spatio-Temporal Feature Distillation Variational Autoencoder

欧阳毅 1汤文燕 1黎晏伶1
扫码查看

作者信息

  • 1. 浙江工商大学管理科学与电子商务学院,浙江杭州 310000
  • 折叠

摘要

针对交通流数据高维非线性和时空依赖性复杂,本文构建了基于特征蒸馏的变分贝叶斯编码器交通流预测模型.对每段时间序列对应的时间窗口特征,构建了基于多模态时间槽和空间槽的交通流特征提取模型.以时空槽特征提取模型作为特征知识蒸馏架构的输入.通过知识蒸馏结构提取的时空特征结晶体,利用教师模型指导学生模型的学习过程,从而提高学生模型的泛化能力.变分贝叶斯编码器对交通流时空特征结晶编码获取交通流数据的隐变量,根据隐变量的生成采样,利用解码器将其解码重构成新的预测值.实验结果表明,本文提出的模型预测性能显著提升,且中长期预测中鲁棒性更优.

Abstract

To improve the accuracy of traffic flow prediction and to solve the problems of high-dimensional nonlin-earity and spatio-temporal dependence of traffic flow,a combined feature distillation and variational Bayes encoders traffic flow forecasting model (ST-DVBE) is proposed. First,to extract the time window characteristics corresponding to each time series,the multi-modal time slots and spatial slots are constructed. Second,with spatio-temporal slot feature extraction mod-el as the input of feature knowledge distillation architecture,and space-time feature crystallization extracted by knowledge distillation structure,the learning process of student model is guided by teacher model,so as to improve the generalization ability of student model. Finally,the variational Bayesian encoder is employed to capture the latent variables of traffic flow data by encoding the crystallization of spatiotemporal features. Utilizing the generated latent variables,the decoder recon-structs them into new predicted values. Experimental results demonstrate a significant enhancement in predictive perfor-mance with the proposed model,especially with better robustness in mid-and long-term forecasting.

关键词

特征蒸馏/多模态时间槽/空间槽/变分贝叶斯/生成式模型/变分推断

Key words

feature distillation/multimodal temporal slots/spatial slots/variational Bayes/generative model/variational inference

引用本文复制引用

基金项目

浙江工商大学"数字+学科建设项目"(SZJ2022C004)

浙江工商大学2023年度省级及以上教学平台自主设立校级教学项目(1310XJ0521036)

出版年

2024
电子学报
中国电子学会

电子学报

CSTPCD北大核心
影响因子:1.237
ISSN:0372-2112
参考文献量4
段落导航相关论文