Search papers, labs, and topics across Lattice.
GFlowState is introduced as a visual analytics system to enhance the interpretability of Generative Flow Network (GFlowNet) training. It provides multiple views, including trajectory networks and transition heatmaps, to analyze sampling behavior, policy evolution, and identify training failures. Case studies demonstrate its utility in debugging and assessing GFlowNets across various applications, ultimately accelerating GFlowNet development.
Uncover hidden GFlowNet training dynamics with GFlowState, a visual analytics tool that reveals how these models explore the sample space and shift sampling probabilities.
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.