On Generalization for Generative Flow Networks

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Krichel, Anas, Malkin, Nikolay, Lahlou, Salem, Bengio, Yoshua
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866911943196934144
author Krichel, Anas
Malkin, Nikolay
Lahlou, Salem
Bengio, Yoshua
author_facet Krichel, Anas
Malkin, Nikolay
Lahlou, Salem
Bengio, Yoshua
contents Generative Flow Networks (GFlowNets) have emerged as an innovative learning paradigm designed to address the challenge of sampling from an unnormalized probability distribution, called the reward function. This framework learns a policy on a constructed graph, which enables sampling from an approximation of the target probability distribution through successive steps of sampling from the learned policy. To achieve this, GFlowNets can be trained with various objectives, each of which can lead to the model s ultimate goal. The aspirational strength of GFlowNets lies in their potential to discern intricate patterns within the reward function and their capacity to generalize effectively to novel, unseen parts of the reward function. This paper attempts to formalize generalization in the context of GFlowNets, to link generalization with stability, and also to design experiments that assess the capacity of these models to uncover unseen parts of the reward function. The experiments will focus on length generalization meaning generalization to states that can be constructed only by longer trajectories than those seen in training.
format Preprint
id arxiv_https___arxiv_org_abs_2407_03105
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle On Generalization for Generative Flow Networks
Krichel, Anas
Malkin, Nikolay
Lahlou, Salem
Bengio, Yoshua
Machine Learning
Generative Flow Networks (GFlowNets) have emerged as an innovative learning paradigm designed to address the challenge of sampling from an unnormalized probability distribution, called the reward function. This framework learns a policy on a constructed graph, which enables sampling from an approximation of the target probability distribution through successive steps of sampling from the learned policy. To achieve this, GFlowNets can be trained with various objectives, each of which can lead to the model s ultimate goal. The aspirational strength of GFlowNets lies in their potential to discern intricate patterns within the reward function and their capacity to generalize effectively to novel, unseen parts of the reward function. This paper attempts to formalize generalization in the context of GFlowNets, to link generalization with stability, and also to design experiments that assess the capacity of these models to uncover unseen parts of the reward function. The experiments will focus on length generalization meaning generalization to states that can be constructed only by longer trajectories than those seen in training.
title On Generalization for Generative Flow Networks
topic Machine Learning
url https://arxiv.org/abs/2407.03105