Enabling Self-Improving Agents to Learn at Test Time With Human-In-The-Loop Guidance
This addresses the challenge for LLM agents in rapidly evolving domains like compliance and risk screening, though it is incremental as it builds on existing self-improving agent concepts.
The paper tackles the problem of LLM agents struggling to adapt to changing rules and domain knowledge in dynamic environments like regulatory compliance, by proposing the ARIA framework that enables continuous learning at test time with human guidance, resulting in significant improvements in adaptability and accuracy compared to baselines, as demonstrated in a deployment serving over 150 million users.
Large language model (LLM) agents often struggle in environments where rules and required domain knowledge frequently change, such as regulatory compliance and user risk screening. Current approaches, like offline fine-tuning and standard prompting, are insufficient because they cannot effectively adapt to new knowledge during actual operation. To address this limitation, we propose the Adaptive Reflective Interactive Agent (ARIA), an LLM agent framework designed specifically to continuously learn updated domain knowledge at test time. ARIA assesses its own uncertainty through structured self-dialogue, proactively identifying knowledge gaps and requesting targeted explanations or corrections from human experts. It then systematically updates an internal, timestamped knowledge repository with provided human guidance, detecting and resolving conflicting or outdated knowledge through comparisons and clarification queries. We evaluate ARIA on the realistic customer due diligence name screening task on TikTok Pay, alongside publicly available dynamic knowledge tasks. Results demonstrate significant improvements in adaptability and accuracy compared to baselines using standard offline fine-tuning and existing self-improving agents. ARIA is deployed within TikTok Pay serving over 150 million monthly active users, confirming its practicality and effectiveness for operational use in rapidly evolving environments.