基于不确定性感知的共形预测与世界模型强化学习的城市交通安全控制
Safe Urban Traffic Control via Uncertainty-Aware Conformal Prediction and World-Model Reinforcement Learning
Joydeep Chandra, Satyam Kumar Navneet, Aleksandr Algazinov, Yong Zhang
arXiv (Cornell University) · 2026-02-04
摘要
城市交通管理需要能够同时预测未来状况、识别异常并采取安全纠正措施的系统,且需具备可靠性保障。我们提出了STREAM-RL这一统一框架,包含三项创新性算法贡献:(1)PU-GAT+,一种不确定性引导的自适应共形预测器,利用预测不确定性通过置信度单调注意力机制动态调整图注意力权重,可实现无分布假设的覆盖性保障;(2)CRFN-BY,一种共形残差流网络,借助归一化流对经不确定性标准化的残差进行建模,在任意依赖关系下可通过Benjamini-Yekutieli方法控制错误发现率;(3)LyCon-WRL+,一种不确定性引导的安全世界模型强化学习智能体,具备李雅普诺夫稳定性证明、经认证的利普希茨界以及基于不确定性传播的想象性推演能力。据我们所知,这是首个能从预测环节到异常检测再到安全策略学习全程传递校准后不确定性、并提供端到端理论保障的框架。在多个真实交通轨迹数据上的实验表明,STREAM-RL的覆盖效率达到91.4%,在经验证的依赖关系下可将错误发现率控制在4.1%,安全率提升至95.2%,而传统PPO算法的安全率仅为69%;同时该框架还能获得更高的奖励值,端到端推理延迟仅为23毫秒。
本摘要由 AI(Hy-MT2-Pro)翻译生成,仅供参考,请以原文为准。
AI 综述
基于全文生成- 主要观点
- 论文提出 STREAM-RL 这一统一框架,核心主张是把经校准的不确定性从预测环节贯穿到异常检测与安全策略学习,并给出端到端的理论保证。其问题域是城市交通中耦合的三项任务:预测须给出满足 P(Y∈[L,U])≥1−α 的区间(RQ1),异常检测须在任意时空依赖下将 FDR 控制在 α 以内(RQ2),控制须在队列长度≤50 辆、最大等待≤120 秒、吞吐不低于基准 0.8 倍三类约束下最大化回报(RQ3)。作者把既有集成式方法的不足概括为三点:温度缩放注意力 α_ij∝exp(e_ij/(1+βσ_i)) 只改变 softmax 温度、无法在置信与不确定邻居之间作出区分;Benjamini-Hochberg 程序在空间自相关与时间持续性下 FDR 保证失效;Lyapunov 安全 RL 的保证所依赖的有界模型误差与 Lipschitz 假设很少被经验核验。对应地提出 PU-GAT+ 的成对不确定性注意力(γ=Softplus(γ̃)≥0 以保证置信度单调)、CRFN-BY 的保形 p 值构造与 BY 校正、LyCon-WRL+ 的谱归一化 Lipschitz 上界与 Lyapunov 证书,并报告 91.4% 覆盖效率、4.1% FDR、95.2% 安全率(PPO 为 69%)与 23ms 端到端推理延迟。作者据此宣称这是首个把校准不确定性从预测经异常检测传播到安全策略学习并具备端到端理论保证的框架。
- 研究方法
- 研究把路网抽象为有向图 G=(V,E,A),节点特征包含流量、速度与密度;PU-GAT+ 以异方差高斯 NLL 训练,先以 PIT 直方图与可靠性图做校准诊断,再进入分裂保形预测:按历史误差统计(均值、标准差、偏度)对节点做 k-means 聚类,逐簇取保形分位数,并按 ACI 递推 α_(t+1)=α_t+γ_ACI(α−err_t) 自适应更新,校准时窗与测试时窗之间留出 τ_gap=24 步的间隔以缓解时间依赖;输入另行分解为趋势(滑动平均)与残差两流,合并不确定性中加入可学习相关系数 ρ=tanh(w_ρ)。异常检测走的是「不确定性归一化残差→条件归一化流→保形 p 值→BY 阈值」的链条:z=(y−μ̂)/(σ̂+ε),用条件 Real-NVP 建模 p_θ(z|c),上下文由邻居 CrossAttn 与因果 Transformer 编码,取负对数似然为异常分数,在截尾校准集 C_trim(τ=0.02)上构造 p 值,最后按 k·α/(m·c_m)、c_m≈ln m+0.577(m=293)阈值化;依赖结构用块自助法(B=1000 次复制,块尺寸 {5,10,20}×{1,2,4})检验,实测 ρ_block=0.34。控制端用 5 成员世界模型集成给出 μ_W 与 σ_W,在留出转移上度量 ε_model,以谱归一化在训练中强制 ‖W_ℓ‖_2≤1 得到 L̄_L 与 J̄_W,据此计算 ε*=(δ_slack+κ·d̄_C)/(L̄_L(1+J̄_W)) 并迭代验证 2–3 轮,Lyapunov 函数由 MLP 平方项与到安全状态的加权二次型构成,RL 状态包含局部排队/等待/吞吐、聚合预测与不确定性、聚合异常 p 值、异常标志与周期编码。实证材料为四套真实轨迹数据(T-Drive 293 格、GeoLife 156 格、Porto 441 格、Manhattan 263 普查区),真实事件标注取自市政记录(T-Drive 23 起、Porto 156 起、Manhattan 89 起),控制实验在 SUMO 的 4×4 网格 16 路口 NEMA 双环相位环境中进行,对照 STGCN、Graph WaveNet、AGCRN、STACI、RIPCN、Isolation Forest、OmniAnomaly、USAD、PPO、CPO、LAMBDA,并做逐组件消融(令 γ=0、以 BH 替换 BY、移除 Lyapunov 约束、δ 敏感性、参数化 p 值对照)与逐模块延迟分解。
- 学术贡献
- 在理论建构层面,论文给出温度缩放注意力无法区分邻居置信度的命题与证明(命题4.1),提出置信度单调注意力并证明其注意力比只由邻居间不确定性差决定、与源节点不确定性无关(命题4.3);进一步把保形分位数与 β-mixing 假设结合得到覆盖率界(定理4.5,含 O(Cρ^τgap/(1−ρ)) 与 O(γ_ACI) 两项),把保形 p 值与 BY 过程结合得到任意依赖下的 FDR 控制(定理4.8),并给出截尾校准保持超均匀性的引理(引理4.7)与有界模型误差下的安全性定理(定理4.14)及阈值 ε* 的显式形式。在方法创新层面,它以谱归一化在训练阶段强制 Lipschitz 上界而非事后估计,使 ε* 成为可核验的训练过程产物,并提出在保形包裹之前先做不确定性引导表征学习的思路。在材料与评估层面,FDR 评估使用市政事故、封路与拥堵记录作为真实异常标注(CRFN-BY 在真实事件上召回 74%、FDR 3.8%),而非仅依赖合成注入。在领域拓展层面,它为城市交通信号控制提供了一条预测—检测—控制一体化的实时管线(23ms,其中预测 8.2ms、流打分 5.1ms、BY 阈值 0.3ms、空间聚合 1.2ms、动作选择 8.2ms),并将覆盖效率在四套数据集上相对 STACI 的 8–15% 提升作为跨域泛化证据。
- 存在局限
- 控制实验只在 SUMO 的 4×4 网格、16 个路口上进行,作者自述规模有限,所得 95.2% 安全率与 92.3% Lyapunov 下降率能否外推到大规模路网未被检验。BY 只保证每个时间步的空间 FDR,跨时间聚合的 FDR 需另接在线 FDR 方法,材料对这一情形没有给出实验;而 BY 的调和修正(m=293 时 c_m≈6.4)以功效为代价,召回由 BH 的 0.76 降至 0.72。真实事件评估把市政记录之外的全部时空点视为原假设,作者承认官方记录本身含噪声,3.8%–4.1% 的 FDR 因而依赖标注协议而非独立真值。覆盖率保证建立在 β-mixing 与 τ_gap=24 步窗口间隔之上,且容忍项含 O(γ_ACI),但材料未给出 γ_ACI 与 ρ 的取值敏感性;作为主要比较指标的「覆盖效率」(Cov.Eff.)在材料中未明确给出定义。谱归一化提供的是训练过程强制上界,作者明确声明不等同于形式化验证级认证;迭代验证只报告 2–3 轮经验收敛而无收敛性证明,且 ε* 依赖政策相关的 d̄_C,安全性结论因此是条件性的;截尾校准的有效性还依赖 τ′≤τ,即校准数据取自已验证正常时段这一域假设。
- 进一步研究方向
- 其一,把控制实验扩展到 16 路口以上的路网乃至真实城市多区域信号控制部署,在同一边界条件下重新测量 ε_model 与 ε* 的大小关系,检验定理4.14 在更高维状态与更长回合下是否仍满足,并记录安全率与推理延迟的变化。其二,以在线 FDR 程序(如材料提及的 Ramdas 等、Javanmard & Montanari 一类)或图感知 FDR 过程、e-values 替换 BY 阈值化,在同一块自助依赖设定下比较跨时间聚合 FDR 与召回,量化 BY 保守性造成的功效损失。其三,以 τ_gap、γ_ACI、截尾比例 τ 为自变量做系统的敏感性实验,标定 O(Cρ^τgap/(1−ρ)) 与 O(γ_ACI) 两项的实际量级,并人为构造 τ′>τ 的污染校准集,观察 p 值偏离 α+0.04 界的方式。其四,将谱归一化得到的 L̄_L、J̄_W 与形式化验证方法(如 Fazlyab 等)给出的界在同一任务上直接对比,量化前者的保守程度及其随网络深度、宽度的变化。其五,为「覆盖效率」给出明确定义,并针对区间宽度与覆盖率之间的权衡做独立消融,使 PU-GAT+ 与 STACI、RIPCN 的比较可复核。
本综述由 AI(deepseek-flash)阅读开放获取全文自动生成,仅供参考,不代表原文作者观点,请以原文为准。
参考文献
相关论文
话题:Traffic Prediction and Management Techniques