← Search

Morris Yau

6 accepted papers

2026

Sequential Parallel Duality in Prefix Scannable Models

ICLR 2026poster

Modern neural sequence models are designed to meet the dual mandate of parallelizable training and fast sequential inference. Recent developments have given rise to various models, such as Gated Linear Attention (GLA) and Mamba, that achieve such ``sequential-parallel duality.'' This raises a natura…

Cited by 0SourceScholar
2025

Blending Complementary Memory Systems in Hybrid Quadratic-Linear Transformers

NeurIPS 2025poster

We develop hybrid memory architectures for general-purpose sequence processing neural networks, that combine key-value memory using softmax attention (KV-memory) with fast weight memory through dynamic synaptic modulation (FW-memory)---the core principles of quadratic and linear transformers, respec…

Cited by 0SourceScholar
2025

Learning Linear Attention in Polynomial Time

NeurIPS 2025oral

Previous research has explored the expressivity of Transformer models in simulating Boolean circuits or Turing machines. However, the efficient learnability of Transformers from data has remained an open question. Our study addresses this gap by providing the first polynomial-time learnability resu…

Cited by 0SourceScholar
2024

Are Graph Neural Networks Optimal Approximation Algorithms?

NeurIPS 2024spotlight

In this work we design graph neural network architectures that capture optimal approximation algorithms for a large class of combinatorial optimization problems, using powerful algorithmic tools from semidefinite programming (SDP). Concretely, we prove that polynomial-sized message-passing algorithm…

2023

Tensor Decompositions Meet Control Theory: Learning General Mixtures of Linear Dynamical Systems

ICML 2023poster

Recently Chen and Poor initiated the study of learning mixtures of linear dynamical systems. While linear dynamical systems already have wide-ranging applications in modeling time-series data, using mixture models can lead to a better fit or even a richer understanding of underlying subpopulations r…

Cited by 8SourcePDFScholar
2020

Classification Under Misspecification: Halfspaces, Generalized Linear Models, and Evolvability

NeurIPS 2020spotlight

In this paper, we revisit the problem of distribution-independently learning halfspaces under Massart noise with rate $\eta$. Recent work resolved a long-standing problem in this model of efficiently learning to error $\eta + \epsilon$ for any $\epsilon > 0$, by giving an improper learner that parti…