news 2026/9/12 6:24:53

Kronos:开源K线预测基础模型,回测表现到底如何?

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
Kronos:开源K线预测基础模型,回测表现到底如何?

Kronos:开源K线预测基础模型,回测表现到底如何?

【免费下载链接】KronosKronos: A Foundation Model for the Language of Financial Markets项目地址: https://gitcode.com/GitHub_Trending/kronos14/Kronos

假设是某个交易日收盘,你盯着一只股票的K线图,想知道接下来几根K线大概是什么形状。开源K线预测模型 Kronos 做的就是这件事:输入一段历史K线数据(开盘、最高、最低、收盘、成交量),直接输出未来若干周期的K线序列。它给的不是“看涨还是看跌”的模糊判断,而是可以画成图的 OHLCV 数值。

它是什么:面向K线序列的开源基础模型

Kronos 是首个面向金融 K 线序列的开源基础模型,在超过 45 个全球交易所的数据上预训练,相关工作已被 AAAI 2026 收录。一句话概括它的流程:输入一段历史K线 → 模型编码后自回归预测 → 输出未来若干周期的K线(含价格与成交量)。

模型按参数量分了几个档位:mini 4.1M、small 24.7M、base 102.3M,均开源可下载,显存不大的设备也可以先跑推理。

Kronos 怎么工作:K线数据先“装箱”,再逐根预测

核心分两步。

第一步是编码(K线数据Token化)。连续数值对神经网络并不友好,Kronos 专门配了一个 tokenizer:它先给每根K线一个“粗粒度 token”,概括这根K线的整体价格位置和量能档位,再在其中挂上“细粒度子 token”,保留价格与成交量的细节。可以类比收拾行李箱:主隔层放“这属于哪一类”,细网袋装“具体偏在哪里”。这套双粒度设计基于二值球量化(BSQ)技术,在明显压缩数据维度的同时保住关键信息。

第二步是自回归预测。Token 序列被送进因果 Transformer(decoder-only 结构),逐根预测下一根K线。有点像写连环画:每一格只能看前面已经画好的格子,不能偷看后面。这和真实交易场景一致——任何时刻只能看到当下为止的历史,模型也不能“作弊”。核心实现可以顺着 model/kronos.py 看,想抠细节从这里入手。

效果怎么看:Kronos 股票预测回测的三个指标

仓库里给了一个回测示例:用 A 股数据跑简单的 Top-K 策略,曲线如下(对应 finetune/ 目录的流水线)。看回测图其实就看三件事:

指标含义怎么读
累计收益策略随时间的累计盈亏与基准曲线(沪深300)对比,持续在上方是第一标准
超额收益策略跑赢基准的部分稳步抬升说明超额是持续的,而不是某段行情带来的
扣费后收益计入交易成本后的结果图里同时画了含成本与不含成本两条线,扣费后仍站得住才更可信

单只股票的预测效果可以看这张图:蓝线是实际收盘与成交量,红线是模型预测值。

看这张图有两点值得注意。其一,这是某只沪市股票 5 分钟K线上、向前预测 120 个周期的结果。其二,红线没有贴住蓝线——模型捕捉的是趋势的整体形状和成交量峰值的位置,而不是逐点对齐。如果哪里给出红线完全压住蓝线的“预测”,反而要打个问号。

3步跑通Kronos第一个预测

第一步,克隆并安装依赖:

git clone https://gitcode.com/GitHub_Trending/kronos14/Kronos cd Kronos pip install -r requirements.txt

第二步,准备数据。模型需要一个标准K线 CSV,包含 timestamps、open、high、low、close、volume、amount 七列;没有成交量可以填 0。格式样例可对照 finetune_csv/data/HK_ali_09988_kline_5min_all.csv。

第三步,跑示例脚本:

python examples/prediction_example.py

它会输出一张历史K线与预测K线上下对照的图。

如果卡住了:先看 README.md 的 Getting Started 一节,示例脚本的完整逻辑有逐步说明;数据格式问题查 finetune_csv/README_CN.md;显存有限就从 Kronos-mini 起步,4.1M 参数、上下文 2048,跑通第一次体验足够。

不止预测价格:高频K线微调和批量并行

高频 5 分钟K线。finetune_csv/ 目录提供在单个 CSV 上微调的完整流水线,仓库内附了港股阿里巴巴(09988)5分钟K线的全量示例:按配置训练后,模型就能在 5 分钟级别上输出预测,下图是示例产出的回测图之一。

