将聚类视为推理:链路思考图学习的 $k$-均值解释
人工智能
2026-05-26 v1 计算与语言
网络与互联网体系结构
摘要
链路思考 (CoT) 提示已在处理 text-attributed graphs (TAGs) 的大型语言模型 (LLM) 推理能力方面显示出前景. 本工作通过将 CoT 图学习重新诠释为 clustering as reasoning 的原则, 提供了链路思考图学习的 -means 解释. 我们观察到, 现有图 CoT 方法依赖于离散架构和固定图表示, 限制了 step-by-step 的语义-拓扑交互和可解释性. 为克服这一限制, 我们提出了统一框架 KCoT, 将 CoT 推理与图表示学习集成. 我们的关键理论结果揭示了 Transformer 块与 -means 算法之间的形式数学对应关系, 使推理可被解释为迭代分配和更新步骤. 基于此洞见, 我们引入了 Semantic Discriminating Prompt, 将这些步骤明确形式化为结构化 CoT 推理, 并提出 structure-grounded alignment 策略以融合拓扑先验与演化 thought-conditioned 表示. 实验表明, 在标准基准上, KCoT consistently 超过最新方法, 验证了将聚类作为 CoT 图学习的原则机制.
引用
@article{arxiv.2605.24867,
title = {Clustering as Reasoning: A $k$-Means Interpretation of Chain-of-Thought Graph Learning},
author = {Xuanting Xie and Zhaochen Guo and Bingheng Li and Xingtong Yu and Zhifei Liao and Zhao Kang and Yuan Fang},
journal= {arXiv preprint arXiv:2605.24867},
year = {2026}
}
备注
Accepted by ICML 2026