中文
相关论文

相关论文: Beyond Linear Attention: Softmax Transformers Impl…

200 篇论文

In-context learning (ICL) is a cornerstone of large language model (LLM) functionality, yet its theoretical foundations remain elusive due to the complexity of transformer architectures. In particular, most existing work only theoretically…

机器学习 · 计算机科学 2024-09-18 Siyu Chen , Heejune Sheen , Tianhao Wang , Zhuoran Yang

In-context learning (ICL) allows Transformers to adapt to novel tasks without weight updates, yet the underlying algorithms remain poorly understood. We adopt a statistical decision-theoretic perspective by investigating simple binary…

机器学习 · 计算机科学 2026-03-13 Faris Chaudhry , Siddhant Gadkari

The transformer architecture, which processes sequences of input tokens to produce outputs for query tokens, has revolutionized numerous areas of machine learning. A defining feature of transformers is their ability to perform previously…

机器学习 · 计算机科学 2025-10-02 Hongbo Li , Lingjie Duan , Yingbin Liang

In-context learning (ICL) is a key building block of modern large language models, yet its theoretical mechanisms remain poorly understood. It is particularly mysterious how ICL operates in real-world applications where tasks have a common…

无序系统与神经网络 · 物理学 2026-04-24 Kaito Takanami , Takashi Takahashi , Yoshiyuki Kabashima

In-context reinforcement learning (ICRL) leverages the in-context learning capabilities of transformer models (TMs) to efficiently generalize to unseen sequential decision-making tasks without parameter updates. However, existing ICRL…

机器学习 · 计算机科学 2026-02-10 Juncheng Dong , Bowen He , Moyang Guo , Ethan X. Fang , Zhuoran Yang , Vahid Tarokh

In-Context Learning (ICL) has been a powerful emergent property of large language models that has attracted increasing attention in recent years. In contrast to regular gradient-based learning, ICL is highly interpretable and does not…

机器学习 · 计算机科学 2024-06-07 Brian K Chen , Tianyang Hu , Hui Jin , Hwee Kuan Lee , Kenji Kawaguchi

Recent work has demonstrated that transformers and linear attention models can perform in-context learning (ICL) on simple function classes, such as linear regression. In this paper, we empirically study how these two attention mechanisms…

机器学习 · 计算机科学 2026-02-20 Ayush Goel , Arjun Kohli , Sarvagya Somvanshi

Recent research has shown that Transformers with linear attention are capable of in-context learning (ICL) by implementing a linear estimator through gradient descent steps. However, the existing results on the optimization landscape apply…

机器学习 · 计算机科学 2024-07-16 Yingcong Li , Ankit Singh Rawat , Samet Oymak

Transformer models have become foundational across a wide range of scientific and engineering domains due to their strong empirical performance. A key capability underlying their success is in-context learning (ICL): when presented with a…

机器学习 · 计算机科学 2026-04-29 Zhen Qin , Jiachen Jiang , Zhihui Zhu

Transformer-based language models excel at in-context learning (ICL), where they can adapt to new tasks based on contextual examples, without parameter updates. In a specific form of ICL, which we refer to as \textit{contextual recall},…

机器学习 · 计算机科学 2026-03-24 Bhavya Vasudeva , Puneesh Deora , Alberto Bietti , Vatsal Sharan , Christos Thrampoulidis

Large autoregressive models like Transformers can solve tasks through in-context learning (ICL) without learning new weights, suggesting avenues for efficiently solving new tasks. For many tasks, e.g., linear regression, the data…

机器学习 · 计算机科学 2025-06-17 Sarthak Mittal , Eric Elmoznino , Leo Gagnon , Sangnie Bhardwaj , Tom Marty , Dhanya Sridhar , Guillaume Lajoie

In-context learning (ICL) derives its power from enabling Large Language Models to adapt to new tasks via prompt-based reasoning alone, entirely bypassing the need for parameter updates. Existing theories primarily study ICL in single-task…

机器学习 · 计算机科学 2026-05-28 Guangyu Li , Meng Ding , Lijie Hu

While large language models (LLMs) exhibit strong reasoning abilities, their performance on complex tasks is often constrained by the limitations of their internal knowledge. A compelling approach to overcome this challenge is to augment…

人工智能 · 计算机科学 2026-03-10 Yaoqi Ye , Yiran Zhao , Keyu Duan , Zeyu Zheng , Kenji Kawaguchi , Cihang Xie , Michael Qizhe Shieh

In-context learning (ICL) of large language models has proven to be a surprisingly effective method of learning a new task from only a few demonstrative examples. In this paper, we study the efficacy of ICL from the viewpoint of statistical…

机器学习 · 统计学 2024-10-03 Juno Kim , Tai Nakamaki , Taiji Suzuki

In-context learning (ICL) has emerged as a powerful paradigm for easily adapting Large Language Models (LLMs) to various tasks. However, our understanding of how ICL works remains limited. We explore a simple model of ICL in a controlled…

机器学习 · 计算机科学 2025-09-03 Omar Naim , Guilhem Fouilhé , Nicholas Asher

The remarkable performance of Large Language Models (LLMs) can be enhanced with test-time computation, which relies on external tools and even other deep learning models. However, existing approaches for integrating non-text modality…

计算与语言 · 计算机科学 2025-12-12 Tianle Zhang , Wanlong Fang , Jonathan Woo , Paridhi Latawa , Deepak A. Subramanian , Alvin Chan

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…

机器学习 · 计算机科学 2025-10-03 Yifei Zuo , Yutong Yin , Zhichen Zeng , Ang Li , Banghua Zhu , Zhaoran Wang

In-context learning (ICL) refers to a remarkable capability of pretrained large language models, which can learn a new task given a few examples during inference. However, theoretical understanding of ICL is largely under-explored,…

机器学习 · 计算机科学 2024-09-27 Tong Yang , Yu Huang , Yingbin Liang , Yuejie Chi

Transformers pretrained on diverse tasks exhibit remarkable in-context learning (ICL) capabilities, enabling them to solve unseen tasks solely based on input contexts without adjusting model parameters. In this paper, we study ICL in one of…

机器学习 · 统计学 2024-03-18 Jingfeng Wu , Difan Zou , Zixiang Chen , Vladimir Braverman , Quanquan Gu , Peter L. Bartlett

Pretrained Transformers demonstrate remarkable in-context learning (ICL) capabilities, enabling them to adapt to new tasks from demonstrations without parameter updates. However, theoretical studies often rely on simplified architectures…

机器学习 · 统计学 2026-02-06 Samet Demir , Zafer Dogan