English

TDA-RC: Task-Driven Alignment for Knowledge-Based Reasoning Chains in Large Language Models

Computation and Language 2026-04-08 v1 Artificial Intelligence

Abstract

Enhancing the reasoning capability of large language models (LLMs) remains a core challenge in natural language processing. The Chain-of-Thought (CoT) paradigm dominates practical applications for its single-round efficiency, yet its reasoning chains often exhibit logical gaps. While multi-round paradigms like Graph-of-Thoughts (GoT), Tree-of-Thoughts (ToT), and Atom of Thought (AoT) achieve strong performance and reveal effective reasoning structures, their high cost limits practical use. To address this problem, this paper proposes a topology-based method for optimizing reasoning chains. The framework embeds essential topological patterns of effective reasoning into the lightweight CoT paradigm. Using persistent homology, we map CoT, ToT, and GoT into a unified topological space to quantify their structural features. On this basis, we design a unified optimization system: a Topological Optimization Agent diagnoses deviations in CoT chains from desirable topological characteristics and simultaneously generates targeted strategies to repair these structural deficiencies. Compared with multi-round reasoning methods like ToT and GoT, experiments on multiple datasets show that our approach offers a superior balance between reasoning accuracy and efficiency, showcasing a practical solution to ``single-round generation with multi-round intelligence''.

Keywords

Cite

@article{arxiv.2604.04942,
  title  = {TDA-RC: Task-Driven Alignment for Knowledge-Based Reasoning Chains in Large Language Models},
  author = {Jiaquan Zhang and Qigan Sun and Chaoning Zhang and Xudong Wang and Zhenzhen Huang and Yitian Zhou and Pengcheng Zheng and Chi-lok Andy Tai and Sung-Ho Bae and Zeyu Ma and Caiyan Qin and Jinyu Guo and Yang Yang and Hengtao Shen},
  journal= {arXiv preprint arXiv:2604.04942},
  year   = {2026}
}

Comments

14 pages, 4 figures