BAM: Bayes with Adaptive Memory
Josue Nassar, Jennifer Rogers Brennan, Ben Evans, Kendall Lowrey
Abstract
Online learning via Bayes' theorem allows new data to be continuously integrated into an agent's current beliefs. However, a naive application of Bayesian methods in non-stationary environments leads to slow adaptation and results in state estimates that may converge confidently to the wrong parameter value. A common solution when learning in changing environments is to discard/downweight past data; however, this simple mechanism of "forgetting" fails to account for the fact that many real-world environments involve revisiting similar states. We propose a new framework, Bayes with Adaptive Memory (BAM), that takes advantage of past experience by allowing the agent to choose which past observations to remember and which to forget. We demonstrate that BAM generalizes many popular Bayesian update rules for non-stationary environments. Through a variety of experiments, we demonstrate the ability of BAM to continuously adapt in an ever-changing world.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on3
- Variational Bayesian UnlearningQuoc Phong Nguyen, Bryan Kian Hsiang Low, Patrick JailletNeurIPS 2020 · 198 citations
- Continual Learning with Bayesian Neural Networks for Non-Stationary DataRichard Kurle, Botond Cseke, Alexej Klushyn, Patrick van der Smagt et al.ICLR 2020 · 82 citations
- Detecting and Adapting to Irregular Distribution Shifts in Bayesian Online LearningAodong Li, Alex Boyd, Padhraic Smyth, Stephan MandtNeurIPS 2021 · 31 citations
Related papers
- Kalman Filter for Online Classification of Non-Stationary DataMichalis K. Titsias, Alexandre Galashov, Amal Rannen-Triki, Razvan Pascanu et al.ICLR 2024 · 14 citations
- Temporal-Difference Variational Continual LearningLuckeciano Carvalho Melo, Alessandro Abate, Yarin GalNeurIPS 2025 · 1 citation
- Deep Reinforcement Learning amidst Continual Structured Non-StationarityAnnie Xie, James Harrison, Chelsea FinnICML 2021 · 43 citations
- AdaMEM: Test-Time Adaptive Memory for Language AgentsYunxiang Zhang, Yiheng Li, Ali Payani, Lu WangICML 2026
- Learning Successor Features with Distributed Hebbian Temporal MemoryEvgenii Aleksandrovich Dzhivelikian, Petr Kuderov, Aleksandr PanovICLR 2025
