← Search

Isys Johnson

4 accepted papers

2024

Zoology: Measuring and Improving Recall in Efficient Language Models

ICLR 2024poster

Attention-free language models that combine gating and convolutions are growing in popularity due to their efficiency and increasingly competitive performance. To better understand these architectures, we pretrain a suite of 17 attention and gated-convolution language models, finding that SoTA gated…

2023

How to Train your HIPPO: State Space Models with Generalized Orthogonal Basis Projections

ICLR 2023poster

Linear time-invariant state space models (SSM) are a classical model from engineering and statistics, that have recently been shown to be very promising in machine learning through the Structured State Space sequence model (S4). A core component of S4 involves initializing the SSM state matrix to a…

2023

Monarch Mixer: A Simple Sub-Quadratic GEMM-Based Architecture

NeurIPS 2023oral

Machine learning models are increasingly being scaled in both sequence length and model dimension to reach longer contexts and better performance. However, existing architectures such as Transformers scale quadratically along both these axes. We ask: are there performant architectures that can scale…

Cited by 59SourcePDFScholar
2021

Combining Recurrent, Convolutional, and Continuous-time Models with Linear State Space Layers

NeurIPS 2021poster

Recurrent neural networks (RNNs), temporal convolutions, and neural differential equations (NDEs) are popular families of deep learning models for time-series data, each with unique strengths and tradeoffs in modeling power and computational efficiency. We introduce a simple sequence model inspired…