English

BayesFlow: A Probability Inference Framework for Meta-Agent Assisted Workflow Generation

Machine Learning 2026-02-02 v1

Abstract

Automatic workflow generation is the process of automatically synthesizing sequences of LLM calls, tool invocations, and post-processing steps for complex end-to-end tasks. Most prior methods cast this task as an optimization problem with limited theoretical grounding. We propose to cast workflow generation as Bayesian inference over a posterior distribution on workflows, and introduce \textbf{Bayesian Workflow Generation (BWG)}, a sampling framework that builds workflows step-by-step using parallel look-ahead rollouts for importance weighting and a sequential in-loop refiner for pool-wide improvements. We prove that, without the refiner, the weighted empirical distribution converges to the target posterior. We instantiate BWG as \textbf{BayesFlow}, a training-free algorithm for workflow construction. Across six benchmark datasets, BayesFlow improves accuracy by up to 9 percentage points over SOTA workflow generation baselines and by up to 65 percentage points over zero-shot prompting, establishing BWG as a principled upgrade to search-based workflow design. Code will be available on https://github.com/BoYuanVisionary/BayesFlow.

Keywords

Cite

@article{arxiv.2601.22305,
  title  = {BayesFlow: A Probability Inference Framework for Meta-Agent Assisted Workflow Generation},
  author = {Bo Yuan and Yun Zhou and Zhichao Xu and Kiran Ramnath and Aosong Feng and Balasubramaniam Srinivasan},
  journal= {arXiv preprint arXiv:2601.22305},
  year   = {2026}
}

Comments

EACL 2026 Finding

R2 v1 2026-07-01T09:26:40.989Z