关于ZAKER 合作
量子位 昨天

2 张 4090 竟能本地微调万亿参数 Kimi K2!趋境联合清华北航把算力门槛击穿了

微调超大参数模型,现在的 " 打开方式 " 已经大变样了:

仅需2-4 张消费级显卡(4090),就能在本地对DeepSeek 671B乃至Kimi K2 1TB这样的超大模型进行微调了。

你没有看错。

这要放以前啊,各路 " 炼丹师 " 是想都不敢这么想的。因为按照传统的方法,类似 Kimi K2 1TB 参数的模型,用 LoRA 微调方案理论上需要高达 2000GB 的显存,而即便是参数量稍小的 DeepSeek-671B 的模型微调也需要 1400G 的显存。

什么概念?

一张 H100(80GB)得十几张起步,说是吞矿也是不足为过了。

而现在微调千亿 / 万亿参数模型的成本能打如此骨折,背后的关键源自两个国产明星项目的联动。

首先就是KTransformers,是由趋境科技和清华 KVCache.AI 共同开源的项目,GitHub 已经斩获 15.3K 星。

KTransformer 此前在大模型推理领域就已声名鹊起,凭借 GPU+CPU 的异构推理的创新路径成为主流推理框架之一,通过 KTransformers 利用单张 4090 可以推理 Kimi K2 1TB 级别大模型。

而这一次,KTransformers 已经支持 LoRA 微调,同样是 Kimi K2 1TB 这样参数的模型,仅 90G 左右的显存即可;微调参数量稍小的 DeepSeek 671B 也仅需 70G 左右的显存。真 · 把成本给打下去了。

另一个国产明星项目,则是LLaMA-Factory,在 GitHub 的星标数超 6 万。它是一个简单易用且高效的大语言模型训练与微调平台,让用户无需编写代码,即可在本地完成上百种预训练模型的微调。

它俩的联动模式是这样的:

LLaMA-Factory 是整个微调流程的统一调度与配置框架,负责数据处理、训练调度、LoRA(Low-Rank Adaptation)插入与推理接口管理。

KTransformers 则作为其可插拔的高性能后端,在相同的训练配置下接管 Attention / MoE 等核心算子,实现异构设备的高效协同。

这时候或许有小伙伴要问了,把 KTransformers 换成其它类似的推理框架不行吗?

答案是,真不行。

例如我们把 KTransformers、HuggingFace 和 Unsloth 三种后端的 LoRA 微调方案放一起比较下效果。

结果显示,KTransformers 为超大规模的 MoE 模型(Kimi K2 1TB 等)提供了 4090 级别的唯一可行方案,并在较小规模的 MoE 模型(DeepSeek-14B)上面也展现了更高的吞吐和更低的显存占用。

嗯,KTransformers 可以说是硬生生把微调超大模型的门槛,从数据中心级拉到了个人工作站级了,而且速度极快。

虽然成本是打下来了,但下一个问题是——效果会不会也打折?

用骨折的开销自定义千亿 / 万亿大模型

大模型用在专业领域的时候,往往令人头疼的一个点就是" 懂得多≠懂得精 ",这就是微调要解决的问题。

而正所谓实践是检验真理的唯一标准,效果打不打折,实测说了算。微调 Kimi K2 1TB 模型需要 90G 左右显存 +2T 左右的内存,微调 DeepSeek 671B 模型需要 70G 左右显存 +1.5T 左右的内存。以下用 DeepSeek 671B 模型为测试样例来看微调效果:

第一个测试的例子,是让 DeepSeek 在微调之后,生成的文字可以有喵娘(CatGirl)效果的语气。

数据集采用的是 NekoQA-10K,是一种面向猫娘语言建模的对话数据集,来看下效果:

微调前,若是提问一个问题:

我舌头坏了怎么办,吃什么柠檬都是酸的。

模型的回答是冷冰冰的 AI 味:1、保持口腔卫生…… 2、避免刺激性食物…… 3、避免酸性食物……

Emmm ……着实是莫得感情。

而微调后,模型的答案变成了:

主人舌头不舒服吗?宝宝好担心喵!(耳朵耸拉下来)柠檬酸是因为柠檬里面有柠檬酸啦,这是正常的喵 ~"。

微调前后,简直判若两 AI,是我们想要的喵味十足的那种。

当然,不止是变喵娘这种整活儿,在严肃的专业领域,KTransformers 的微调能力同样能打。

接下来的测试,采用了非洲医疗数据集(AfriMed-QA),包含医疗选择题和简答题,是垂直领域微调的试金石。

在这些具有明确风格和知识需求的任务上,经过 KTransformers 后端 LoRA 微调的模型,各项评测指标(如 BLEU、ROUGE、Accuracy)均获得了大幅提升。

这些个例子还都是开胃菜,微调背后真正有意思的,是开启了 AI 的个性化定制时代。

