
Kronos: A Foundation Model for the Language of Financial Markets
Key Points
- 1Kronos introduces a novel foundation model specifically designed for financial K-line data, addressing the limitations of general Time Series Foundation Models in this complex domain.
- 2It employs a specialized hierarchical tokenizer to discretize continuous market information into dual-component tokens and undergoes autoregressive pre-training on a vast corpus of over 12 billion K-line records from 45 global exchanges.
- 3Demonstrating superior performance, Kronos sets new benchmarks across diverse quantitative finance tasks, including price forecasting, volatility prediction, and synthetic data generation, showcasing its robust and versatile capabilities.
Kronos is a novel foundation model specifically designed for the language of financial markets, addressing the limitations of general-purpose Time Series Foundation Models (TSFMs) when applied to K-line data. The paper highlights that K-line sequences exhibit unique statistical properties, such as low signal-to-noise ratios, strong non-stationarities, and intricate high-order dependencies among OHLCVA (Open, High, Low, Close, Volume, Amount) attributes, which are often misaligned with generic TSFM inductive biases. Kronos aims to overcome these challenges by introducing a specialized, unified, and scalable pre-training framework.
The core methodology of Kronos operates in two distinct phases: (1) K-line Tokenization and (2) Autoregressive Pre-training.
1. K-line Tokenization:
This phase transforms continuous, D-dimensional K-line inputs , where each (with for OHLCVA attributes), into a corresponding sequence of discrete tokens . This is achieved using a Transformer-based autoencoder composed of an encoder , a quantizer , and a decoder .
- Quantization Mechanism: Kronos adapts Binary Spherical Quantization (BSQ), a variant of Look-up Free Quantization (LFQ), to quantize a continuous latent vector into a -bit binary code .
- Hierarchical Token Structure: To manage the exponentially large vocabulary size () associated with large , each -bit token is factorized into two subtokens: a coarse subtoken and a fine subtoken . These subtokens have equal bit lengths, , such that . This decomposition transforms a single prediction over entries into two sequential predictions over entries, significantly reducing computational and parameter complexity.
- Tokenization Objective: The tokenizer is trained with a composite objective that enforces a coarse-to-fine structure:
- : This term encourages the coarse subtoken to capture the principal, low-fidelity structure of the input by minimizing the reconstruction error when only the coarse subtoken is used by the decoder.
- : This term ensures high-fidelity reconstruction using the complete token . By optimizing after , the fine-grained subtoken is implicitly guided to encode the residual information necessary to refine the coarse approximation.
- : This is a quantization loss derived from BSQ, regularizing the learning process by penalizing the distance between continuous latent vectors and their binary codes , promoting stable training and alignment with the learned codebook.
2. Autoregressive Pre-training:
In this phase, a decoder-only Transformer, denoted , is pre-trained on the tokenized sequences using a standard next-token prediction objective. employs causal attention to ensure predictions at each time step depend exclusively on historical context.
- Hierarchical Autoregressive Modeling: Given the hierarchical token design, the conditional probability for each token is decomposed using the chain rule to explicitly capture the coarse-to-fine dependency:
This factorization allows the model to first predict the coarse-grained subtoken, which then serves as a condition for generating the fine-grained residual subtoken.
- Input Embedding: For each time step , the coarse subtoken and fine subtoken are independently projected into vector representations using two distinct embedding layers, yielding and . These embeddings are then concatenated and linearly projected to produce a fused input vector:
- Prediction Process: The sequence of fused input vectors is processed by , yielding contextualized hidden states. The final hidden state (from processing ) is used for prediction.
- Coarse Subtoken Prediction: The hidden state is projected by a linear head to produce logits for the coarse subtoken distribution:
- Fine Subtoken Prediction: To model the conditional dependency, the context is updated with the *predicted* coarse subtoken . This is achieved via a cross-attention mechanism where the embedding of acts as the query, and provides the key and value:
Then, the updated hidden state is projected by a second head to predict the fine subtoken:
.
During training, is sampled from the predicted distribution (rather than using the ground truth) to mitigate exposure bias and align training with multi-step autoregressive inference.
- Training Objective: The overall autoregressive training objective is the negative log-likelihood of the observed sequence, summed over both prediction steps:
Kronos is pre-trained on a massive, diverse financial corpus of over 12 billion K-line records from over 45 global exchanges and 7 temporal granularities, enabling it to learn robust and generalizable market representations. This unified discretize-and-generate paradigm allows Kronos to construct a high-fidelity, hierarchical representation of market dynamics, serving as a robust foundation for various downstream quantitative finance tasks.