martin592/hybrid_AuGMEnT

Development of the memory dynamics in AuGMEnT network to enable learning on hierarchical tasks, like 12ax. We introduced a differentiable leaky control system that allows the network to forget non-relevant cues. Thus, we implemented a variant of the AuGMEnT network, named hybrid AuGMEnT, which employs an hybrid memory with both leaky and non-leaky cells in order to learn on multiple termporal scales.

Python