Topological Active Inference for Task Disambiguation
Yangbo Wei, Zhen Huang, Shaoqiang Lu, Junhong Qian, Chen Wu, Lei He
Abstract
In open-ended domains, natural language instructions are often underspecified, mapping to multiple valid yet functionally distinct latent intents. Although Large Language Models (LLMs) excel at generation, their interactive disambiguation remains limited by semantic blindness: they may spend clarification turns distinguishing superficial syntactic variants rather than resolving substantive intent differences. We propose Topological Active Inference (TAI), a geometric framework that recasts task disambiguation as intent-manifold contraction. TAI uses Persistent Homology to recover persistent intent clusters from sampled solutions, filtering short-lived syntactic variations while preserving robust semantic structure under mild separability assumptions. It then synthesizes clarifying questions as semantic separators and selects them with Topological Expected Information Gain (TEIG), which optimizes uncertainty reduction over intent clusters rather than individual candidates. This reduces the effective hypothesis space from sampled solutions to latent intents and yields logarithmic interaction complexity under balanced-split conditions. Experiments across code, visualization, and navigation tasks show that TAI resolves user intent with fewer turns and remains robust to noisy feedback, and smaller model scales.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 1cf3fcb2-be3c-4767-bb35-270e14083774Builds on10
- Training language models to follow instructions with human feedbackLong Ouyang, Jeffrey Wu, Xu Jiang, Diogo Almeida et al.NeurIPS 2022 · 24,707 citations
- The Effects of Reward Misspecification: Mapping and Mitigating Misaligned ModelsAlexander Pan, Kush Bhatia, Jacob SteinhardtICLR 2022 · 293 citations
- AmbigQA: Answering Ambiguous Open-domain QuestionsSewon Min, Julian Michael, Hannaneh Hajishirzi, Luke ZettlemoyerEMNLP 2020 · 162 citations
- Active Example Selection for In-Context LearningYiming Zhang, Shi Feng, Chenhao TanEMNLP 2022 · 84 citations
- Active Prompting with Chain-of-Thought for Large Language ModelsShizhe Diao, Pengcheng Wang, Yong Lin, Rui Pan et al.ACL 2024 · 40 citations
Related papers
- Active Task Disambiguation with LLMsKasia Kobalczyk, Nicolás Astorga, Tennison Liu, Mihaela van der SchaarICLR 2025
- Large Language Models as Topological Thinkers: A Benchmark on Graph Persistent HomologyHao Li, Hao Wan, Yixue Huang, Yuzhou Chen et al.ICML 2026
- Uncertainty-Aware Clarification in LLM Agents with Information GainMengyi DENG, Zhiwei Li, Xin Li, Tingyu ZHU et al.ICML 2026 · 1 citation
- The Shape of Adversarial Influence: Characterizing LLM Latent Spaces with Persistent HomologyAideen Fay, Inés García-Redondo, Qiquan Wang, Haim Dubossarsky et al.ICLR 2026 · 8 citations
- Ambig-SWE: Interactive Agents to Overcome Underspecificity in Software EngineeringSanidhya Vijayvargiya, Xuhui Zhou, Akhila Yerukola, Maarten Sap et al.ICLR 2026 · 35 citations
