中文
相关论文

相关论文: Reparametrizing Shampoo and SOAP for Subspace Basi…

200 篇论文

The growing scale of Large Language Models (LLMs) has necessitated the development of parameter-efficient fine-tuning techniques. Low-Rank Adaptation (LoRA) has emerged as a promising approach, reducing the number of trainable parameters by…

机器学习 · 计算机科学 2025-09-01 Jessica Liang , Anirudh Bharadwaj

Learning multiple tasks sequentially requires neural networks to balance retaining knowledge, yet being flexible enough to adapt to new tasks. Regularizing network parameters is a common approach, but it rarely incorporates prior knowledge…

机器学习 · 计算机科学 2025-12-22 Joanna Sliwa , Frank Schneider , Nathanael Bosch , Agustinus Kristiadi , Philipp Hennig

While FP8 attention has shown substantial promise in innovations like FlashAttention-3, its integration into the decoding phase of the DeepSeek Multi-head Latent Attention (MLA) architecture presents notable challenges. These challenges…

机器学习 · 计算机科学 2026-04-29 Yifan Zhang , Zunhai Su , Shuhao Hu , Rui Yang , Wei Wu , Yulei Qian , Yuchen Xie , Xunliang Cai

Pipeline parallelism (PP) is widely used for training large language models (LLMs), yet its scalability is often constrained by high activation memory consumption as the number of in-flight microbatches grows with the degree of PP. In this…

机器学习 · 计算机科学 2025-07-01 Xinyi Wan , Penghui Qi , Guangxing Huang , Min Lin , Jialin Li

In this paper, we further investigate and refine the subspace-constrained preconditioning technique to enhance the theoretical and numerical convergence properties of randomized iterative methods for solving linear systems. In particular,…

数值分析 · 数学 2026-05-29 Yonghan Sun , Hou-Duo Qi , Deren Han , Jiaxin Xie

Researchers have presented systems for efficiently analysing video data at scale using sampling algorithms. While these systems effectively leverage the temporal redundancy present in videos, they suffer from three limitations. First, they…

数据库 · 计算机科学 2021-04-06 Jaeho Bang , Pramod Chunduri , Joy Arulraj

Self-supervised learning (SSL) is a powerful paradigm for learning from unlabeled time-series data. However, popular methods such as masked autoencoders (MAEs) rely on reconstructing inputs from a fixed, predetermined masking ratio. Instead…

机器学习 · 计算机科学 2026-03-03 Duy Nguyen , Jiachen Yao , Jiayun Wang , Julius Berner , Animashree Anandkumar

The Key-Value (KV) cache is a crucial component in serving transformer-based autoregressive large language models (LLMs), enabling faster inference by storing previously computed KV vectors. However, its memory consumption scales linearly…

机器学习 · 计算机科学 2024-10-07 Rongzhi Zhang , Kuang Wang , Liyuan Liu , Shuohang Wang , Hao Cheng , Chao Zhang , Yelong Shen

Object: Modern computational MRI denoising approaches are often designed assuming fixed k-space coverage. This contrasts with earlier acquisition-design literature that leveraged k-space coverage modifications (e.g., reducing spatial…

信号处理 · 电气工程与系统科学 2025-11-12 Jiayang Wang , Justin P. Haldar

Data compression is a powerful tool for managing massive but repetitive datasets, especially schemes such as grammar-based compression that support computation over the data without decompressing it. In the best case such a scheme takes a…

数据结构与算法 · 计算机科学 2019-06-04 Travis Gagie , Tomohiro I , Giovanni Manzini , Gonzalo Navarro , Hiroshi Sakamoto , Yoshimasa Takabatake

Efficient inference of large language models (LLMs) is hindered by an ever-growing key-value (KV) cache, making KV cache compression a critical research direction. Traditional methods selectively evict less important KV cache entries, which…

机器学习 · 计算机科学 2025-12-01 Yuxuan Tian , Zihan Wang , Yebo Peng , Aomufei Yuan , Zhiming Wang , Bairen Yi , Xin Liu , Yong Cui , Tong Yang

Large language models encounter critical GPU memory capacity constraints during long-context inference, where KV cache memory consumption severely limits decode batch sizes. While existing research has explored offloading KV cache to DRAM,…

机器学习 · 计算机科学 2026-03-31 Qiuyang Zhang , Kai Zhou , Ding Tang , Kai Lu , Cheng Li , Zhenyu Yang , Peng Xu , Jiguang Wan

Quantum error correction, which utilizes logical qubits that are encoded as redundant multiple physical qubits to find and correct errors in physical qubits, is indispensable for practical quantum computing. Surface code is considered to be…

机器学习 · 计算机科学 2025-09-15 Hoshitaro Ohnishi , Hideo Mukai

Catastrophic forgetting has been a major challenge in continual learning, where the model needs to learn new tasks with limited or no access to data from previously seen tasks. To tackle this challenge, methods based on knowledge…

机器学习 · 计算机科学 2023-06-29 Qiao Gu , Dongsub Shim , Florian Shkurti

Large Language Models (LLMs) confront significant memory challenges due to the escalating KV cache with increasing sequence length. As a crucial technique, existing cross-layer KV cache sharing methods either necessitate modified model…

机器学习 · 计算机科学 2025-08-25 Yixuan Wang , Haoyu Qiao , Lujun Li , Qingfu Zhu , Wanxiang Che

As large language models (LLMs) continue to scale, the memory footprint of key-value (KV) caches during inference has become a significant bottleneck. Existing approaches primarily focus on compressing KV caches within a single prompt or…

计算与语言 · 计算机科学 2025-12-18 Xinye Zhao , Spyridon Mastorakis

Efficient LLM inference research has largely focused on reducing the cost of each decoding step (e.g., using quantization, pruning, or sparse attention), typically applying a uniform computation budget to every generated token. In practice,…

机器学习 · 计算机科学 2026-05-12 Yash Akhauri , Mohamed S. Abdelfattah

Reinforcement learning (RL) has become central to enhancing reasoning in large language models (LLMs). Yet on-policy algorithms such as Group Relative Policy Optimization (GRPO) often suffer in early training: noisy gradients from…

机器学习 · 计算机科学 2026-03-19 Ziyan Wang , Zheng Wang , Xingwei Qu , Qi Cheng , Jie Fu , Shengpu Tang , Minjia Zhang , Xiaoming Huo

Self-supervised learning (SSL) has become a core technique in speech processing, but the high dimensionality of its representations makes discretization essential for improving efficiency. However, existing discretization methods still…

音频与语音处理 · 电气工程与系统科学 2025-11-11 Xueqing Li , Hao Ma , Zehan Li , Rujin Chen , Boyu Zhu , Ruihao Jing , Jian Kang , Jie Li , Chi Zhang , Xiao-Lei Zhang , Xuelong Li

Non-uniform message quantization techniques such as reconstruction-computation-quantization (RCQ) improve error-correction performance and decrease hardware complexity of low-density parity-check (LDPC) decoders that use a flooding…

信号处理 · 电气工程与系统科学 2021-04-20 Caleb Terrill , Linfang Wang , Sean Chen , Chester Hulse , Calvin Kuo , Richard Wesel , Dariush Divsalar