项目详情

Kronos

首个开源金融 K 线基础模型,学会市场的语言

GitHub 地址 MIT PythonPyTorch

更新于 2026-07-23

← 返回汇总
项目速览

一句话定位

首个开源金融 K 线基础模型,学会市场的语言

首个开源金融 K 线(candlestick)基础模型。专用 tokenizer 把连续 OHLCV 数据量化成层级离散 token,再用自回归 Transformer 在 45+ 全球交易所、120 亿+ K 线记录上预训练。零样本即可胜任价格预测、波动率预测、合成数据生成多种量化任务。价格预测 RankIC 比头部 TSFM 高 93%,波动率 MAE 低 9%,合成保真度 +22%。清华+京东出品,AAAI 2026 录用,32.7k Stars。

项目速览

核心数据

32.7k
GitHub Stars
12B+
K 线训练记录
45+
全球交易所
93%
RankIC 提升

decoder-only Transformer · 四档模型(4.1M–499.2M)· Hugging Face 开源 · 含微调脚本 · 作者含清华张长水、李建,京东徐伟

为什么存在

金融时序建模的三重困境

通用 TSFM 不懂金融

通用时序基础模型(TimesFM/Moirai 等)针对平稳信号设计,在金融高噪声、肥尾、非平稳数据上表现不佳,常输给非预训练架构。

任务模型各自为政

价格预测、波动率预测、合成数据生成各训各的模型,无法共享表征,数据利用率低。

LLM 读不懂价格

GPT/Claude 擅长新闻情感,却无法直接处理 OHLCV 价格序列——语言模型与数字信号存在鸿沟。

核心理念

市场数据也是一种"语言"

把 LLM 的"预测下一个 token"范式移植到 K 线——前提是先把连续价格"词汇化"。Kronos 的关键洞察:金融市场的语法藏在 OHLCV 的演化模式里。

原始 OHLCV 专用 Tokenizer 离散 token 序列 自回归 Transformer 预测下一个 token

就像 GPT 预测句子下一个词,Kronos 预测价格序列下一个 token。预训练让模型学会跨市场、跨资产的通用时序表征。

核心架构

两阶段预训练框架

Stage 1
专用 Tokenizer
将连续、多维 K 线数据(OHLCV)量化为层级离散 token。同时保留价格动态与交易活跃度两类模式。提供 2K 与 base 两种词表规格。
Stage 2
自回归 Transformer
在 token 序列上用 next-token 预测目标预训练。decoder-only 架构,学到的表征可迁移到下游多种量化任务,无需重新设计模型。
性能基准

零样本碾压

任务对比基线Kronos 提升
价格序列预测头部通用 TSFMRankIC +93%
价格序列预测最佳非预训练架构RankIC +87%
波动率预测MAE -9%
合成 K 线生成生成保真度 +22%

全部为零样本(zero-shot)设置——无需在目标数据上微调即可达到。RankIC 是量化选股的核心指标,衡量预测值与真实收益的排序相关性。

模型族

四档容量,按需选择

模型参数量上下文长度Tokenizer开源
Kronos-mini4.1M20482K
Kronos-small24.7M512Base
Kronos-base102.3M512Base
Kronos-large499.2M512Base

mini:边缘部署/移动端,2048 上下文看更长历史。
small:消费级 GPU 可跑,多数场景均衡之选。

base:生产级量化工作。
large:未开源,疑似留给商业合作。模型均在 Hugging Face NeoQuasar/ 命名空间下。

快速上手

三步出预测

# 1. 装依赖(Python 3.10+)
pip install -r requirements.txt
# 2. 加载模型与 tokenizer
from model import Kronos, KronosTokenizer, KronosPredictor
tokenizer = KronosTokenizer.from_pretrained("NeoQuasar/Kronos-Tokenizer-base")
model = Kronos.from_pretrained("NeoQuasar/Kronos-small")
predictor = KronosPredictor(model, tokenizer, max_context=512)
# 3. 预测(df 含 open/high/low/close 列,volume 可选)
pred_df = predictor.predict(
    df=x_df, x_timestamp=x_ts, y_timestamp=y_ts,
    pred_len=120,
    T=1.0,       # 温度采样
    top_p=0.9,    # 核采样
    sample_count=1  # 多路径平均
)

KronosPredictor 封装了归一化、tokenization、逆变换全流程。另有 predict_batch 支持多资产并行预测,利用 GPU 并行加速回测。

能力边界

不只是预测

价格预测

零样本预测未来 K 线 OHLCV。支持概率预测——多次采样生成分布,而非单点。

波动率预测

预测未来价格波动幅度,MAE 比基线低 9%。期权定价与风险管理关键输入。

合成数据生成

生成保真度提升 22% 的仿真 K 线。用于策略回测增广、隐私保护数据共享。

三种任务共享同一套预训练权重——这是"基础模型"范式的核心价值:一次预训练,多任务复用。

适用场景

谁该用 Kronos

微调与生态

适配你的市场

2025 年 8 月开放微调脚本。在预训练权重上继续训练,让模型适配特定资产类别或市场状态。

风险提示

必须清醒

潜力评估

未来空间

Domain-specific Foundation Model 是 AI 的下一 frontier,Kronos 证明了金融领域的可行性。32.7k stars + AAAI 2026 录用,说明学界与工业界都认可其范式价值。两阶段框架(专用 tokenizer + 自回归预训练)可迁移到任何"连续信号词汇化"场景——能源、气象、工业传感。若微调生态成熟 + large 模型择机开源,有望成为金融 AI 的"BERT 时刻"——一个所有量化工作都绕不开的基座。局限在于:金融市场的有效性意味着 alpha 衰减极快,模型能力的"半衰期"可能短于通用领域。


上一个
OmniRoute
下一个
croc
1 / 14