arXiv Computation and Language By Tianjun Zhong, Linyang He, Ziyang Li, Nima Mesgarani

LLMs Think with DAGs: Probing the Latent Graph Structure of Reasoning

Read the original on arXiv Computation and Language →

The paper introduces Reasoning DAG Probing, a method that treats premises, intermediate conclusions, and final answers as nodes in a directed acyclic graph (DAG). By training lightweight probes on LLM hidden states, the authors show that DAG structure is encoded in the models, peaking in intermediate layers and becoming stronger with larger models. During autoregressive generation, the latent DAG structure emerges progressively, anticipating answer correctness and enabling correction of some incorrect outputs by steering activations toward the ground‑truth DAG.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv Computation and Language.

arXiv AI
Sep 10

A*-Thought-V2: Efficient Latent Reasoning via Geometric Dynamics of LLM

A*-Thought-V2 is a framework that models Chain-of-Thought reasoning as a geometric trajectory in a 3D PCA space, using explicit-implicit latent tokens to compress steps that deviate from the main question-to-solution direction. The method measures alignment angles to decide which steps remain text and which become latent, and introduces stepwise embedding forcing and label forcing to train the architecture. Experiments on Qwen models show up to 2.6% accuracy gains, halved response length, and significant reductions in computation and training time.

By Xiaoang Xu, Siyuan Liu, Shuo Wang, Junlan Feng, Fanyu Meng, Zhu Zhang, Jixun Wang, Xiaorong Wang, Zihan Zhou, Xin Li, Chaojun Xiao, Yiming Zhang, Huijia Wu, Liuyu Xiang, Peipei Li, Zhaofeng He