AX 爱鲜报

MiniMind 迷你大脑

从零训练 64M 参数超小语言模型的开源项目,2 小时、约 3 元成本即可在单张 3090 上完成全流程复现,覆盖预训练、SFT、LoRA、RLHF、RLAIF 与 Agentic RL。

爱鲜报评分

0.0

0 人评分

63k

星数

是

中文

Python

主语言

是

活跃

21

贡献者

人工分析

EDITORIAL

推荐理由

这是目前少见的把 LLM 全链路算法用 PyTorch 原生从 0 实现的教学级项目,不依赖 transformers/trl/peft 的高层封装,适合想真正看懂大模型内部机制的开发者。64M 的体量让个人显卡也能跑通预训练到强化学习全流程,配套中文文档与视频讲解,学习曲线友好。

适用场景

  • LLM 原理学习与源码级研究
  • 个人 GPU 上从零复现大模型训练
  • 高校教学与课程实验
  • LoRA/SFT/DPO/GRPO 等算法实践
  • 小模型私有数据微调与部署验证

优点

  • 全流程算法从 0 用 PyTorch 原生实现,无高层框架黑盒
  • 模型仅 64M,单卡 3090 约 2 小时即可完成 SFT 复现
  • 覆盖 Pretrain、SFT、LoRA、DPO、PPO/GRPO/CISPO、Tool Use、蒸馏等完整链路
  • 中文文档详尽,配套 B 站视频与在线体验 Demo
  • 兼容 transformers、llama.cpp、vllm、ollama 等主流生态

缺点

  • 模型规模极小,实际生成能力与商用大模型差距明显
  • 训练流程涉及强化学习与分布式,部分环节需一定深度学习基础
  • 版本迭代较快,旧模型与旧代码存在兼容性断裂风险
  • 完整复现仍需可用的 NVIDIA GPU 与 CUDA 环境
上手难度 进阶

MiniMind 迷你大脑是什么

MiniMind 是一个「从 0 开始训练大语言模型」的开源项目。它用约 3 元 GPU 租用成本、2 小时训练时间,就能在单张 NVIDIA 3090 上训练出参数量约 64M 的超小语言模型。主线最小版本体积约为 GPT-3 的 1/2700,目标是让普通个人显卡也能完整跑通大模型训练与复现。

项目不只是「能跑」,更强调「能懂」:所有核心算法均使用 PyTorch 原生实现,不依赖 transformers / trl / peft 等高层抽象接口,因此每一行训练代码都可以被阅读和修改。

核心特性

  • 完整训练链路:覆盖 Pretrain、SFT、LoRA、RLHF(DPO)、RLAIF(PPO / GRPO / CISPO)、Tool Use、Agentic RL、自适应思考与模型蒸馏。
  • 极简模型结构:提供 Dense 与 MoE 两种结构,主线对齐 Qwen3 / Qwen3-MoE 生态。
  • 自带分词器:Tokenizer 基于 BPE + ByteLevel 训练,支持 <tool_call>、<think> 等模板标记。
  • 全阶段开源数据:数据经过收集、蒸馏、清洗与去重,开箱即用。
  • 生态兼容:可对接 llama.cpp、vllm、ollama 等推理引擎,以及 Llama-Factory 等训练框架。
  • 服务与界面:提供兼容 OpenAI API 协议的极简服务端,以及基于 Streamlit 的聊天 WebUI。

已发布模型

模型 参数量 发布时间
minimind-3 64M 2026.04.01
minimind-3-moe 198M-A64M 2026.04.01
minimind2 104M 2025.04.26
minimind2-small 26M 2025.04.26
minimind-v1 108M 2024.09.01

训练流程概览

数据准备 → Tokenizer 训练 → Pretrain → SFT → LoRA 微调
        → RLHF(DPO) → RLAIF(PPO/GRPO/CISPO) → Tool Use / Agentic RL

适合谁

  • 想真正理解 LLM 内部机制、不愿停留在「调 API」层面的开发者;
  • 需要在小规模数据上验证训练算法的高校与科研人员;
  • 希望用个人显卡完整跑通一次大模型训练流程的爱好者。

注意事项

  • 项目迭代较快,跨版本模型与代码可能存在不兼容,升级前建议先看更新日志。
  • 64M 模型的能力有限,更适合作为学习与实验载体,而非生产级对话模型。
  • 完整复现需要可用的 NVIDIA GPU 与 CUDA 环境。

评论

登录后才可评论与评分

0 条评论

  • 暂无评论,来写第一条吧。