神经符号路由器让边缘设备推理更可靠
「研究提出神经符号路由器,将结构化推理查询分发给确定性符号求解器,仅把开放式应用题交给小型语言模型。系统用L*文法推断算法学习确定性有限自动机,在树莓派4B上达到100%路由准确率与98.3%整体准确率,响应最快仅需1毫秒。」
在边缘硬件上直接运行语言模型,意味着无需网络连接即可获得低延迟且保护隐私的推理能力。然而,能够塞进这类设备的小型模型,恰恰在计算机最擅长的那类任务上表现糟糕,例如算术、代数与形式逻辑问题。一篇新发表的论文认为,这种不可靠性在相当程度上是可以避免的。
论文指出,许多看似需要推理的查询,实际上是结构确定性的,可以用快速而精确的符号方法直接求解。让概率模型去近似这些任务,等于用准确率和能耗换取几乎没有的收益。基于这一判断,作者提出了一种神经符号路由器:它对每一条输入查询进行分类,然后分派给成本最低且能给出正确答案的求解器,把结构化任务交给确定性引擎,只把开放式文字应用题留给小型语言模型。
更值得注意的是路由逻辑的获得方式。研究团队没有手工编写路由规则,而是借助L*文法推断算法,学习出一个确定性有限自动机。在这一过程中,小型语言模型充当成员资格查询的判定器,而人工标注的数据则充当等价判定器,两者配合逐步逼近目标语言。这种把模型从执行者转变为规则发现辅助者的思路,使得路由行为可以被形式化描述与验证,而不是依赖难以维护的经验规则。
实验在一台树莓派4B上进行,硬件配置为8GB内存且没有GPU。评测使用了来自DeepMind数学数据集、GSM8K与RuleTaker的100条未参与训练的提示。结果显示,学习得到的路由策略实现了100%的路由准确率;在512个token的推理预算下,整体准确率达到98.3%,其中开放式文字应用题为93.3%。作为对比,最强的智能体基线Program-of-Thought只有72.0%,而在配备相同求解器的情况下,一个工具调用式智能体的准确率仅为58.7%。
效率方面的差距同样明显。由于格式化查询根本不会送进模型,路由器处理这类请求只需1至11毫秒。在30个token的配置下,它的运行速度比Program-of-Thought快8.8倍,能效高出2.8倍。对于电池容量与散热能力都受限的边缘设备而言,这种数量级的差距往往决定了一项功能能否真正落地。
这项工作带来的启示不止于性能数字。它提示研究者重新审视推理任务的构成:并非所有被冠以推理之名的请求都需要神经网络去近似。把确定性的部分交还给确定性工具,把模糊与开放的部分留给模型,是一种更符合成本收益原则的分工方式。同时,用文法推断从数据中自动获得路由规则,也避免了人工规则库随任务扩张而迅速膨胀的问题,让系统在多类任务并存的场景中仍可保持可控的行为边界。
当然,这种架构的有效性依赖于能否把查询稳定地转化为符号求解器可接受的格式,以及在面对自动机之外的新任务分布时能否保持稳健。论文在特定数据集上给出了令人信服的结果,但更大规模、更多样化的真实场景仍有待检验。尽管如此,神经符号路由在资源受限设备上展示了一条兼顾准确率、延迟与能耗的现实路径,也为小型语言模型如何与经典符号方法协同提供了可复用的范式。
来源:Heooo AI工具导航