批量并行。预测器提供predict_batch接口,把多只股票的序列放进列表一次并行预测(走 GPU 并行),适合“把一篮子股票都扫一遍”的场景。examples/yuce/ 目录下还有多只个股的完整分析案例(如 000021),每个案例同时输出K线预测图、成交量预测和市场因素评分的 JSON 报告,方便二次开发成自己的信号。

微调适配。想让模型专门针对某个市场或某只标的,finetune/ 提供基于 Qlib 的四步流水线:数据预处理 → 微调 tokenizer → 微调 predictor → 回测评估,训练只需两条torchrun命令,不用改模型代码。

适合谁用、边界在哪里

适合谁:想搞懂K线序列模型架构的从业者;想给技术分析之外多一个“第二意见”的投资者;以及想把 AI 预测组件嵌入既有策略的开发者。

已知局限,仓库自己也讲得很直白:

  • 效果依赖历史数据质量。数据缺失、噪音大,输出会明显漂移。
  • 回测收益不等于实盘收益。README 明确说明微调流水线只是演示,并非生产级系统;简单 Top-K 回测也没有完全建模滑点和市场冲击。
  • small/base 模型的上下文上限是 512 根K线,输入窗口超过会被自动截断。
  • 预测来自概率采样(温度、top_p 可调),输出是一个区间估计,不是确定值。

一句话:它是分析K线结构的工具,不是印钞机。

写在最后

Kronos 的价值,在于把“K线预测”这件以往只能靠黑盒商业系统的事,变成了可复现、可开源、可自己动手验证的事——模型、数据格式、微调流水线全部公开。

第一步很简单:克隆仓库,备好一份K线 CSV,运行examples/prediction_example.py。看到那条红色预测曲线之后,剩下的都是调参问题。

【免费下载链接】KronosKronos: A Foundation Model for the Language of Financial Markets项目地址: https://gitcode.com/GitHub_Trending/kronos14/Kronos

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/9/12 6:24:27

10 分钟把 RTSP 摄像头接进低延迟流媒体:go2rtc 新手完整教程

10 分钟把 RTSP 摄像头接进低延迟流媒体:go2rtc 新手完整教程 【免费下载链接】go2rtc Ultimate camera streaming application 项目地址: https://gitcode.com/GitHub_Trending/go/go2rtc go2rtc 是一个 Go 写的摄像头流媒体程序,一路 RTSP 进来…

作者头像 李华
网站建设 2026/9/12 6:23:21

Gitee研发一体化选型实战:从代码托管到CI/CD的完整闭环

/* MD / 富文本中的 .toc(含博客园搬家等嵌套结构);.toc-box 在侧栏,不受影响 */#content_views .toc,/* 编辑器常在目录前后插入空 p(:empty 仍占 20px),一并去掉避免顶空隙 */#content_views.markdown_views > p:empty:has(+ .toc),#content_views.markdown_views …

作者头像 李华
网站建设 2026/9/12 6:22:22

ESP32-S3 N16R8嵌入式开发实战:PlatformIO工程化与工业级项目结构

1. 为什么选ESP32-S3 N16R8?不是参数堆砌,而是真实开发场景的“够用省心”刚拿到那块印着“ESP32-S3-N16R8”的小板子时,我第一反应不是看数据手册,而是把它插进电脑——USB口一亮,设备管理器里直接跳出一个“Silicon …

作者头像 李华
网站建设 2026/9/12 6:22:12

STM32F103用USART1+TIM2驱动DHT11单总线协议

简介:本资源是一套基于STM32F103VET6微控制器实现DHT11温湿度传感器单线通信的完整嵌入式开发工程,面向嵌入式初学者、课程设计学生及STM32入门开发者,解决MCU与数字传感器协议级对接难、时序控制不精准等典型实践痛点,适用于智能…

作者头像 李华
网站建设 2026/9/12 6:21:09

teamai-cli:构建团队级AI命令行接口的实战指南

1. “teamai-cli”不是新工具,而是开发者对CLI生态焦虑的具象化投射最近在多个技术社区和内部协作群中,频繁刷到“teamai-cli”这个关键词——它既没出现在npm官方registry的热门包榜单里,也没在GitHub上拥有超过50星的独立仓库,更…

作者头像 李华
网站建设 2026/9/12 6:19:44

研究生论文写作工具对比:千笔与学术猹评测

1. 研究生论文写作工具对比:千笔写作工具与学术猹深度评测作为一名经历过研究生阶段的科研狗,我深知论文写作过程中查阅文献、整理笔记、规范引用的痛苦。今天要评测的两款工具——千笔写作工具和学术猹,都是近年来在研究生群体中口碑不错的神…

作者头像 李华