Tasteful AI agents improve decisions in complex tasks

The Tasteful Agent: Measuring and Improving Taste in Long-Horizon Tasks

Artificial Intelligence

Summary

AI agents often need to make a series of decisions that affect long tasks, but existing tests only check if the final result is successful, not whether the agent made good choices along the way. The authors created Taste-Bench, a new test that asks AI models to pick the best option at key decision points without knowing the outcomes, revealing how good the agents’ judgment is. Their experiments show even top models struggle with these decisions, especially when the best choice depends on information only seen later. They also found it is possible to train AI to make better choices by teaching one model from another that has seen the full outcome.

What this means in practice

  • For software engineers: Improve AI assistants’ decision making to build better software by evaluating and training their choices at critical development steps.
  • For product development teams: Enhance AI tools that recommend research directions by enabling evaluation and refinement of decision quality during explorations.

Authors

Wenbo Pan, Zhichao Liu, Shujie Liu, Jingying Zeng, Chin-Yew Lin, Xianfeng Tang, Yan Lu, Qi He, Xiaohua Jia

Abstract

LLM agents increasingly work on long-horizon tasks, and the decisions they make along the way, such as which hypothesis to test or which implementation to build on, determine the outcome of the whole run. Making these decisions well is becoming a key capability for both engineering and research agents. We refer to the ability to make good long-horizon decisions as the taste of an agent. While existing benchmarks measure the end-to-end success of agents on long-horizon tasks, none of them measures the taste of an agent. To address this problem, we build Taste-Bench, a benchmark of taste questions constructed automatically from trajectories that agents produced in engineering and research tasks. Each question presents a decision fork, a point in a trajectory where multiple directions are available and one of them leads to a better outcome, and the evaluated model chooses among these directions without seeing what happens after the fork. We mine these forks automatically from parallel attempts at the same task and from detours inside a single trajectory, without needing human annotation. We evaluate frontier models on Taste-Bench and find that the best model answers only 59.7% of the questions correctly. We further find that forks whose deciding evidence appears later in the trajectory are much harder for every model, and that a larger reasoning budget does not improve the accuracy. Finally, we show that taste can be trained. We distill the judgment of a teacher that has seen the outcome into a student model, and the student makes better decisions on unseen tasks and improves end-to-end success on held-out SWE-bench Pro tasks.