English
Related papers

Related papers: VORT: Adaptive Power-Law Memory for NLP Transforme…

200 papers

We study learned memory tokens as a computational scratchpad for a single-block Universal Transformer with Adaptive Computation Time (ACT) on Sudoku-Extreme, a combinatorial reasoning benchmark. Memory tokens are empirically necessary: no…

Machine Learning · Computer Science 2026-05-05 Grigory Sapunov

Multi-scale Vision Transformer (ViT) has emerged as a powerful backbone for computer vision tasks, while the self-attention computation in Transformer scales quadratically w.r.t. the input patch number. Thus, existing solutions commonly…

Computer Vision and Pattern Recognition · Computer Science 2022-07-12 Ting Yao , Yingwei Pan , Yehao Li , Chong-Wah Ngo , Tao Mei

Sequential decision-making algorithms such as reinforcement learning (RL) in real-world scenarios inevitably face environments with partial observability. This paper scrutinizes the effectiveness of a popular architecture, namely…

Machine Learning · Computer Science 2024-05-31 Chenhao Lu , Ruizhe Shi , Yuyao Liu , Kaizhe Hu , Simon S. Du , Huazhe Xu

Many complex physical systems admit natural decomposition into an exactly solvable component and a perturbative correction. Rather than training neural networks to learn complete trajectories from scratch, we introduce Neural Network…

Computational Physics · Physics 2025-12-02 Zhenhao Chen , Mutian Shen , Boris Fain , Zohar Nussinov

Hardware neural networks that implement synaptic weights with embedded non-volatile memory, such as spin torque memory (ST-MRAM), are a major lead for low energy artificial intelligence. In this work, we propose an approximate storage…

Emerging Technologies · Computer Science 2018-10-26 Nicolas Locatelli , Adrien F. Vincent , Damien Querlioz

Vision Transformers (ViTs) excel in semantic recognition but exhibit systematic failures in spatial reasoning tasks such as mental rotation. While often attributed to data scale, this work argues that the limitation arises from the…

Computer Vision and Pattern Recognition · Computer Science 2026-05-28 Siyi Lyu , Quan Liu , Feng Yan

Deep networks are powerful function approximators, but they typically store many different computations in shared weight matrices, making it difficult to selectively reuse or adapt parts of them when a familiar structure appears in novel…

Machine Learning · Computer Science 2026-05-28 Niclas Pokel , Benjamin F. Grewe

Deep Learning Reduced Order Models (ROMs) are becoming increasingly popular as surrogate models for parametric partial differential equations (PDEs) due to their ability to handle high-dimensional data, approximate highly nonlinear…

Machine Learning · Computer Science 2026-04-09 Iva Mikuš , Boris Muha , Domagoj Vlah

Weight-Decomposed Low-Rank Adaptation (DoRA) extends LoRA by decoupling weight magnitude from direction, but its forward pass requires the row-wise norm of W + sBA, a computation that every major framework we surveyed implements by…

Machine Learning · Computer Science 2026-03-24 Alexandra Zelenin , Alexandra Zhuravlyova

Recent advancements in large language models (LLMs) based on transformer architectures have sparked significant interest in understanding their inner workings. In this paper, we introduce a novel approach to modeling transformer…

Machine Learning · Computer Science 2025-04-17 Anh Tong , Thanh Nguyen-Tang , Dongeun Lee , Duc Nguyen , Toan Tran , David Hall , Cheongwoong Kang , Jaesik Choi

Deploying Vision Transformers (ViTs) on near-sensor analog accelerators demands training pipelines that are explicitly aligned with device-level noise and energy constraints. We introduce a compact framework for silicon-photonic execution…

Emerging Technologies · Computer Science 2026-04-07 Xuming Chen , Deniz Najafi , Chengwei Zhou , Pietro Mercati , Arman Roohi , Mohsen Imani , Mahdi Nikdast , Shaahin Angizi , Gourav Datta

Parameter sharing in recursive transformers reduces model size but collapses layer-wise expressivity. We propose Mixture of LoRAs (MoL), a lightweight conditional-computation mechanism that inserts Low-Rank Adaptation (LoRA) experts inside…

Machine Learning · Computer Science 2025-12-18 Mohammadmahdi Nouriborji , Morteza Rohanian , Omid Rohanian

We introduce and analyze a **memory-weighted velocity operator** \(\mathscr{V}_{\alpha,\beta}\) as a mathematical framework for describing rates of change in systems with time-varying, power-law memory. The operator employs two independent…

Mathematical Physics · Physics 2026-01-23 Jiahao Jiang

Spin-torque transfer magnetic random access memory (STT-MRAM) is a promising emerging non-volatile memory (NVM) technology with wide applications. However, the data recovery of STT-MRAM is affected by the diversity of channel raw bit error…

Information Theory · Computer Science 2024-10-08 Xingwei Zhong , Kui Cai , Peng Kang , Guanghui Song , Bin Dai

The recently developed pure Transformer architectures have attained promising accuracy on point cloud learning benchmarks compared to convolutional neural networks. However, existing point cloud Transformers are computationally expensive…

Computer Vision and Pattern Recognition · Computer Science 2022-05-26 Cheng Zhang , Haocheng Wan , Xinyi Shen , Zizhao Wu

Test-time compute has emerged as a key paradigm for enhancing LLM reasoning, yet prevailing approaches like Best-of-N and majority voting apply uniform depth across inputs, wasting computation on simple queries while potentially…

Artificial Intelligence · Computer Science 2025-09-30 Yixin He , Lumingyuan Tang

Unlike recurrent models, conventional wisdom has it that Transformers cannot perfectly model regular languages. Inspired by the notion of working memory, we propose a new Transformer variant named RegularGPT. With its novel combination of…

Computation and Language · Computer Science 2023-05-09 Ta-Chung Chi , Ting-Han Fan , Alexander I. Rudnicky , Peter J. Ramadge

Hamiltonian Operator Inference has been introduced in [Sharma, H., Wang, Z., Kramer, B., Physica D: Nonlinear Phenomena, 431, p.133122, 2022] to learn structure-preserving reduced-order models (ROMs) for Hamiltonian systems. This approach…

Numerical Analysis · Mathematics 2024-05-10 Yuwei Geng , Jasdeep Singh , Lili Ju , Boris Kramer , Zhu Wang

We apply the influence-adaptive Walsh geometry of a companion theory paper (arXiv:2605.01637) to extreme low-bit weight-only LLM quantization. The recipe is one math-invariant transformation: WHT-rotate each linear layer's weight matrix and…

Machine Learning · Computer Science 2026-05-26 Gorgi Pavlov

The self-attention mechanism is the key to the success of transformers in recent Large Language Models (LLMs). However, the quadratic computational cost $O(n^2)$ in the input sequence length $n$ is a notorious obstacle for further…

Machine Learning · Computer Science 2024-10-17 Yingyu Liang , Heshan Liu , Zhenmei Shi , Zhao Song , Zhuoyan Xu , Junze Yin