mlsanityData to 14 Oct 2026

GFlowState: Visualizing the Training of Generative Flow Networks Beyond the Reward

arXiv 2604.21830 ↗First posted 23 Apr 2026cs.LG cs.HCcited by 0

Not on the map. Why?

Abstract (arXiv)

We present GFlowState, a visual analytics system designed to illuminate the training process of Generative Flow Networks (GFlowNets or GFNs). GFlowNets are a probabilistic framework for generating samples proportionally to a reward function. While GFlowNets have proved to be powerful tools in applications such as molecule and material discovery, their training dynamics remain difficult to interpret. Standard machine learning tools allow metric tracking but do not reveal how models explore the sample space, construct sample trajectories, or shift sampling probabilities during training. Our solution, GFlowState, allows users to analyze sampling trajectories, compare the sample space relative to reference datasets, and analyze the training dynamics. To this end, we introduce multiple views, including a chart of candidate rankings, a state projection, a node-link diagram of the trajectory network, and a transition heatmap. These visualizations enable GFlowNet developers and users to investigate sampling behavior and policy evolution, and to identify underexplored regions and sources of training failure. Case studies demonstrate how the system supports debugging and assessing the quality of GFlowNets across application domains. By making the structural dynamics of GFlowNets observable, our work enhances their interpretability and can accelerate GFlowNet development in practice.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 22 of its 34 references that are in the map. First posted on arXiv 23 Apr 2026.

Builds on

22 of 34 references.

References · 22

  1. 2021Flow Network based Generative Models for Non-Iterative Diverse Candidate Generationcited by 392
  2. 2021GFlowNet Foundationscited by 290
  3. 2019Tackling Climate Change with Machine Learningcited by 265
  4. 2022Trajectory balance: Improved credit assignment in GFlowNetscited by 251
  5. 2020The Open Catalyst 2020 (OC20) Dataset and Community Challengescited by 246
  6. 2018Visual Analytics in Deep Learning: An Interrogative Survey for the Next Frontierscited by 117
  7. 2023Towards Understanding and Improving GFlowNet Trainingcited by 75
  8. 2018Seq2Seq-Vis: A Visual Debugging Tool for Sequence-to-Sequence Modelscited by 69
  9. 2016Graying the black box: Understanding DQNscited by 68
  10. 2023GFlowNets for AI-Driven Scientific Discoverycited by 63
  11. 2020CNN Explainer: Learning Convolutional Neural Networks with Interactive Visualizationcited by 57
  12. 2020A Survey of Visual Analytics Techniques for Machine Learningcited by 56
10 more
  1. 2018GAN Lab: Understanding Complex Deep Generative Models using Interactive Visual Experimentationcited by 41
  2. 2021Why? Why not? When? Visual Explanations of Agent Behavior in Reinforcement Learningcited by 16
  3. 2019ConfusionFlow: A model-agnostic visualization for temporal analysis of classifier confusioncited by 15
  4. 2022Visual Comparison of Language Model Adaptationcited by 14
  5. 2023Revealing the Unwritten: Visual Investigation of Beam Search Trees to Address Language Model Prompting Challengescited by 6
  6. 2020InstanceFlow: Visualizing the Evolution of Classifier Confusion on the Instance Levelcited by 5
  7. 2024generAItor: Tree-in-the-Loop Text Generation for Language Model Explainability and Adaptationcited by 5
  8. 2018Juniper: A Tree+Table Approach to Multivariate Graph Visualizationcited by 3
  9. 2024EvolvED: Evolutionary Embeddings to Understand the Generation Process of Diffusion Modelscited by 3
  10. 2025Catalyst GFlowNet for electrocatalyst design: A hydrogen evolution reaction case studycited by 3

Built on by

Not cited yet.