
通用时序模型很难适配金融市场高噪声、非平稳的数据特性,Kronos作为全球首批开源K线专属基础模型,把行情序列转化为可被AI识别的“市场语言”,一套框架覆盖多市场走势推演、量化信号实验,大幅降低量化研究的建模门槛。

Kronos是开源解码器式金融基座模型,论文已收录AAI 2026会议,训练素材覆盖全球45家以上交易所的行情数据。区别于通用时间序列工具,它独创分层K线分词架构,专门处理OHLCV多维行情序列,完整配套预测、微调、回测整套开源代码,GitHub热度持续走高,当前Star数量突破3.4万。
常规时序工具做行情推演时短板明显,Kronos针对性优化,对比表格如下:
现存建模痛点 | Kronos优化方案 |
|---|---|
通用模型不懂K线结构,易被短期杂波干扰 | 分层BSQ分词,分离大势粗信号与短线细波动 |
一套模型无法兼容股票、加密货币等多品类行情 | 统一token范式,更换数据源即可直接推理 |
从零搭建训练、回测流程耗时极久 | 内置Qlib配套微调与回测完整脚本 |
大体积模型普通设备无法运行 | 开放mini/small/base轻量化权重,低配机器可部署 |
单次只能单条资产推演,批量实验效率低 | 封装batch批量预测接口,GPU并行提速 |
整套系统分为两大核心模块,逻辑清晰易懂:
项目开放三款可商用开源权重,大尺寸large模型暂未开源,参数与适配场景整理如下:
模型名称 | 参数量 | 上下文长度 | 硬件适配场景 |
|---|---|---|---|
Kronos-mini | 4.1M | 2048 | 普通笔记本CPU、轻量化快速实验 |
Kronos-small | 24.7M | 512 | 家用游戏显卡,日常单资产推演 |
Kronos-base | 102.3M | 512 | 中端GPU,批量多标的回测实验 |
Kronos-large | 499.2M | 512 | 未开源,机构专业量化训练 |
所有开源模型均可从Hugging平台一键拉取权重,无需本地重新预训练。
环境最低要求Python3.10及以上版本,全程命令行操作,无复杂编译流程:
git clone https://github.com/shiyu-coder/Kronos
cd Kronos
pip install -r requirements.txt
# 如需A股微调回测额外安装qlib
pip install pyqlib
python examples/prediction_example.py
Windows、Linux、macOS系统均可兼容,无系统专属限制,低配设备优先选用mini模型降低算力消耗。
仅需少量代码即可完成历史K线推演未来行情,完整可运行示范:
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)
# 载入本地行情csv(需包含open/high/low/close列)
import pandas as pd
df = pd.read_csv("stock_5min.csv")
# 设置回溯窗口与预测周期
lookback = 400
pred_len = 120
# 生成未来K线预测数据
pred_df = predictor.predict(
df=df.loc[:lookback-1],
pred_len=pred_len,
T=1.0, top_p=0.9, sample_count=1
)
# 输出预测开盘、收盘、成交量数据
print(pred_df[["open","close","volume"]].head())
工具自动完成数据归一、反归一流程,无需手动处理数值缩放,批量多资产调用可使用predict_batch函数提升效率。
内置基于Qlib的完整微调流水线,适配国内个股行情,分四步执行:
finetune/config.py,配置本地qlib数据路径、模型保存文件夹;python finetune/qlib_data_preprocess.py
# 分词器微调
torchrun --standalone --nproc_per_node=2 finetune/train_tokenizer.py
# 主模型微调
torchrun --standalone --nproc_per_node=2 finetune/train_predictor.py
python finetune/qlib_test.py --device cuda:0
微调完成后输出的预测信号仅作研究素材,不能直接作为交易依据。
依托模型输出的量价推演信号,搭建Top-K选股、波段跟踪等策略,快速验证历史周期内策略收益表现,缩短策略迭代周期。
股票、加密货币、海外股指统一一套推理代码,无需针对不同市场重新搭建建模逻辑,适合多品种同步研究。
轻量化mini模型算力门槛低,高校金融、计算机专业可用于时序AI、量化相关课题实验,配套完整论文与开源代码便于复现实验结论。
项目自带BTC/USDT实时线上演示页面,输入历史区间自动生成未来走势预览,可嵌入内部教学演示页面。



Kronos依靠专属K线分词架构、多档位轻量化开源权重、完整微调回测流水线,补齐通用AI在金融时序领域的短板,给量化研究者、在校科研人员提供低成本建模工具。使用者可根据设备算力挑选适配模型,结合内置示例快速开展行情推演与策略实验,同时需理性区分模型研究结论与真实市场交易的边界。
#Kronos #金融AI #量化开源项目 #K线预测 #大模型教程 #长鑫科技
注:本文仅作为 开源项目AI 技术、量化工具及数据分析 的科普探讨分析,所涉代码、模型及数据仅供学习交流。绝不构成任何投资建议、个股推荐或操作指导。市场有风险,投资需谨慎,请勿仅凭本文内容做出交易决策。