Parason: Revealing Subtask and Trial Parallelism in LLM Reasoning
本文介绍了 Parason,这是一个通过结构化语法和专门的训练算法来识别并利用子任务并行性和尝试并行性,从而显著加速大语言模型推理的框架,在保持准确性的同时,在复杂的数学基准测试上实现了约 1.7 倍的加速。
原始论文采用 CC BY 4.0 许可(http://creativecommons.org/licenses/by/4.0/)。 这是对下方论文的AI生成解释。它不是由作者撰写或认可的。如需技术准确性,请参阅原始论文。 阅读完整免责声明
大语言模型在解决复杂问题方面已变得异常出色,从破解困难的数学谜题到编写复杂的代码。这种能力通常源于一个简单的策略:让模型思考更长的时间。通过在给出答案之前生成一段长长的、循序渐进的思维链,这些系统可以处理棘难的逻辑并避免简单的错误。然而,这种方法存在一个显著的物理限制。这些模型标准的工作方式就像一个人按顺序逐字阅读书籍,严格遵循先后顺序。每一个想法都必须等待前一个想法完成后才能开始。当问题极其困难时,思维链可能会延伸至数十万字,这意味着用户可能需要等待几分钟、几小时甚至几天才能得到一个答案。这种延迟使得该技术无法应用于像编程助手或交互式导师这样对速度要求极高的实时工具。
研究人员长期以来一直怀疑,答案在于并行思考。与其采用一条长长的单线思维,不如想象一支专家团队同时在问题的不同部分开展工作。一些现有的系统尝试通过将一个大任务分解为若干个较小的、独立的碎片,分别解决每个碎片,然后组合结果来实现这一点。虽然这很有帮助,但这种方法忽略了智能推理真正运作的核心部分。在面对真正困难的问题时,一个聪明的思考者不仅仅是分解问题,他们还会尝试不同的想法,观察哪些失败了,并保留那些奏效的想法。他们同时探索多条路径,舍弃死胡同,并沿着有希望的方向前进。本文介绍了一个名为 Parason 的新系统,它教会语言模型正是这样做,将缓慢的、单列排队的思考过程转变为快速的多车道高速公路。
研究人员首先分析了来自先进模型的数千条推理轨迹,以观察它们究竟是如何思考的。他们发现,思考过程不仅仅是单一的步骤线条,而是两种截然不同的活动类型的混合。第一种类型类似于建筑施工队:一个团队负责打地基,另一个负责搭建框架,第三个负责安装屋顶。所有这些任务都是必要的,并且可以同时进行。研究人员称之为“子任务并行”(subtask parallelism)。第二种类型则更像是同时测试多种理论的侦探:侦探尝试一个嫌疑人,发现并不符合,于是立即尝试另一个,而同时一名同事在检查第三种可能性。在这些路径中,只需要其中一条正确即可破案,但侦探必须探索所有路径才能找到正确的那一个。研究人员称之为“试验并行”(trial parallelism)。
他们的发现令人惊讶。在最困难的问题中,“侦探”式的思考方式——即尝试许多不同的想法并丢弃失败案例——占据了大部分的工作量。事实上,对于一些最先进的模型,这种基于试验的探索占据了超过三分之二的推理步骤。以往只专注于将任务分解为较小碎片的系统,本质上忽略了这些模型解决难题最常见的方式。它们是在为“建筑施工队”进行优化,而真正的核心工作却是由“侦探”完成的。
为了解决这个问题,该团队构建了一个框架,教导模型识别何时应该拆分任务,以及何时应该同时尝试多个想法。他们创建了一套特殊的规则,类似于一种结构化语言,允许模型说:“这里有三种解决此问题的方法,我将同时尝试它们。”随后,系统并行运行这些尝试,并收集结果。如果某次尝试失败,模型只需继续执行下一个,而无需等待失败的尝试完成后才开始下一个。这与旧的方法相比是一个重大的转变,在旧方法中,模型必须完成一个想法,意识到错误后,再一个接一个地开始下一个。
训练过程使用了一种奖励机制,鼓励模型既要正确,又要快速。模型不仅因得出正确答案而受到表扬,还因其组织思考的方式能够让计算机更快地完成工作而受到表扬。通过同时运行这些并行分支,该系统有效地缩短了得出结论所需的时间。在针对困难数学竞赛的测试中,这种新系统与标准的顺序处理方法相比,实现了约 1.7 倍的平均加速,同时保持了同样高水平的准确度。这意味着一个原本可能需要十分钟解决的问题,现在大约只需六分钟,且不会牺牲解题质量。
研究人员还观察到,问题越难,这种并行方法带来的帮助就越大。在最困难的问题上,节省的工作量是最多的。这是因为难题自然会产生更多的死胡同,并需要更多对不同可能性的探索。通过允许模型并行探索这些可能性,而不是一个接一个地进行,该系统将曾经漫长、缓慢的等待变成了一个高效的过程。这项研究表明,人工智能推理的未来不仅在于思考得更久,还在于同时向多个方向思考,模仿人类专家应对世界上最复杂挑战的方式。
您所在领域的论文太多了?
获取与您研究关键词匹配的最新论文每日摘要——附技术摘要,使用您的语言。