arXiv:2512.06183cs.LG2025-12被引 1

用物理约束和掩码感知扩散模型,从稀疏数据重建高速路速度场。

PMA-Diffusion: A Physics-guided Mask-Aware Diffusion Framework for TSE from Sparse Observations

  • 设计掩码感知训练策略,让扩散模型学习稀疏观测下的速度分布规律。
  • 仅5%观测覆盖率下,三项误差指标均优于现有方法。
  • 适合交通流估计、智能交通系统等需处理稀疏传感数据的场景。

高分辨率高速公路交通状态信息对智能交通系统至关重要,但传统环形检测器和探针车辆采集的数据往往过于稀疏且噪声大,难以捕捉交通流的细节动态。本文提出PMA-Diffusion,一种基于物理引导的掩码感知扩散框架,用于从稀疏不完整观测中重建缺失的高速公路速度场。该方法在稀疏观测速度场上直接训练扩散先验,采用单掩码与双掩码两种掩码感知训练策略。推理阶段,物理引导后验采样器通过交替执行反向扩散更新、观测投影及基于自适应各向异性平滑的物理投影,实现缺失速度场的重建。在I-24 MOTION数据集上,不同可见率条件下进行测试。即使在严重稀疏情况下(仅5%可见率),PMA-Diffusion在三项重建误差指标上均超越其他基线模型。此外,使用稀疏观测训练的PMA-Diffusion几乎达到在完整观测数据上训练的基线模型性能。结果表明,结合掩码感知扩散先验与物理引导后验采样器,可为真实传感稀疏条件下的交通状态估计提供可靠且灵活的解决方案。

原文摘要 · Abstract (English)

High-resolution highway traffic state information is essential for Intelligent Transportation Systems, but typical traffic data acquired from loop detectors and probe vehicles are often too sparse and noisy to capture the detailed dynamics of traffic flow. We propose PMA-Diffusion, a physics-guided mask-aware diffusion framework that reconstructs unobserved highway speed fields from sparse, incomplete observations. Our approach trains a diffusion prior directly on sparsely observed speed fields using two mask-aware training strategies: Single-Mask and Double-Mask. At the inference phase, the physics-guided posterior sampler alternates reverse-diffusion updates, observation projection, and physics-guided projection based on adaptive anisotropic smoothing to reconstruct the missing speed fields. The proposed framework is tested on the I-24 MOTION dataset with varying visibility ratios. Even under severe sparsity, with only 5% visibility, PMA-Diffusion outperforms other baselines across three reconstruction error metrics. Furthermore, PMA-diffusion trained with sparse observation nearly matches the performance of the baseline model trained on fully observed speed fields. The results indicate that combining mask-aware diffusion priors with a physics-guided posterior sampler provides a reliable and flexible solution for traffic state estimation under realistic sensing sparsity.

交通估计扩散模型稀疏观测

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。