☰
从数据到权重:Qwen3.6-35B-A3B-AntiLoop-NVFP4 的 AntiLoop SFT 训练 + KL 正则化全流程揭秘
2026/10/21 10:47:27 网站建设 项目流程

【免费下载链接】Qwen3.6-35B-A3B-AntiLoop-NVFP4

项目地址:https://ai.gitcode.com/hf_mirrors/mlx-community/Qwen3.6-35B-A3B-AntiLoop-NVFP4
点击查看免费下载

Qwen3.6-35B-A3B-AntiLoop-NVFP4 是一个专为修复大模型"无限循环"问题而生的文本生成模型。它基于 Qwen3.6-35B-A3B,通过 178 条精心构造的训练数据完成 AntiLoop SFT 训练,并用 KL 正则化防止"矫枉过正",最终以 NVFP4 混合精度量化交付。本文将从训练数据、SFT + KL 正则化流程、能力保持评测,一路讲到混合精度权重的无损 MLX 转换,带你完整看懂这条"从数据到权重"的链路。

一、AntiLoop 解决什么问题:大模型"死循环"

用过大模型思考模式的朋友可能都见过这种场景:模型在推理阶段开始反复回忆、反复重新计数、反复自我验证,明明已经接近答案,却卡在无效循环里出不来,直到把 token 额度耗尽。

AntiLoop 正是针对这类病理性自我验证与枚举循环(pathological self-verification and enumeration loops)做的"窄幅微调"(narrow fine-tune)——只修行为,不动其他能力。

为了量化这种"循环",作者专门构建了一个名为LoopHard的评测集:285 个专门诱发无效循环的枚举型提示词,由 GLM-5.2 按"收敛感知"标准判卷——能得出系统结论的验证不算循环,发现自己打转并能自行退出算"恢复"。核心指标是judged loops(被判为循环的样本数)。

四组对照结果非常直观:

模型 / 设置被判循环数循环率
NVIDIA NVFP4 原始版72 / 28525.26%
AntiLoop NVFP410 / 2853.51%
NVIDIA NVFP4 +presence_penalty=1.530 / 28510.53%
AntiLoop NVFP4 +presence_penalty=1.51 / 2850.35%

也就是说,仅靠 AntiLoop 微调,循环率就从 25.26% 降到 3.51%;再叠加采样参数presence_penalty=1.5,进一步压到 0.35%。配对比较的双侧 McNemar 检验p = 2.98e-8,效果高度显著。

二、训练数据:178 条"带掩码"的监督目标 🎯

很多人会问:修复循环行为,需要多少数据?答案是——最终一轮只用了 178 条。

这份数据(Qwen3.6-35B-A3B-AntiLoop-SFT数据集)的设计有两个关键细节:

  • loss_start_char边界标记:每条样本中,病态循环的前缀被刻意保留为条件上下文(conditioning context),而不是监督目标。模型学习的是"看到这种循环前缀后如何优雅退出",而不是"从头学会一段正常回答"。
  • KL 锚点单独存放:用于 KL 正则化的锚点数据被单独生成,不包含在这 178 条 SFT 数据里,两者各司其职。

这种"掩码监督"让数据效率极高:178 条样本足以教会模型在正确的时刻刹车。

三、训练流程:SFT + KL 正则化,为什么两者缺一不可

AntiLoop 适配器的训练采用了标准监督微调(SFT)流程,但额外加了一味"药":

用基础模型在日常生活类提示词上单独生成的、不循环的锚点样本,计算 KL 散度损失来约束训练。

用一句话概括这套流程:

  1. SFT 项:让模型学会从循环前缀中恢复、正常完成回答;
  2. KL 正则项:拉住模型在普通任务上的分布,别让它为了"不循环"而改变日常说话方式。

这正是后文能力保持评测能通过的原因——KL 正则化充当了"能力护栏"。另外,推理时不需要任何 PEFT 适配器,LoRA 更新已经合入权重,拿来即用;模型也完整保留了语言架构、分词器、聊天模板和 262,144 token 的原生上下文配置。

四、修了循环,能力还在吗?📊

这是所有行为微调最让人担心的问题。作者在匹配运行时 LoRA 配置下,对比了第二轮 AntiLoop 适配器和 FP8 参考模型:

基准测试FP8 参考AntiLoop FP8
GPQA Diamond167 / 198(84.34%)166 / 198(83.84%)
GSM8K1273 / 1319(96.51%)1270 / 1319(96.29%)

两项差异在统计上均不显著——换句话说,循环率下降近 86%,而推理与数学能力几乎无损。

