核心摘要

Kronos 是首个面向金融 K 线数据的开源基础模型,论文被 AAAI 2026 接收。核心创新是两阶段架构:

  1. 专用分词器:将连续 OHLCV 数据量化为层次化离散令牌,捕获金融数据高噪声特性
  2. 自回归 Transformer:在这些离散令牌上进行预训练,服务于多种量化任务

模型系列

模型参数量上下文开源
Kronos-mini410 万2048
Kronos-small2470 万512
Kronos-base1.023 亿512
Kronos-large4.992 亿512

关键特性

  • 跨市场预训练:覆盖全球 45+ 交易所 K 线数据
  • 概率性预测:通过温度(T)和 top-p 采样支持多路径预测
  • 批量并行:支持多个资产同时预测(GPU 并行)
  • 端到端微调:提供 Qlib 数据准备 → 微调 → 回测完整流水线
  • 在线演示:BTC/USDT 未来 24 小时预测可视化 Demo

技术栈

Python + PyTorch,支持 torchrun 分布式训练,数据接口兼容 Qlib,实验管理支持 Comet.ml。