Suzu-2

输入法候选排序小模型。给定上文和同一拼音下的候选词，对候选打分；用于 Rime 等输入法的候选重排。

• 架构：gpt2，12 层、512 维、8 头，字符级词表 6000 字，约 40.9M 参数；Q8_0 量化文件约 41.9 MiB。
• 训练：以 Qwen3.5 为教师蒸馏，语言模型损失加候选排序的 KL；语料为维基百科中文（见 NOTICE.md）。
• 文件：suzu-2.q8_0.gguf、suzu-2.q8_0.gguf.vocab.txt（一行一个字符，行号即编号：0 填充、1 未知、2 句首）。

输入与输出

输入为句首符、上文最后至多 30 个字、候选的字；输出为各字对数概率之和，越大越优。需推理端自行做字符到编号的映射，不使用 llama.cpp 的分词器。

评测

评测集 16000 题（维基百科中文，易例与难例各半，按真实比例加权），首选准确率：混合 92.8%（Qwen3.5-0.8B 为 92.7%，Qwen3.5-9B 为 95.1%），难例（纯模型）72.4%（0.8B 为 70.7%）。桌面 CPU 2 线程约 21 ms，Zenfone 10 上中位 20.1 ms，常驻内存约 115 MB。

适用范围与局限

• 只覆盖简体中文，语料偏书面语，未在日常聊天文本上评测。
• 混合先验权重在评测集上选取，数字略乐观。
• 仅用于候选重排，不做文本生成。

许可与署名

权重按 CC BY-SA 4.0 许可。允许使用、修改和商用，但须署名并注明改动。基于它的衍生品（如微调出的新模型）须按相同条款发布。

训练数据来自维基百科中文（CC BY-SA 3.0 与 GFDL），教师为 Qwen3.5（Apache-2.0）。使用时须保留署名，详见 NOTICE.md。

复现

训练与评测代码：Suzu；输入法集成（推理核心与三端接入）：suzu-ime；词库与配置：rime-ice-Custom。