All talks

Seminar Talks

The Illusion of Deep Learning Architecture
Ali Behrouz
Date Friday, Oct. 16
Time 14:30
Place N10_302, Institute of Computer Science
Description

Designing efficient and effective architectural backbones as well as optimization algorithms have been in the core of research efforts to enhance the capability of foundation models. We reconceptualize neural architectures as well as optimization algorithms as associative memory modules that learn a mapping of keys and values using an internal objective, referred to as attentional bias. Surprisingly, we observed that most existing sequence models leverage either (1) dot-product similarity, or (2) L2 regression objectives as their attentional bias. Going beyond these objectives, we present a set of alternative attentional bias configurations along with their effective approximations to stabilize their training procedure. We then reinterpret forgetting mechanisms in modern deep learning architectures as a form of retention regularization, providing a novel set of forget gates for sequence models. Building upon these insights, we present Miras, a general framework to design deep learning architectures based on four choices of: (i) associative memory architecture, (ii) attentional bias objective, (iii) retention gate, and (iv) memory learning algorithm. We present novel sequence models, including Titans, Atlas, and Hope, that show exceptional performance in tasks such as language modeling, commonsense reasoning, and recall intensive tasks, even outperforming Transformers and other modern linear recurrent models.

 

Zoom link: Click Here

Details