acceptodds
Under review as a conference paper at ICLR 2027

When Are GFlowNets Scalable? A Regret Analysis of Sample Efficiency

Abstract

Generative Flow Networks (GFlowNets) are designed to learn distributions over combinatorial objects whose underlying state spaces can be prohibitively large. While substantial progress has been made in improving GFlowNet training, through better trajectory sampling, exploration, and gradient estimation, a basic theoretical question remains unsolved: *when is a GFlowNet statistically scalable, and how does the structure of the task determine its sample efficiency?* We study this question through regret, where the performance gap is the trajectory KL divergence from the reward-proportional target distribution. Our central finding is that the relevant statistical scale is the representation dimension , rather than the state space size. We first establish an oracle-sampling lower bound of . This bound reveals a data-starved regime: when , the worst-case average performance gap does not vanish, so the trajectory budget must reach the scale to leave this regime. We then investigate whether the statistical scaling of the lower bound can be approached constructively. We develop an optimistic policy-learning algorithm that leverages a shared -dimensional representation of policy values. A first-order analysis yields regret , and a second-order analysis further improves the dependence on the trajectory budget to be logarithmic under the stated assumptions. Together, these results identify representation dimension as a fundamental measure of GFlowNet learnability: enormous state spaces of GFlowNet can remain statistically scalable when their relevant value structure admits a compact shared representation, whereas without such compression the effective dimension can approach the state space size. This provides a formal step toward the theory of the statistical scalability of GFlowNets.

open until 14 Dec 2026

est. 32% chance this paper gets accepted at ICLR 2027.

Reject 68%Accept 32%

What do you think this paper will get?

All positions stay anonymous.

Related papers

Loading the map…

Discussion (0)

Sign in to comment.