Delen
Turning delayed outcomes into meaningful feedback that helps AI agents learn more effectively

Helping AI learn from delayed feedback

2 september 2026

PhD researcher Yudi Zhang developed new methods that help AI agents learn which decisions contributed to success or failure, even when feedback only arrives at the end of a task.

/

Artificial intelligence is becoming increasingly capable of performing tasks independently. AI agents can navigate environments, make decisions, work together with other agents, and use large language models to carry out complex tasks. But as these tasks become longer and more complicated, a fundamental challenge remains: how does an AI agent know which of its decisions were responsible for the final outcome?

Imagine an AI agent completing a task that requires hundreds of decisions. If it succeeds, it receives a reward. If it fails, it receives nothing. But neither outcome tells the agent which decisions along the way were useful and which ones caused the problem. This makes it difficult for the agent to learn from its experience.

This challenge is known as the credit assignment problem. It is particularly important for AI systems that learn through interaction, where feedback may be sparse, delayed, or difficult to interpret.

PhD researcher investigated how this delayed feedback can be turned into more useful guidance for AI agents. She developed new methods that identify which decisions, information, and agents contributed to an outcome, allowing AI systems to learn more effectively while also making their learning process easier to understand. She defended her PhD thesis at the Department of Mathematics and Computer Science on Tuesday, September 1, 2026.

From a final score to meaningful feedback

Zhang鈥檚 research first focused on reinforcement learning, a form of machine learning in which an AI agent learns by interacting with its environment and receiving rewards.

In many reinforcement-learning tasks, the reward only becomes available after a sequence of decisions has been completed. Zhang developed methods to redistribute this final reward across the individual steps that contributed to the outcome. Her approach also identifies which parts of the agent鈥檚 observations are relevant to the reward and which are not.

This provides the agent with more informative feedback during learning. At the same time, it offers insight into why the agent succeeded or failed.

Zhang extended this approach to AI agents that learn directly from images. Visual environments contain a huge amount of information, but only some of it is relevant to the task. Zhang developed a method that breaks complex visual observations down into different components, allowing the AI agent to distinguish features that influence its reward from those that are unrelated to the outcome. This helps the agent focus on the information that matters when learning how to make successful decisions.

Giving credit to the right AI agent

The problem becomes more complex when AI agents have to cooperate. If several agents work together towards a shared goal, they may receive a single team reward when the task is completed. But this does not reveal how much each individual agent contributed.

Zhang developed a method that breaks down the team鈥檚 overall result and attributes contributions to individual agents and decision steps. This gives cooperating agents more detailed feedback about their role in the team鈥檚 success or failure.

Such information can help AI agents coordinate more effectively and learn from collective outcomes. It also makes the behaviour of multi-agent systems easier to interpret.

Teaching language-model agents from experience

The research also explores how these ideas can be applied to agents powered by large language models (LLMs). LLM-based agents are increasingly being used to perform tasks that require multiple decisions and interactions rather than simply generating a single response.

Zhang developed a method that extracts understandable rules from previous experiences and uses these rules to guide future decisions. This allows an agent to reuse knowledge it has gained from earlier tasks.

In another approach, she converts the final outcome of a task into feedback for the intermediate decisions that led to it. This allows language-model agents to improve through repeated interaction, without requiring humans to provide detailed feedback for every individual step.

/

Making AI more interpretable and better at learning

A key finding of Zhang鈥檚 research is that interpretability can play a more active role in artificial intelligence. It is not only useful to understand an AI agent鈥檚 behaviour after it has made a decision. Making the information, decisions, and contributions that matter explicit can also help the agent learn more efficiently and reliably.

Across different types of AI agents, Zhang鈥檚 research shows that delayed outcomes can be transformed into specific, step-by-step learning signals without changing the original objective of the task.

This could contribute to the development of more transparent and dependable AI systems for applications such as robotics, autonomous decision-making, digital assistants, and cooperation between multiple AI agents. As AI systems increasingly learn and act with less human supervision, understanding which decisions lead to successful outcomes may be essential for making them more effective, reliable, and interpretable.


PhD researcher Yudi Zhang. 

  • Supervisors

    Mykola Pechenizkiy, Meng Fang

Written by

Bouri, Danai
(Communications Advisor M&CS)

More on AI and Data Science

Latest news

keep following us