---
title: OrderFlowDiT: Transport-Regularized Diffusion Transformer for Probabilistic Limit-Order Supply Forecasting
url: https://www.ml-quant.com/papers/arxiv/2610.22727/
site: ML-Quant (https://www.ml-quant.com)
updated: 2026-10-23
license: Summaries CC BY 4.0; links go to the original sources
index: https://www.ml-quant.com/llms.txt
identifier: arXiv:2610.22727
source_url: https://arxiv.org/abs/2610.22727
featured: 2026-10-23
citations: unknown
topic: Trading, Microstructure & Execution
---


# OrderFlowDiT: Transport-Regularized Diffusion Transformer for Probabilistic Limit-Order Supply Forecasting

OrderFlowDiT forecasts unnormalized bid-ask volume fields using unbalanced optimal transport regularization, reducing transport cost by 45.8 percent relative to mean squared error alone.

- Source: https://arxiv.org/abs/2610.22727
- Identifier: arXiv:2610.22727
- Released: 2026-10-19
- First featured: Quant Letter No. 136 (2026-10-23): https://www.ml-quant.com/issues/2026-10-23/
- Citations (Semantic Scholar): not tracked
- Published in: not yet
- Topic: Trading, Microstructure & Execution
- Authors: Jung Hun Phee, Tucker Balch

## Abstract (arXiv, CC0)

We study the conditional distribution of limit-order volume submitted over the next five minutes, given the current order book and recent order flow. The task requires capturing price placement, total volume, and predictive uncertainty jointly. Coordinate-wise diffusion losses do not explicitly account for price displacement, while normalizing fields to unit mass removes information about total volume. Agreement on selected summary statistics also does not establish fidelity to the joint field distribution. We introduce OrderFlowDiT, a conditional Diffusion Transformer for unnormalized bid and ask volume-at-price fields, trained with an unbalanced optimal-transport (UOT) auxiliary loss. We also develop a full-field evaluation framework that combines a transport cost sensitive to price placement and total volume with proper scoring rules and calibration diagnostics. In the primary comparison on 1950 validation conditions across 30 U.S. equities, adding UOT reduces full-field transport cost by 45.8% and the transport energy score by 26.1% relative to training with mean squared error alone. Both measures improve for every stock. These findings support incorporating price geometry and volume into generative modeling, with field accuracy and predictive uncertainty evaluated together.

## Related

- [Detecting Multilevel Manipulation from Limit Order Book via Cascaded Contrastive Representation Learning](https://www.ml-quant.com/papers/arxiv/2508.17086/): The study suggests a learning framework to enhance the detection of trade-based manipulation in financial markets, with Transformer-based architectures proving most successful.
- [Deep attentive survival analysis in limit order books: estimating fill probabilities with convolutional-transformers](https://www.ml-quant.com/papers/arxiv/2306.05479/): A deep learning method outperforms other approaches in estimating filltimes of limit orders.
- [Event History Over Scale: Compact Transformers for Low-Latency Limit Order Book Forecasting](https://www.ml-quant.com/papers/arxiv/2610.02917/): MBOFormer and MBOFusion, two causal transformers with under 70 kilobyte serialized state, predict limit order book price trends with sub-millisecond latency across three markets and four prediction horizons.
- [Painting the market: generative diffusion models for financial limit order book simulation and forecasting](https://www.ml-quant.com/papers/arxiv/2509.05107/): A new method for simulating financial market data transforms limit order book data into an image format and uses diffusion models to predict future states, showing top performance on LOB-Bench.
- [Bimodal dynamics of the artificial limit order book stock exchange with autonomous traders](https://www.ml-quant.com/papers/arxiv/2508.17837/): The paper uncovers the inherent bistability and complex dynamics of an artificial stock market exchange, which emerge from micro-level trading rules.
- [Reinforcement learning for trade execution with market and limit orders](https://www.ml-quant.com/papers/arxiv/2507.06345/): The paper presents a reinforcement learning framework for optimal trade execution, using multivariate logistic-normal distributions, which outperforms traditional benchmark strategies.