arXiv Machine Learning By Timofei Gritsaev, Evgeny Katasonov, Ildus Sadrtdinov, Esmeralda S. Whitammer, Dmitry Vetrov

On the Generation Order in GFlowNets

Read the original on arXiv Machine Learning →

The paper investigates how the order in which components are generated in GFlowNets affects learning and convergence. It shows that existing GFlowNet algorithms often fail to discover useful generation orders, and introduces a new training recipe—Learning-Order GFlowNet—that learns interpretable orders without explicit annotations. Experiments demonstrate that the new method improves convergence, and analysis of gradient signal-to-noise ratio suggests that higher SNR correlates with faster convergence and better generation orders.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv Machine Learning.

arXiv Machine Learning
Aug 31

Trajectory balance: Improved credit assignment in GFlowNets

The paper introduces a new learning objective called trajectory balance for Generative Flow Networks (GFlowNets), aiming to improve credit assignment across long action sequences. It demonstrates that minimizing this objective yields a policy that samples exactly from the target distribution. Experiments on four domains show that trajectory balance enhances convergence, sample diversity, and robustness to long sequences and large action spaces.

By Esmeralda S. Whitammer, Moksh Jain, Emmanuel Bengio, Chen Sun, Yoshua Bengio
arXiv Machine Learning
Aug 6

Stable GFlowNets with TV Monitoring and Probabilistic Guarantees

arXiv:2605. 01729v2 Announce Type: replace Abstract: Generative Flow Networks (GFlowNets) learn to sample states proportional to an unnormalized reward.

By Zengxiang Lei, Ananth Shreekumar, Jonathan Rosenthal, Ruoyu Song, Alvaro A. Cardenas, Daniel J. Fremont, Dongyan Xu, Satish Ukkusuri, Z. Berkay Celik
arXiv AI
Oct 27

Hierarchical Cooperative GFlowNets

Hierarchical Cooperative GFlowNets (HCG) dynamically partition the search space to coordinate multiple GFlowNets, preventing them from collapsing onto the same high‑reward regions. HCG imposes an ordering where the top model explores the full space while lower‑ranked models focus on uncovered areas, using a shared classifier and a normalizer‑weighted aggregation rule to maintain reward‑proportional sampling. Experiments on multimodal distributions, sequence generation, combinatorial optimization, and biological sequence design demonstrate that HCG improves mode coverage and target approximation, enabling broader complementary exploration.

By Pedro Dall'Antonia, Juan Belieni, Ana Carolina Erthal, Saullo H. G. de Oliveira, Eliezer de Souza da Silva, Diego Mesquita