Transformers and Cortical Waves: Encoders for Pulling In Context Across Time

ML TAVR Transformer
2024年01月25日
transformer网络(如ChatGPT和其他大型语言模型)的能力引起了世界的关注。它们的关键计算机制依赖于将完整的输入序列(例如,句子中的所有单词)转换为一个长的“编码向量”,使transformer能够学习自然序列中的长程时间依赖关系。具体来说,应用于该编码向量的“自注意力”通过计算输入序列中单词对之间的关联来增强transformer中的时间上下文。我们认为,神经活动的波,无论是在单个大脑皮层区域内还是在整个大脑尺度上跨越多个区域,都可以实现类似的编码原则。通过将最近的输入历史封装为每个时间点的单个空间模式,皮层波可能使时间上下文从感觉输入序列中提取出来,这是transformer使用的相同计算原理。
The capabilities of transformer networks such as ChatGPT and other Large Language Models (LLMs) have captured the world's attention. The crucial computational mechanism underlying their performance relies on transforming a complete input sequence - for example, all the words in a sentence into a long "encoding vector" - that allows transformers to learn long-range temporal dependencies in naturalistic sequences. Specifically, "self-attention" applied to this encoding vector enhances temporal context in transformers by computing associations between pairs of words in the input sequence. We suggest that waves of neural activity, traveling across single cortical regions or across multiple regions at the whole-brain scale, could implement a similar encoding principle. By encapsulating recent input history into a single spatial pattern at each moment in time, cortical waves may enable temporal context to be extracted from sequences of sensory inputs, the same computational principle used in transformers.
许愿