StableLM

Stability AI 开源语言模型系列,提供高效预训练与微调基座

访问官网 github.com
StableLM 官网页面截图
github.com 官网页面截图,实际界面以官网为准。

StableLM是什么

StableLM 是 Stability AI 开发的一系列开源语言模型,旨在为研究者和开发者提供高性能、可复现的自然语言处理基座。该仓库持续更新最新的模型检查点,涵盖从轻量级到中等规模参数的多种架构,支持社区在透明许可下进行创新与应用。

系列模型包括采用多轮次训练策略的 StableLM-3B-4E1T,以及经过指令微调的 StableLM-Tuned-Alpha 版本。这些模型基于 LLaMA 类似架构改进而来,使用 Rotary Position Embeddings 和 LayerNorm 等技术优化推理效率与稳定性,并在 Hugging Face 等平台提供交互式演示。

所有基础模型均在严格筛选的开源数据集上进行训练,如 Falcon RefinedWeb、RedPajama-Data 和 The Pile(剔除特定子集),确保数据质量与多样性。StableLM 致力于推动小型高效模型的研究,探索在有限计算资源下实现最优性能的路径。

主要功能

  • 高效参数设计StableLM-3B-4E1T 仅含约 28 亿参数,却通过在 1 万亿 token 上进行 4 个轮次的训练,实现了与更大模型相媲美的损失值表现。
  • 改进的 Transformer采用类似 LLaMA 的解码器-only 结构,但引入了 Rotary Position Embeddings 以提升吞吐量,并使用带偏置项的 LayerNorm 替代 RMSNorm。
  • 高质量训练数据训练数据混合了 Falcon RefinedWeb、RedPajama-Data 和 The Pile 等开源大型数据集,并剔除了 Books3 等特定子集以保证合规性与质量。
  • 开源许可友好基础模型如 StableLM-3B-4E1T 和 StableLM-Alpha 系列通常在 CC BY-SA-4.0 等开放许可下发布,便于学术研究与商业应用探索。
  • 零样本评估基准提供在 ARC、BoolQ、HellaSwag 等多个标准基准上的零样本评估结果,方便用户横向对比模型在常识推理、问答等任务上的能力。
  • 持续迭代更新仓库持续发布新版本的检查点,包括补丁修复后的 Alpha v2 模型以及针对特定任务优化的 RLHF 微调版本如 StableVicuna。

如何使用

  1. 访问 GitHub 官方仓库页面,浏览 README 以了解当前可用的模型列表及其特性。
  2. 根据需求选择合适的模型版本,例如需要轻量级部署可选择 StableLM-3B-4E1T。
  3. 查看提供的 YAML 配置文件和技术报告,获取完整的超参数设置和训练细节。
  4. 通过 Hugging Face Spaces 在线体验聊天模型,或下载权重在本地环境中进行推理测试。
  5. 如需针对特定下游任务优化,建议使用基础模型进行微调,并参考官方提供的评估脚本。

应用场景

  • 学术研究实验利用开源权重和详细的技术报告,研究多轮次训练对语言模型性能的影响及数据效率。
  • 轻量级应用部署在资源受限的边缘设备或服务器上部署 3B 参数模型,实现低延迟的文本生成与对话功能。
  • 自定义任务微调基于高质量的基础预训练模型,针对垂直领域的客服、写作或代码辅助任务进行指令微调。

价格 / 免费额度

价格与免费额度以官网为准。

适合人群

  • AI 研究人员
  • 开源社区开发者
  • 边缘计算工程师
  • 自然语言处理爱好者

常见问题

StableLM-3B-4E1T 中的 4E1T 是什么意思?

它表示模型在 1 万亿(1 Trillion)token 的数据集上进行了 4 个轮次(4 Epochs)的训练,旨在研究重复数据对模型性能的影响。

我可以将 StableLM 用于商业项目吗?

这取决于具体模型的许可证。例如,部分基础模型使用 CC BY-SA-4.0,而某些微调版本可能使用不同的许可,请务必检查官方仓库中每个模型的具体许可证说明。

StableLM 与 LLaMA 有什么关系?

StableLM 的架构类似于 LLaMA,但在位置编码和归一化层等方面进行了修改。此外,部分早期版本如 StableVicuna 是基于 LLaMA 的衍生模型进行的进一步微调。

标签

  • 开源大模型
  • 自然语言处理

返回模型与资源