Yao co-authors FALCON, unifying SSMs and fast-weight memory
TL;DR
- Andrew Chi-Chih Yao and ten co-authors posted FALCON on arXiv August 27, framing selective state-space models and fast-weight memories as one online learning rule.
- The family comprises three squared-error regression updates (Falcon-1/2/3) plus three inner-product variants (Falcon-1A/2A/3A), each in recurrent, masked-parallel, and chunk-parallel forms.
- The abstract claims competitive language-modeling performance and improved length extrapolation on variable-digit addition, but names no baselines and publishes no benchmark numbers.
Eleven researchers, including Andrew Chi-Chih Yao, have posted a paper on arXiv arguing that recurrent fast-weight memories and selective state-space models are the same object: an online learning rule that updates a fixed-size recurrent state as each token arrives.
The paper, titled 'Fast Weight Attention for Continual Learning,' derives normalized first-order updates for squared-error regression as a family of three: 'Falcon-1 (a scalar NLMS update), Falcon-2 (its per-column extension), and Falcon-3 (a sliding-window mini-batch update).' Each has an inner-product analog (Falcon-1A/2A/3A), and the authors publish recurrent, masked-parallel, and chunk-parallel implementations with what they call 'numerically stable positive-decay renormalization.'
The empirical bar is modest. Representative variants 'remain competitive in language modeling and improve length extrapolation on variable-digit addition,' the abstract reports. It names no baselines and publishes no benchmark numbers.
The framing is what stakes the paper. If the derivation holds, selective state-space models and fast-weight networks are two parameterizations of the same regression step, one the authors say 'separates temporal alignment, plasticity, forgetting, and bounded rehearsal in recurrent sequence models.'
Originally reported by paper
Read the original article →Original headline: Turing Laureate Andrew Yao Co-Authors FALCON, Unifying SSMs and Fast-Weight Memories as Online Learning Rules