An all-transistor physical-learning transformer can be trained to perform in-context learning

arXiv cs.LG Papers

Summary

该论文提出了一种全晶体管物理学习transformer,通过在双电路中引入额外电路元件恢复双向信息流,并利用亚阈值MOSFET的指数电流定律构建注意力机制,仿真表明该电路可被训练执行简单的上下文学习任务。

arXiv:2610.12571v1 Announce Type: new Abstract: Most systems trained by equilibrium propagation or coupled learning are in the same class as multi-layer perceptrons: while they may be universal function approximators, they struggle to learn associative tasks which are easy for transformers. Equilibrium propagation relies on a bidirectional flow of information through the physical system between input and output. The reverse flow is broken by more complex nonlinear systems, such as those performing multiplication which is needed to express dot-product attention. Here we explicitly restore this bidirectional flow by introducing extra circuit elements for this purpose into a double or dyadic circuit. We then use these ideas to construct an attention mechanism exploiting the exponential current law of sub-threshold MOSFET devices. We show that in simulations this circuit can be trained to perform a simple in-context learning task, despite using realistic reverse components which transmit only an approximate gradient. For this training to succeed we need cross-entropy loss (not mean squared error). We generate nudges appropriate for this loss using a novel floating circuit element, based on the same MOSFET sub-circuit used for softmax. We train the resistors in this circuit using the Lion optimiser, which can be implemented using one capacitor per parameter to store local memory of past gradient estimates.
Original Article
View Cached Full Text

Cached at: 10/12/26, 06:41 AM

# An all-transistor physical-learning transformer can be trained to perform in-context learning
Source: [https://arxiv.org/abs/2610.12571](https://arxiv.org/abs/2610.12571)
[View PDF](https://arxiv.org/pdf/2610.12571)

> Abstract:Most systems trained by equilibrium propagation or coupled learning are in the same class as multi\-layer perceptrons: while they may be universal function approximators, they struggle to learn associative tasks which are easy for transformers\. Equilibrium propagation relies on a bidirectional flow of information through the physical system between input and output\. The reverse flow is broken by more complex nonlinear systems, such as those performing multiplication which is needed to express dot\-product attention\. Here we explicitly restore this bidirectional flow by introducing extra circuit elements for this purpose into a double or dyadic circuit\. We then use these ideas to construct an attention mechanism exploiting the exponential current law of sub\-threshold MOSFET devices\. We show that in simulations this circuit can be trained to perform a simple in\-context learning task, despite using realistic reverse components which transmit only an approximate gradient\. For this training to succeed we need cross\-entropy loss \(not mean squared error\)\. We generate nudges appropriate for this loss using a novel floating circuit element, based on the same MOSFET sub\-circuit used for softmax\. We train the resistors in this circuit using the Lion optimiser, which can be implemented using one capacitor per parameter to store local memory of past gradient estimates\.

## Submission history

From: Michael Abbott \[[view email](https://arxiv.org/show-email/67c99064/2610.12571)\] **\[v1\]**Thu, 8 Oct 2026 18:01:05 UTC \(739 KB\)

Similar Articles

Spatial Induction Heads: In-Context Learning of Multidimensional Cellular Automata

arXiv cs.LG

本文提出"空间感应头"(spatial induction heads),一种两层 gather-and-match 电路,解释 Transformer 如何在多维细胞自动机的序列化数据中进行上下文学习。研究表明,该电路可逼近贝叶斯后验并在确定性规则上实现精确预测,训练后的两层 Transformer 能泛化到未见过的规则。

@Phoenixyin13: If the full score is 10, I would honestly give this MIT paper's SMT idea and writing an 8. The paper proposes Supervised Memory Training, using Transformer as a super teacher to first distill in parallel the most important things to remember at each moment…

X AI KOLs Timeline

This paper proposes Supervised Memory Training (SMT), which uses Transformer as a super teacher to distill memory states in parallel, then trains RNN with one-step supervised learning, achieving fully parallel training and reducing gradient path from O(T) to O(1), significantly improving long-range dependency learning.

You could have invented transformers from memory

Hacker News Top

一篇技术博客,通过"关联记忆"的视角重新推导 Transformer:从线性层的外积记忆,到 MLP 的容量提升,再到自注意力的近乎无界记忆容量,并给出各层可存储记忆数量的启发式估算(如 64 维键可从约 64 条提升到上百万条记忆)。

@snowboat84: Continuing the discussion on applying physical models in AI. Today's Transformer uses attention to allow information at different positions in a sequence to interact, but this mixing is very likely lossy and irreversible. Independent pieces of information get blurred and lost as they mix, and it's impossible to precisely reverse-engineer the input from the output. What we want is a different kind of interaction...

X AI KOLs Timeline

The author discusses the problem of information loss and irreversibility caused by the Transformer's attention mechanism, and proposes drawing inspiration from the physical model of soliton propagation to design a reversible, zero-loss interaction layer as a direction for improving existing AI model architectures.