Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

AMATA: Adaptive Multi-Agent Trajectory Alignment for Knowledge-Intensive Question Answering

About

Despite substantial advances in large language models (LLMs), generating factually consistent responses for knowledge-intensive question answering remains challenging. These difficulties are primarily due to hallucinations and the limitations of LLMs in bridging long-tail knowledge gaps. To address this, we propose AMATA, an Adaptive Multi-Agent Trajectory Alignment framework that dynamically integrates external knowledge to improve response interpretability and factual grounding. Our architecture leverages six specialized agents that collaboratively perform structured actions for complex question reasoning. We formalize multi-agent collaboration with external tools as a trajectory preference alignment problem, incorporating question-aware agent customization and inter-agent preference harmonization. AMATA introduces two principal innovations: (1) Intra-Trajectory Preference Learning, which learns objective-oriented preferences to prioritize critical agents, and (2) Inter-Agent Dependency Learning, which captures cross-agent tool dependencies through a novel dependency-aware direct preference optimization technique. Empirical results show that AMATA consistently outperforms baseline approaches, knowledge-augmented frameworks, and LLM-based trajectory systems on five established knowledge-intensive QA benchmarks. Further analysis demonstrates the efficiency of our method in reducing token consumption.

Taolin Zhang, Dongyang Li, Chen Chen, Qizhou Chen, Jiuheng Wan, Xiaofeng He, Chengyu Wang, Richang Hong• 2026

Related benchmarks

TaskDatasetResultRank
Question AnsweringARC Challenge
Accuracy (ARC)74.11
598
Question AnsweringPopQA
Exact Match49.62
133
Question AnsweringASQA--
59
Multi-task EvaluationAggregate (HealthQA, ARC-C, PopQA, Squad1, ASQA)
Average Score62.8
8
Question AnsweringSQuAD v1.1
Accuracy37.8
8
Question AnsweringHealthQA
Accuracy78.74
8
Showing 6 of 6 rows

Other info

Follow for update