[控场AI]
· 5 分钟阅读· 2,818 字

StochLOB:重尾得分匹配提升订单簿趋势预测稳健性

StochLOB:重尾得分匹配提升订单簿趋势预测稳健性

StochLOB用跳跃与截断稳定噪声替代布朗高斯腐蚀,提升订单簿预测的尾部稳健性。

StochLOB 针对扩散模型在限价订单簿预测中的高斯噪声假设缺陷,提出在训练阶段引入复合泊松跳跃-Lévy 噪声与截断 α-稳定噪声,以贴近订单流的重尾、突发和跳跃特性。其关键设计是腐蚀仅发生在训练阶段,推理时保持确定性预测器,无需反向扩散采样,部署复杂度不变。得分目标的径向因子可解释为后验期望腐蚀精度,对大残差按潜在腐蚀尺度加权,增强对极端事件的容忍度。在 Coinbase LOB 数据上的留存测试显示常规区间保持竞争力,尾部压力诊断指标改善。A 股迁移研究为探索性质,泛化能力有待独立验证。

当扩散模型遇上真实市场:布朗噪声的局限

得分匹配(Score Matching)与扩散模型在生成建模领域大放异彩,其经典设定是用布朗运动驱动的高斯噪声逐步腐蚀数据。然而,这套"轻尾"假设放到金融限价订单簿(Limit Order Book, LOB)场景下,问题就暴露出来了。订单流本身具有重尾、突发和跳跃特征——挂单与撤单常常成簇出现,价格偶尔发生离散跳变,这些都和高斯噪声的平滑、连续、尾部衰减快的性质相矛盾。

换句话说,用一个轻尾前向随机微分方程(SDE)去描述订单簿的破坏过程,本身就是一种失真的抽象。来自 rss 的这篇工作 StochLOB 正是从这个矛盾切入,提出了一套"仅训练阶段"的得分匹配框架:改变前向扩散的腐蚀规律,但部署时的预测器保持确定性。

rss source: StochLOB 论文页面

得分匹配与扩散模型:得分匹配的核心思想是学习数据分布对数密度的梯度场 (\nabla_x \log p(x)),即所谓"得分"。扩散模型则通过一个前向过程逐步向数据注入噪声(通常是布朗运动驱动的高斯噪声),将数据分布逐渐转化为易于采样的先验分布(如标准高斯),再训练一个神经网络学习反向去噪过程。经典 DDPM、Score-Based Generative Model 等都属于这一框架。其理论保证很大程度上依赖于高斯噪声的良好性质:连续路径、解析可解的前向分布、以及尾部快速衰减。一旦数据本身具有重尾或跳跃特征,这些假设就不再成立,模型需要"用错误的腐蚀过程"去学习数据分布,导致对极端样本的建模偏差。

限价订单簿(LOB):限价订单簿是交易所撮合系统的核心数据结构,记录某一时刻所有未成交的买单(bid)与卖单(ask)的价格与数量。订单流包含限价挂单、市价成交、撤单三类事件,其到达过程呈现明显的聚集性与重尾性:大单往往成批出现,撤单在行情剧烈波动时集中爆发,价格偶尔发生跳变。LOB 的统计特征因此与高斯假设格格不入,是检验生成模型鲁棒性的天然试验场。

两种非高斯噪声:跳跃与截断稳定分布

StochLOB 的核心改动在于噪声注入方式。它不再只用布朗高斯噪声,而是引入两种受非高斯 SDE 启发的替代方案:

复合泊松方差混合(jump-Lévy 噪声)。通过复合泊松过程构造跳跃结构,让训练样本中出现有限跳变,贴近订单流中真实的突发大单、瞬间撤单等事件。

截断 α-稳定噪声(positive-stable 高斯尺度混合)。α-稳定分布天然具备重尾特性,作者对其做截断处理以控制数值稳定性,从而在训练中注入重尾残差。

这两种腐蚀机制的关键工程价值在于:它们只在训练阶段起作用,推理时无需反向时间扩散采样。这意味着模型部署的复杂度和延迟不受影响,预测器依然是确定性的。

