【免费下载链接】Hy-Memory-30B-A3B
Hy-Memory-30B-A3B 是腾讯混元推出的紧凑型长期记忆写入模型(约 3B 激活参数),用于在 Mem0、MemOS、Hy-Memory 等外部记忆框架中决定"记什么、怎么记、如何更新"。本文带你完整拆解它的 128 专家 MoE 架构、3B 激活参数设计与 256K 长上下文的实现细节,并附上跨框架评测表现与快速上手方法。
上图展示了 Hy-Memory-30B-A3B 在四个记忆框架(harness)上的宏平均得分,其中 EverMemOS 为训练中未见的留出框架,模型平均分 55.49,整体领先同级开源与闭源模型。
一、它是什么:专攻"记忆写入"的长期记忆模型
大多数大模型负责"回答问题",而 Hy-Memory-30B-A3B 专门负责记忆系统中的写路径(write path):
- 判断哪些对话信息值得保留(用户事实、偏好、代码上下文等)
- 决定如何精准表达这些记忆
- 当新证据出现时,决定如何更新已有记忆库
它的定位是"可插拔的记忆写入器"——可以直接替换 Mem0、MemOS、Hy-Memory(覆盖 Chat、Working、Coding 三类场景)中的通用 LLM,甚至能零样本迁移到训练中完全没见过的 EverMemOS 框架上。
二、架构拆解:128 专家 MoE 与 3B 激活参数
所有架构细节都在 config.json 中,关键配置如下:
| 配置项 | 数值 | 含义 |
|---|---|---|
num_experts | 128 | 每个 MoE 层共 128 个专家 |
num_experts_per_tok | 8 | 每个 token 只激活 8 个专家 |
moe_intermediate_size | 1536 | 单个专家中间维度 |
num_hidden_layers | 48 | 总层数 |
first_k_dense_replace | 1 | 第 1 层为稠密层,其余 47 层为 MoE |
hidden_size | 2048 | 隐藏层维度 |
num_attention_heads/num_key_value_heads | 32 / 4 | GQA 分组查询注意力 |
max_position_embeddings | 262144 | 支持 256K 上下文 |
vocab_size | 120832 | 词表大小 |
几个值得注意的设计点:
1. "总参 30B,激活约 3B"的 MoE 结构。全模型共 300 多亿参数(51 个权重分片),但每个 token 只路由到 8 个专家,实际计算量仅对应约 3B 激活参数——推理成本接近小型模型,能力接近大模型。
2. 专家路由细节。配置显示moe_router_use_sigmoid与moe_router_enable_expert_bias均为true,即采用带专家偏置的 sigmoid 路由(类似 DeepSeek-V3 的无辅助损失设计),并配有num_shared_experts: 1共享专家兜底,保证每个 token 都有基础容量。
3. 注意力效率。32 个查询头共享 4 个 KV 头(GQA),配合qk_norm: true的 QK 归一化,为长上下文稳定性服务。
权重文件组织可参考 model.safetensors.index.json,51 个model-000XX-of-00051.safetensors分片按层与专家切分,总大小约 60GB(bf16 精度)。
三、256K 长上下文如何实现
config.json中max_position_embeddings: 262144,即256K token上下文;tokenizer_config.json 中model_max_length同为 262144,两者严格对齐。
对记忆写入模型而言,长上下文意味着:单次可处理超长对话历史、大型代码仓库上下文,从而一次性抽取完整记忆,而不必依赖粗暴的截断。位置编码采用 RoPE(rope_theta: 11158840.0,见 config.json),配合 GQA 与 QK 归一化共同支撑超长序列推理。
四、两阶段训练:OPD 蒸馏 + 记忆强化学习
模型在与部署完全相同的框架环境中分两阶段训练(详见 README.md 的 "Two-Stage Training" 章节):
- Memory On-Policy Distillation(OPD,记忆在线策略蒸馏):在"学生模型自己写出的状态链"上提供密集的 teacher 监督信号,教会模型基本的记忆写入格式与规范。
- Attributed Memory Reinforcement Learning(带归因的记忆强化学习):以最终记忆库效果为导向做奖励归因,按 chunk 粒度分配信用,并把得分最高的记忆库滚动传递到下一个训练窗口。
这种"先蒸馏、后强化"的方式,让模型学会的不是泛泛的摘要,而是"写出的记忆能在真实框架中跑分更高"。
五、跨框架实测表现
评测固定 harness 版本、提示词、检索配置与答题模型,只更换记忆写入器,在 LoCoMo、LongMemEval、PersonaMem-v2、Travel SPS、Shopping PS、Code Memory Bench 六项指标上打分:
| 记忆写入器 | Mem0 | Hy-Memory | MemOS | EverMemOS(留出) | 平均 |
|---|---|---|---|---|---|
| Hy-Memory-30B-A3B | 54.02 | 60.55 | 50.51 | 56.89 | 55.49 |
| Claude Opus 5 | 53.43 | 60.53 | 50.50 | 54.04 | 54.62 |
| DeepSeek V4.1 Flash | 49.73 | 60.20 | 49.17 | 55.47 | 53.64 |
| Qwen3.6-35B-A3B | 47.51 | 55.87 | 47.36 | 50.75 | 50.37 |
| Qwen3-30B-A3B | 39.81 | 45.67 | 45.44 | 46.57 | 44.37 |
亮点:作为约 3B 激活的轻量模型,它在四个框架上全部拿到第一,尤其在未参与训练的 EverMemOS 上以 56.89 大幅领先,验证了跨框架泛化能力。完整六指标明细见 README.md 的 "Detailed Benchmark Results"。
六、快速上手:vLLM 部署 + OpenAI 兼容 API
模型支持 vLLM(已验证 v0.31.0)部署,通过--served-model-name hy-memory启动后,即可用 OpenAI 兼容接口调用。最小示例:
from openai import OpenAI client = OpenAI(base_url="http://127.0.0.1:8000/v1", api_key="EMPTY") response = client.chat.completions.create( model="hy-memory", messages=[ {"role": "system", "content": "Extract explicit user facts and preferences... " 'Return only a JSON object with a "memories" list of strings.'}, {"role": "user", "content": "My name is Alex. I live in Seattle and work as a software engineer."}, ], temperature=0.0, max_tokens=512, extra_body={"chat_template_kwargs": {"reasoning_effort": "no_think"}}, )提示:通过reasoning_effort: "no_think"关闭思考模式,可获得更低的记忆写入延迟;对话模板定义见 chat_template.jinja。
七、文件结构速览
| 文件/目录 | 说明 |
|---|---|
| config.json | 模型架构配置(MoE、256K 上下文等全部参数) |
| model.safetensors.index.json | 51 个权重分片的索引映射 |
| tokenizer.json / tokenizer_config.json | 分词器与词表配置(词表 120832) |
| chat_template.jinja | 对话渲染模板 |
| README.md | 模型卡:训练流程、评测、Quickstart |
| LICENSE | Apache-2.0 开源协议 |
| figures/fig3_cross_harness_colm.png | 跨框架评测得分对比图 |
八、小结
Hy-Memory-30B-A3B 证明了"记忆写入"可以是一个独立的模型任务:128 选 8 的 MoE 架构让 30B 总参数只需约 3B 激活成本,256K 长上下文保证超长对话不截断,OPD + 记忆强化学习两阶段训练则直接优化了真实框架中的记忆效果。如果你正在搭建长期记忆应用(Chat、Agent、Coding 场景),它几乎可以"即插即用"地替换掉你的记忆写入器。
【免费下载链接】Hy-Memory-30B-A3B
相关推荐
小白也能看懂的Qwen3-30B-A3B:128个专家、激活仅3B参数的高效MoE架构全解析
小白也能看懂的Qwen3 30B A3B:128个专家、激活仅3B参数的高效MoE架构全解析 Qwen3 30B A3B 是一款总参数 300 多亿、每个 to
30B参数仅激活3B的秘密:InternVL3.5-30B-A3B MoE专家路由机制源码级解析
30B参数仅激活3B的秘密:InternVL3.5 30B A3B MoE专家路由机制源码级解析 InternVL3.5 30B A3B Pretrained
一张显卡就能跑:30B 总参只激活 3B,Hy-Memory-30B-A3B 本地部署上手
一张显卡就能跑:30B 总参只激活 3B,Hy Memory 30B A3B 本地部署上手 在 Agent 应用越来越依赖"长期记忆"的当下,一个尴尬的矛盾始终
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考