English(EN) Step-TP: A Grounded, Step-Level Dataset with Chain-of-Thought Reasoning for LLM-Guided Tensor Program Optimization

新数据集通过逐级推理增强LLM张量程序优化能力

作者 PulseAugur 编辑部 · [2 个来源] · 2026-05-25 15:29

研究人员推出Step-TP，一个旨在提升大型语言模型（LLMs）优化张量程序能力的新数据集。现有方法通常缺乏逐级监督和可解释性，阻碍了LLMs在复杂优化任务上的表现。Step-TP提供了原子级的、逐级监督以及结构化的链式思考推理，使LLMs能够通过理解中间程序状态来做出更可靠的单步决策。 AI

影响该数据集旨在提升LLMs在特定程序优化领域的性能，可能带来更高效的AI模型编译和执行。

排序理由该集群包含一篇详细介绍用于AI研究的新数据集的研究论文。

AI 生成摘要 · Google Gemini · 来自 2 个来源。我们如何撰写摘要 →

报道来源 [2]

arXiv cs.AI TIER_1 English(EN) · Mengfan Liu, Da Zheng, Junwei Su, Chuan Wu · 2026-05-26 04:00

Step-TP：一个基于链式思考推理的、面向LLM指导的张量程序优化的基础性、步进式数据集

arXiv:2605.25954v1 Announce Type: cross Abstract: Despite the strong reasoning capabilities of large language models (LLMs), optimizing the execution efficiency of tensor programs remains challenging due to the need for precise, composable transformation decisions. Recent LLM-gui…
arXiv cs.AI TIER_1 English(EN) · Chuan Wu · 2026-05-25 15:29

Step-TP：一个基于LLM引导的张量程序优化的、具有思维链推理的、分步式数据集

Despite the strong reasoning capabilities of large language models (LLMs), optimizing the execution efficiency of tensor programs remains challenging due to the need for precise, composable transformation decisions. Recent LLM-guided approaches frame tensor program optimization a…