Lévy 过程与 α-稳定分布:Lévy 过程是一类具有独立平稳增量的随机过程,布朗运动是其特例。与布朗运动不同,Lévy 过程允许跳跃,复合泊松过程即是最简单的跳跃型 Lévy 过程——在随机时刻发生随机大小的跳变。α-稳定分布是广义中心极限定理的极限分布,其尾部按幂律衰减,方差(当 α<2 时甚至均值)可能不存在,因而能刻画极端事件频发的现象。α=2 时退化为高斯分布,α<2 时尾部显著变厚。由于 α-稳定分布方差无穷,直接用于神经网络训练会导致梯度爆炸,因此实际应用中常采用截断版本,即在超过某一阈值后截断尾部或改用轻尾分布,以兼顾重尾建模与数值稳定性。

得分目标的意外收获:径向因子即后验精度

StochLOB 提出的得分目标有一个耐人寻味的解释。其径向因子 (h_t(r)) 可以被理解为"后验期望腐蚀精度"——也就是说,当模型遇到一个大幅残差时,它会根据"最可能产生该残差的潜在腐蚀尺度"来加权,而不是一刀切地惩罚。

这种机制让模型对异常值更具容忍度,同时又不会完全忽略它们。对于订单簿预测而言,这一点尤其重要:市场中的极端事件(如闪崩、流动性骤降)往往是信息量最大的样本,但传统高斯假设下的损失函数会将其视为噪声而过度惩罚。StochLOB 的得分目标则提供了一种更贴合市场统计特性的加权逻辑。

后验期望腐蚀精度:在扩散模型中,前向过程通常写作 (x_t = \sqrt{\bar{\alpha}_t} x_0 + \sqrt{1-\bar{\alpha}_t} \epsilon),其中 (\bar{\alpha}_t) 控制信号保留程度,(1-\bar{\alpha}_t) 可视为腐蚀强度或噪声方差。当噪声分布非高斯时,给定观测残差 (r = x_t - \sqrt{\bar{\alpha}_t} x_0),其对应的潜在腐蚀尺度 (t) 不再是确定值,而是一个后验分布。径向因子 (h_t(r)) 正是对该后验分布下"有效腐蚀精度"的加权期望。直观上,当残差很大时,模型会推断它更可能来自一个腐蚀程度更高(噪声更强)的潜在状态,从而降低对该样本的惩罚权重;反之小残差则对应低腐蚀状态。这种自适应加权使损失函数对异常值具有鲁棒性,同时保留其信息量,与金融数据中"极端事件即信号"的直觉一致。

实测表现:正常行情稳健,尾部压力更优

作者在 Coinbase 的 LOB 数据上按时间顺序做了留存测试。结果显示,所提变体在未受腐蚀的 BTC 及山寨币测试区间上保持竞争力,没有因为改动噪声模型而牺牲常规性能。

更值得关注的是若干"已实现尾部窗口"的压力诊断指标得到改善。这说明模型在极端行情窗口下的表现更为可靠——对于风控和实盘交易而言,尾部稳健性的提升往往比平均精度的提升更有价值。

此外,论文还做了一项探索性的 Feishu A 股迁移研究,检验概率输出在交易成本和交易规则约束下是否仍然可用。这一部分被作者定性为"探索性",但它触及了一个真实痛点:模型在学术数据集上表现良好,不代表它在有涨跌停、T+1、手续费等约束的真实市场里依然可用。

从研究到落地的距离

StochLOB 提供了一个清晰的思路:与其在推理阶段做复杂的扩散采样,不如在训练阶段用更贴近真实市场统计特性的腐蚀过程来塑造模型。这种"训练重、推理轻"的设计对工程落地友好。

代码与配置文件已在 GitHub 开源(仓库名 Penny),对订单簿预测、高频交易信号、以及更广义的金融时序重尾建模感兴趣的读者可以进一步验证。

不过也要看到,论文的 A 股迁移部分仍是探索性质,样本和数据覆盖有限;不同交易所、不同资产类别的订单流统计特性差异较大,StochLOB 的两种噪声方案的泛化能力还有待更多独立复现来检验。

#StochLOB#得分匹配#限价订单簿#重尾分布#扩散模型#趋势预测#高频交易
分享:

相关推荐