五、从数据到权重:NVFP4 混合精度 + 无损 MLX 转换

训练出的行为最终要固化进权重。本模型是一个混合精度 NVIDIA ModelOpt 部署检查点,量化策略按模块分层(详见 config.json 中的mlx_hybrid_format字段):

  • 注意力密集层(如self_attn、linear_attn投影矩阵):FP8 E4M3 权重 + FP8 动态输入激活;
  • MoE 专家与共享专家、lm_head:NVFP4(E2M1 权重量化,group size 16,E4M3 块级 scale)。

模型共 40 层、每层 256 个专家,架构为Qwen3_5MoeForConditionalGeneration(线性注意力与全注意力 3:1 混合)。

在转到 MLX 侧时,convert_qwen36_modelopt_hybrid_to_mlx.py 做的是格式转换而非重量化:

  • FP8 字节直接打包进 MLXuint32张量,配合单位 E8M0 载波 scale 精确还原原始值;
  • NVFP4 的 E2M1 半字节与 E4M3 块 scale按位保留(bit-identical);
  • BF16/FP32 张量维持源精度。

mlx_conversion_manifest.json 中weight_bytes_requantized: false就是这份承诺的落笔:1,475 个张量、41 个语言分片,转换全程未引入任何额外量化误差(注意:激活仍用模型 dtype,不保证与 ModelOpt 推理逐位一致)。运行时模型文件为 modeling_mlx_qwen36_modelopt_hybrid.py,MLX-LM 通过本地model_file钩子直接加载,不改动已安装的包。

六、快速上手:MLX-LM 一键运行 ✅

想在自己的 Apple 设备上体验这个"不循环"的模型,只需要两步:

1. 安装经过验证的运行时版本

pip install -U "mlx-lm==0.31.3"

2. 直接生成

python -m mlx_lm generate \ --model mlx-community/Qwen3.6-35B-A3B-AntiLoop-NVFP4 \ --trust-remote-code \ --prompt "Solve: 27 * 43" \ --max-tokens 512

几个实用参数提示:

  • 打包默认采样参数为temperature=0.6、top_p=0.95、top_k=20(见 generation_config.json);
  • 聊天模板保留了enable_thinking开关(chat_template.jinja),LM Studio 等客户端可正常暴露思考开关;
  • 若追求评测同款"极致抗循环"效果,可在服务端追加presence_penalty=1.5;
  • 本仓库为纯文本版,有意省略了视觉塔(333 个 BF16 视觉张量)与 MTP 草稿权重,图片/视频输入不受支持。

七、局限性与注意事项

引用 README.md 中的官方提醒,使用前三思:

  • 这是窄幅行为微调,不是通用对齐或安全模型,不承诺输出安全、无偏见;
  • 循环率(LoopHard)是任务特定、基于评审模型的指标,不能外推为通用安全/事实性分数;
  • GPQA / GSM8K 的能力保持验证用的是运行时 LoRA + FP8 基座,并非这个 NVFP4 工件本身,且presence_penalty=1.5下的能力保持尚未评测;
  • 源检查点的固定 scale FP8 合并是对 BF16 LoRA 合并的近似,微小更新可能被舍入或截断。

总结:一张表看懂全流程

环节关键做法
训练数据178 条掩码监督目标,loss_start_char把循环前缀当上下文而非标签
训练目标SFT 恢复行为 + KL 散度约束日常分布
效果循环率 25.26% → 3.51%(叠加presence_penalty=1.5后 0.35%)
能力保持GPQA 83.84%、GSM8K 96.29%,与参考模型无显著差异
权重形态FP8 + NVFP4 混合精度,无损转入 MLX,1,475 张量 / 41 分片
使用门槛一条mlx_lm generate命令,无需额外适配器

AntiLoop 的思路给所有大模型实践者一个清晰启示:修复一个具体行为缺陷,未必需要海量数据和全参数重训——精准构造的掩码数据、一个恰到好处的 KL 护栏、再配上严格的对照评测,178 条样本就能把 25% 的死循环率压到 3.5%,而能力几乎分毫不动。

【免费下载链接】Qwen3.6-35B-A3B-AntiLoop-NVFP4

项目地址:https://ai.gitcode.com/hf_mirrors/mlx-community/Qwen3.6-35B-A3B-AntiLoop-NVFP4
点击查看免费下载
上一篇:GARbro社区贡献指南:如何参与项目开发与维护
下一篇:WhyNotWin11开源项目分析:LGPL-3.0许可证完全解读指南

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

需要专业的网站建设服务?

联系我们获取免费的网站建设咨询和方案报价,让我们帮助您实现业务目标

立即咨询