← Search

Yifei Zuo

2 accepted papers

2026

Local Linear Attention: An Optimal Interpolation of Linear and Softmax Attention For Test-Time Regression

ICLR 2026poster

Transformer architectures have achieved remarkable success in various domains. While efficient alternatives to Softmax Attention have been widely studied, the search for more expressive mechanisms grounded in theoretical insight—even at greater computational cost—has been relatively underexplored. I…

Cited by 3SourcecodeScholar
2025

BRiTE: Bootstrapping Reinforced Thinking Process to Enhance Language Model Reasoning

ICML 2025poster

Large Language Models (LLMs) have demonstrated remarkable capabilities in complex reasoning tasks, yet generating reliable reasoning processes remains a significant challenge. We present a unified probabilistic framework that formalizes LLM reasoning through a novel graphical model incorporating lat…

Cited by 1SourcePDFScholar