Urban traffic flow prediction is a critical task in intelligent transportation systems (ITS), yet it faces challenges such as complex spatiotemporal dependency modeling and incomplete observation data. To address the widespread issue of point-wise missing data and the limited robustness of existing methods, this paper proposes a conditional diffusion-based traffic flow prediction framework, referred to as RDPFlow. The proposed method integrates a missing-aware masking mechanism with a heterogeneous external condition guidance strategy, enabling unified modeling of both traffic flow prediction and data imputation. By explicitly indicating missing regions, RDPFlow guides the model to distinguish between actual zero values and unobserved entries, and introduces external factors such as holidays and weather conditions for semantic enhancement, thereby improving the model’s adaptability to realistic missing patterns. Experimental results on TaxiBJ dataset show that RDPFlow consistently outperforms State-of-The-Art (SOTA) methods under both complete and missing data scenarios, achieving lower prediction errors and stronger generalization robustness.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

RDPFlow: A Conditional Diffusion Model for Traffic Flow Prediction with Point-Wise Missing Data

  • Yuan Xu,
  • Chen-Yang Yan,
  • Qun-Xiong Zhu,
  • Ming-Qing Zhang,
  • Wei Ke,
  • Chong-Xing Ji,
  • Yang Zhang

摘要

Urban traffic flow prediction is a critical task in intelligent transportation systems (ITS), yet it faces challenges such as complex spatiotemporal dependency modeling and incomplete observation data. To address the widespread issue of point-wise missing data and the limited robustness of existing methods, this paper proposes a conditional diffusion-based traffic flow prediction framework, referred to as RDPFlow. The proposed method integrates a missing-aware masking mechanism with a heterogeneous external condition guidance strategy, enabling unified modeling of both traffic flow prediction and data imputation. By explicitly indicating missing regions, RDPFlow guides the model to distinguish between actual zero values and unobserved entries, and introduces external factors such as holidays and weather conditions for semantic enhancement, thereby improving the model’s adaptability to realistic missing patterns. Experimental results on TaxiBJ dataset show that RDPFlow consistently outperforms State-of-The-Art (SOTA) methods under both complete and missing data scenarios, achieving lower prediction errors and stronger generalization robustness.