以前我们用大模型,基本上就是 " 模型有啥,你用啥 ";但现在,成本打下来了之后,玩法就多了去了:

定制你的专属风格:不只是喵娘,你可以用自己的聊天记录、邮件、文档去微调,打造一个 " 你风格 " 的写作助手,让它帮你回邮件、写周报,口吻跟你一模一样。

打造私有知识库助手:把公司内部的 SOP、技术文档、法律条文喂给它,微调出一个只为你公司服务的 AI 专家,问啥都懂,而且数据不出本地,绝对安全。

创造灵魂伴侣: 把你喜欢的某个角色、某位历史人物的语料丢进去,微调一个能随时随地和你角色扮演的聊天机器人。

深入垂直领域: 就像原稿里提到的,用专业数据集(比如医疗、法律)微调,模型在特定领域的表现会暴涨。这在严肃的专业领域同样有用。

这意味着,大模型不再是少数机构的专属技术,而成为高校、团队乃至个人都能驾驭的创意工具。算力门槛的消失,让更多垂直需求与独特想法得以实现,从而催生前所未有的应用创新。

对企业而言,KTransformers 带来的低成本微调能力,也为落地大模型提供了新选项:

企业不再需要赌一个大而全的通用模型,而是可以快速在多个业务方向上进行测试,用私有数据喂出一个个懂自家业务的 AI 专家——无论是客服、营销还是内部知识管理,迭代效率和投资回报率都远超以往。

这,才是低成本微调的真正魅力:它让大模型从一个高高在上的全知全能工具,变成了每个人、每个企业都能随心定制的专属生产力。

用起来也是超方便的

而且啊,KTransformers 和 LLaMA-Factory 联动之下,操作方式也变得非常简单。

趋境科技为 KTransformers 本次封装了 wheel 包,避免本地编译,安装极简。

你只需同时安装 KTransformers 和 LLaMA-Factory 环境,把 use_kt 设置为 true,并指定相应的 kt_optimize_rule YAML 文件,然后像往常一样启动 LLaMA-Factory 的训练命令。

LLaMA-Factory 会自动负责所有的数据处理、训练调度、LoRA 插入。而 KTransformers 则会作为即插即拔的高性能后端,在底层默默接管所有 Attention 和 MoE 的核心算子,实现 GPU+CPU 的高效协同。

若是用一个公式来总结二者的联动,或许可以是:

底层极致性能(KTransformers)+ 上层易用性(LLaMA-Factory)= 微调界的平民法拉利

至于背后的原理,我们可以简单总结为一套组合拳:

第一拳:把最重的包袱甩给 CPU。 MoE 模型最吃显存的专家层,KTransformers 直接让 CPU 内存来扛。GPU 解放出来专心算它擅长的。结果:671B 的模型,显存占用从 1400GB+ 理论值,硬是被压到了 70GB!

第二拳:LoRA 和高性能算子无缝合体。 简单说,它搞了个新设计,让你在享受 KTransformers 极致速度的同时,还能无缝插入 LoRA 微调,两边的好处都占了。

第三拳:榨干 CPU。 甩给 CPU 的任务,也不是让它摸鱼。KTransformers 集成了 Intel AMX 指令集,让 CPU 处理 AI 运算也猛得一批。

KTransformers 背后的团队——趋境科技与清华 KVCache.AI,值得再次被提及。

趋境科技在异构推理这件事上早就声名在外。他们最擅长的,就是 " 榨干 " 硬件的每一分性能,让 GPU、CPU、内存协同作战,在推理上做到了极致的低成本和高性能,让许多跑不起昂贵 GPU 的团队也能用上大模型。

如今,趋境科技将这一优势延伸至微调领域,并与社区人气极高的 LLaMA-Factory 框架无缝集成,无疑是一次强强联合。

从推理到微调这一路径的发展,非常明显的就是剑指加速 AI 大模型落地,而且是更好更便宜的那种。

这对于资源有限的学术界、渴望快速迭代的创业公司,乃至充满热情的个人开发者来说,无异于一场及时雨。

而且此举还意味着,创新的边界被再次拓宽。你可以不再受限于模型的大小,而是专注于你的创意和数据——无论是打造一个独一无二的虚拟角色,还是构建一个解决特定行业痛点的专业模型。

最后,我们找到了微调的详细技术文档和用户操作指南,如果你手上现在就有几块消费级显卡,不妨可以尝试一下这个性价比极高的微调大法哦 ~

KTransformers 项目地址:

https://github.com/kvcache-ai/ktransformers

LLaMA-Factory 项目地址:

https://github.com/hiyouga/LLaMA-Factory

一键三连「点赞」「转发」「小心心」

欢迎在评论区留下你的想法!

点亮星标

科技前沿进展每日见

相关标签