AAAI 2026technical0 citations
Loss-Guided Auxiliary Agents for Overcoming Mode Collapse in GFlowNets
Idriss Malek, Aya Laajil, Abhijith Sharma, Eric Moulines, Salem Lahlou
Abstract
Although Generative Flow Networks (GFlowNets) are designed to capture multiple modes of a reward function, they often suffer from mode collapse in practice, getting trapped in early-discovered modes and requiring prolonged training to find diverse solutions. Existing exploration techniques often rely on heuristic novelty signals. We propose Loss-Guided GFlowNets (LGGFN), a novel approach where an auxiliary GFlowNet
BibTeX
@inproceedings{aaai2026_lossguidedauxili,
title = {Loss-Guided Auxiliary Agents for Overcoming Mode Collapse in GFlowNets},
author = {Idriss Malek and Aya Laajil and Abhijith Sharma and Eric Moulines and Salem Lahlou},
booktitle = {AAAI 2026},
year = {2026}
}