PACT: Privileged Trace Co-Training for Multi-Turn Tool-Use Agents

22d ago · Global · primary source: export.arxiv.org

A new training framework called PACT aims to improve how AI agents handle multi-step tasks that require tool use, according to a paper submitted to arXiv in 2026 [1]. The method uses expert demonstrations as optimization signals during training without relying on them when the agent is actually running [2]. Multi-turn tool-use agents must reason, call external tools, and adapt to observations across several interaction turns [2]. Training these agents has presented a persistent trade-off. Reinforcement learning (RL) matches the prompt-only inference setting but often suffers from sparse rewards and weak credit assignment, while supervised fine-tuning (SFT) on expert traces provides dense process supervision but can over-constrain the model to fixed trajectories [2]. The PACT framework, short for Privileged trAce Co-Training, attempts to resolve this by keeping rollout generation prompt-only and using expert traces solely as training-time optimization signals [2]. It does so through two complementary signals: a trace-conditioned RL surrogate that evaluates prompt-only rollouts under expert-trace context, and a component-aware SFT loss that supervises reasoning prefixes and tool-calls with annealed strength [2]. To reduce over-reliance on the training-only trace context, PACT introduces a prompt-only anchoring mechanism [2]. The authors also provide a latent-trace view that connects the two trace-based objectives and explains how expert traces can guide optimization without being used during rollout generation [2]. Experiments were conducted on three benchmarks — FTRL, BFCL, and ToolHop — and the results showed that PACT consistently improves over strong SFT- and RL-based baselines [2]. The paper was posted on arXiv, an open-access repository of electronic preprints that has hosted scientific papers since 1991 and now receives about 24,000 submissions per month [8]. arXiv papers are moderated but not peer reviewed [8]. The repository supports community-developed tools through its arXivLabs framework, which allows collaborators to build features such as bibliographic explorers and code finders directly on article pages [6][7].

applicationresearch-paperinfrastructuretool-release

Background sources we checked (9)
  • arxiv.org ↗ Multi-turn tool-use agents must reason, call tools, and adapt to observations across several interaction turns. Post-training such agents is challenging, as reinforcement learning often suffers from sparse rewards and weak credit assignment despite matching the prompt-only infere…
  • en.wikipedia.org ↗ Russia–European Union relations are the international relations between the European Union (EU) and Russia. Russia borders five EU member states: Estonia, Finland, Latvia, Lithuania and Poland; the Russian exclave of Kaliningrad is surrounded by EU members. Until the radical brea…
  • en.wikipedia.org ↗ A city is a human settlement of a substantial size. The term "city" has different meanings around the world and in some places the settlement can be very small. Even where the term is limited to larger settlements, there is no universally agreed definition of the lower boundary f…
  • info.arxiv.org ↗ arXiv Labs - arXiv info | arXiv e-print repository Skip to content # arXiv Labs Attention arXiv Users: arXiv Labs is pausing new proposals ## What are arXiv Labs? arXiv Labs are a way for the community to contribute new, useful features to arXiv. These integrations are avail…
  • blog.arxiv.org ↗ arXivLabs: a space for community innovation – arXiv blog arXiv has launched a new, formalized framework enabling innovative collaborations with individuals and organizations. “Members of our community want to contribute tools that enhance the arXiv experience, and we val…
  • info.arxiv.org ↗ arXivLabs: Showcase - arXiv info | arXiv e-print repository ... # arXivLabs: Showcase ... arXiv is surrounded by a community of researchers and developers working at the cutting edge of information science and technology. ... While the arXiv team is focused on our core mission—pr…
  • en.wikipedia.org ↗ arXiv (pronounced as "archive"—the X represents the Greek letter chi ⟨χ⟩) is an open-access repository of electronic preprints and postprints (known as e-prints) approved for posting after moderation, but not peer reviewed. It consists of scientific papers in the fields of mathem…
  • en.wikipedia.org ↗ 14 (fourteen) is the natural number following 13 and preceding 15.…
  • en.wikipedia.org ↗ A large language model (LLM) is a type of machine learning model designed for natural language processing tasks such as language generation. LLMs are language models with many parameters, and are trained with self-supervised learning on a vast amount of text.…

Sources covering this (2)

Spot something wrong? Report an issue