3B模型新SOTA！开源AI让日常调用不同大模型更简单
NEXA AI 投稿
量子位 | 公众号 QbitAI
大模型，大，能力强，好用！
但单一大模型在算力、数据和能耗方面面临巨大的限制，且消耗大量资源。
而且目前最强大的模型大多为闭源，对AI开发的速度、安全性和公平性有所限制。
AI大模型的未来发展趋势，需要怎么在单一大模型和多个专门化小模型之间做平衡和选择？
针对如此现状，两位斯坦福校友创办的NEXA AI，提出了一种新的方法：
采用functional token整合了多个开源模型，每个模型都针对特定任务进行了优化。
他们开发了一个名叫Octopus v4的模型，利用functional token智能地将用户查询引导至最合适的垂直模型，并重新格式化查询以实现最佳性能。
介绍一下，Octopus v4是前代系列模型的演化，擅长选择和参数理解与重组。
此外，团队还探索了使用图作为一种多功能数据结构，有效地协调多个开源模型，利用Octopus模型和functional token的能力。
通过激活约100亿参数的模型，Octopus v4在同级别模型中实现了74.8的SOTA MMLU分数。
Octopus系列模型
这里要重点介绍一下Octopus-V4-3B。
它拥有30亿参数，开源，是Nexa AI设想中的语言模型图的主节点。
该模型专为MMLU基准测试话题定制，能够高效地将用户查询转换成专业模型可以有效处理的格式。
它擅长将这些查询正确引导至相应的专业模型，确保精确且有效的查询处理。
Octopus-V4-3B具备以下特点：
紧凑尺寸：Octopus-V4-3B体积紧凑，使其能在智能设备上高效、迅速地运行。
准确性：利用functional token设计准确地将用户查询映射到专业模型，提高了其精度。
查询重格式化：帮助将自然人类语言转换为更专业的格式，改善了查询描述，从而获得更准确的响应。
Nexa AI把语言模型作为图中的节点整合，并提供了针对实际应用定制的系统架构。
此外，讨论了使用合成数据集对Octopus模型进行训练的策略，强调了这种语言模型图在生产环境中的系统设计。
从Octopus v2提取的用于分类的语言模型
研究人员在Octopus v2论文中介绍了一种名为functional token的分类方法。
Octopus v2模型有效地处理了这一任务：
图中的语言模型作为节点
考虑一个定义为：G=(N，E)。
其中N代表图中的各种节点，E代表连接这些节点的边。
节点分为两种类型：
一，主节点Nm，它们通过将查询定向到合适的工作节点Nω并传递执行任务所需的信息来协调查询。
二，工作节点，接收来自主节点的信息并执行所需的任务，使用Octopus模型来促进进一步的协调。
节点信息传输过程如下图所示。
为了处理用户查询q并生成响应y，研究人员将概率建模为：
对于只涉及一个工作节点的单步任务，该过程可以定义为：
这里，P(Nω,ph|q;Nm)使用Octopus v2模型为𝑁m选择最佳的相邻工作节点并重新格式化查询为𝑞ℎ，这是重构的查询。
概率P(y|qh;Nω)由位于工作节点的语言模型计算。
对于多步任务，通常在多代理工作流中，该过程涉及多个节点之间的几次顺序交互，如下：
使用图进行多步骤操作的任务规划
多步骤任务规划中，所有功能列在上下文中提交给语言模型，生成基于用户查询的计划。
传统方法在处理长功能描述时有局限性，尤其是参数少于10B的模型。
基于图的方法确保只考虑与特定节点相关的邻近节点，显著减少了选择的复杂性。
语言模型图的系统设计
以下详细说明了复杂图中每个节点代表一个语言模型的系统架构，利用多个Octopus模型进行协调。
在准备生产部署时，整合一个负载均衡器以有效管理系统需求至关重要。
然后，研究团队将系统划分为几个可管理的组件，强调核心方法：
首先是工作节点部署。
每个工作节点Nω对应一个单独的语言模型。
团队建议为这些节点采用无服务器架构，特别推荐使用Kubernetes进行基于内存使用和工作负载的强大自动缩放。
其次是主节点部署。
主节点应使用不超过10B参数的基础模型（实验中使用3B模型），以便在边缘设备上部署。
每个工作节点与一个Octopus模型接口，以增强协调。
如Octopus v2所示，可以集成一个紧凑的Lora模型以扩展functional token的能力。
建议使用单一基础模型并补充多个Loras，每个工作节点一个。
推荐使用开源的LoraX库来管理这种配置的推理操作。
再者是通讯。
工作节点和主节点分布在不同设备上，不限于单一设备。
因此，互联网连接对于节点之间的数据传输至关重要。
虽然主节点可能位于智能设备上，工作节点则托管在云端或其他设备上，结果返回智能设备。
为了支持数据缓存需求，包括聊天历史存储，推荐使用Redis，一个高性能的内存数据库，促进分布式缓存。
实验
研究人员详细介绍了框架进行的实验，通过多节点协作提高语言模型性能。
采用了17种不同的模型进行MMLU任务，Octopus v4模型将用户查询定向到相关的专业模型并适当重格式化。
MMLU包含57个独特的任务，分为17个整合组。
专业模型根据基准得分、人气和用户推荐从Hugging Face精选。
并非所有任务都有专门模型，例如人文学科和社会科学目前无专门模型，但Llama3模型通过系统提示调整模拟专业能力。
未来工作与讨论
当前，NEXA AI的GitHub 项目专注于开发语言模型的图形框架，目前处于起始阶段。
团队计划通过整合多种垂直特定模型并加入Octopus v4模型来增强这一框架，以多代理能力为特征。
未来版本将在此存储库中展示更强大的图形表示。
GitHub Repo将由Nexa AI维护，团队今后旨在为多种垂直领域开发紧凑、专门化的模型。
与更大模型的缩放法则、数据集相比，NEXA AI的框架无限制，并且可以创建一个大型图。
此外，团队正在开发多模态模型Octopus 3.5，能够处理视觉、音频和视频数据；完成开发后，将被纳入图形框架。
论文链接：https://arxiv.org/pdf/2404.19296
GitHub： https://github.com/NexaAI/octopus-v4
试用Octopus v4：https://huggingface.co/NexaAIDev/Octopus-v4
投稿请发邮件到：
标题注明【投稿】，告诉我们：
你是谁，从哪来，投稿内容
附上论文/项目主页链接，以及联系方式哦
我们会（尽量）及时回复你
点这里👇关注我，记得标星哦～
微信扫码关注该文公众号作者