英伟达发布全新小型模型 Nemotron-Nano-9B-V2:免费商用且性能卓越
小型模型正在掀起一场风暴,而英伟达也不甘落后。在麻省理工学院和谷歌相继推出能在智能手表和智能手机上运行的小型 AI 模型后,英伟达发布了其最新的小型语言模型(SLM)——Nemotron-Nano-9B-V2。该模型在多个基准测试中表现出色,并在特定测试中达到了同类产品的最高水平。
专为效率和推理而生
Nemotron-Nano-9B-V2的参数量为90亿,虽然比一些数百万参数的微型模型要大,但它比之前的120亿参数版本显著减小,并专门针对单个英伟达 A10GPU 进行了优化。英伟达 AI 模型后训练总监 Oleksii Kuchiaev 解释说,这种调整是为了适配 A10这款热门的部署 GPU。此外,Nemotron-Nano-9B-V2是一款混合模型,能处理更大的批次,速度比同等规模的 Transformer 模型快6倍。
该模型支持多达九种语言,包括中、英、德、法、日、韩等,并擅长处理指令跟踪和代码生成任务。其预训练数据集和模型本身都已在 Hugging Face 和英伟达的模型目录中提供。
融合 Transformer 与 Mamba 架构
Nemotron-Nano-9B-V2基于 Nemotron-H 系列,该系列融合了 Mamba 和 Transformer 架构。传统的 Transformer 模型虽然强大,但在处理长序列时会消耗大量内存和计算资源。而 Mamba 架构则引入了选择性状态空间模型(SSM),能够以线性复杂度处理长信息序列,从而在内存和计算开销上更具优势。Nemotron-H 系列通过用线性状态空间层替换大部分注意力层,在长上下文处理上实现了2-3倍的吞吐量提升,同时保持了高精度。
独特的推理控制功能
这款模型的一大创新是其内置的“推理”功能,允许用户在模型输出最终答案前进行自我检查。用户可以通过简单的控制符(如 /think
或 /no_think
)来开启或关闭此功能。模型还支持运行时“思考预算”管理,开发者可以限制用于内部推理的令牌数量,从而在准确性和延迟之间取得平衡。这对于客户支持或自主代理等对响应速度有要求的应用场景尤为关键。
严格的开放许可,瞄准企业应用
英伟达根据其开放模型许可协议发布了 Nemotron-Nano-9B-V2,该协议对企业友好且高度宽松。英伟达明确表示,企业可以自由地将该模型用于商业用途,并且无需为使用该模型支付费用或版税。
尽管如此,协议仍有一些核心要求,例如用户必须遵守内置的安全机制、在重新分发模型时进行归属标注,并遵守相关法律法规。英伟达表示,该协议旨在确保负责任和合乎道德的使用,而不是通过限制商业规模来盈利。这使得 Nemotron-Nano-9B-V2成为了那些希望在降低成本和延迟的同时,保持高精度的企业开发者的理想选择。
空空如也!