Complex-Valued Phase-Coherent Transformer
cs.LG
Submitted: 2026-05-11
Updated: 2026-09-24
Comments: 26 pages, 17 tables (no figures). Companion Lean 4 formalization of Theorems 1 and 2 at https://github.com/leohio/phase-coherent-transformer-r-d
Code: https://github.com/torchcvnn/torchcvnnhttps:
License: http://creativecommons.org/licenses/by/4.0/
The gist: Complex-valued Transformers have largely inherited softmax attention from real-valued architectures.
Terminology
Abstract
Complex-valued Transformers have largely inherited softmax attention from real-valued architectures. However, row-normalised token competition is not necessarily aligned with phase-preserving computation. In this paper, we introduce the Phase-Coherent Transformer (PCT), which applies a real-valued, element-independent, smooth gate to L2-normalised complex query-key similarities. PCT replaces token competition with token-non-competing attention and is designed to preserve phase information across layers. Across mid-scale benchmarks spanning long-range memory, hierarchical long-range reasoning, positional retrieval, phase-based memory and superposition, and image classification, PCT shows strong generalisation across task categories. Under parameter-fair comparison, PCT consistently outperforms both the standard softmax Transformer and its direct complex-valued counterpart. Moreover, even on tasks traditionally considered difficult for complex-valued neural networks, such as NIAH and LRA-Text, PCT remains competitive with Multiscreen, the strongest real-valued NN baseline in our comparison. Experiments introducing gates that deliberately violate the PCT conditions show that the design is not incidental: smooth gates that preserve negatively aligned phase components remain strong, whereas gates that delete such components collapse on long-range retrieval, and gates whose outputs become excessively large suffer clear performance degradation. PCT also shows no depth-related accuracy collapse across the tested depth range. These results support introducing multi-layer phase-coherent structure into attention as a promising design principle for achieving generalisation in complex-valued Transformers.
Sources
- Attention Is All You Need
- Deep Complex Networks
- Building Blocks for a Complex-Valued Transformer Architecture
- Holographic Transformers for Complex-Valued Signal Processing: Integrating Phase Interference into Self-Attention
- Rethinking Attention with Performers
- Replacing softmax with ReLU in Vision Transformers
- Theory, Analysis, and Best Practices for Sigmoid Self-Attention
- Rethinking Attention: Polynomial Alternatives to Softmax in Transformers
- Screening Is Enough
- Long Range Arena: A Benchmark for Efficient Transformers
- Efficiently Modeling Long Sequences with Structured State Spaces
- On the Parameterization and Initialization of Diagonal State Space Models
- Diagonal State Spaces are as Effective as Structured State Spaces
- Simplified State Space Layers for Sequence Modeling
- Resurrecting Recurrent Neural Networks for Long Sequences
- Mega: Moving Average Equipped Gated Attention
- DeepNet: Scaling Transformers to 1,000 Layers
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks