Search papers, labs, and topics across Lattice.
This paper introduces BOT-Orch, a bandit-based framework for adaptive agent orchestration that explicitly models uncertainty in agent reliability and output distributions using Optimal Transport (OT) regularization. BOT-Orch formulates orchestration as a multi-armed bandit problem where the reward is regularized by the OT distance between agent output distributions and task-specific reference distributions. The framework achieves $\mathcal{O}(\sqrt{T})$ regret and demonstrably outperforms standard bandit and heuristic baselines in heterogeneous, non-i.i.d. agent environments.
Coordinating AI agents gets a reliability boost: BOT-Orch uses bandit learning and Optimal Transport to intelligently delegate tasks, even when agents are unpredictable.
Adaptive orchestration of heterogeneous agents requires making sequential delegation decisions under uncertain and evolving agent behaviour, e.g., coordinating specialised AI models with varying reliability, cost, and response quality. While prior work on agent orchestration focuses on performance or cost, uncertainty in agent reliability and output distributions is typically not modelled explicitly at the orchestration level. In this work, we study the problem of adaptive orchestration of heterogeneous agents under uncertainty, where a meta-controller must decide when to delegate to an agent, accounting for reliability, cost, and uncertainty. We propose BOT-Orch, a lightweight framework that recasts orchestration as a bandit problem over agents, regularized by OT distances between agent output distributions and task-specific reference distributions. We show that the regularised orchestration enjoys $\mathcal{O}(\sqrt{T})$ regret under standard assumptions, and provably induces preference ordering among agents with identical mean rewards but differing distributional alignment. Empirically, we demonstrate that BOT-Orch outperforms standard bandit and heuristic baselines in synthetic but adversarial task allocation settings with heterogeneous, non-i.i.d. agent behaviour.
