English
Related papers

Related papers: Knowledge Offloading: Decomposing LLMs into Sparse…

200 papers

How many of a neural network's parameters actually encode task-specific information? We investigate this question with LottaLoRA, a training paradigm in which every backbone weight is drawn at random and frozen; only low-rank LoRA adapters…

Machine Learning · Computer Science 2026-04-14 Hananel Hazan , Yanbo Zhang , Benedikt Hartl , Michael Levin

Knowledge distillation typically involves transferring knowledge from a Large Language Model (LLM) to a Smaller Language Model (SLM). However, in tasks such as text matching, fine-tuned smaller models often yield more effective…

Computation and Language · Computer Science 2025-07-09 Mingzhe Li , Jing Xiang , Qishen Zhang , Kaiyang Wan , Xiuying Chen

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…

Machine Learning · Computer Science 2025-08-25 Yixuan Wang , Haoyu Qiao , Lujun Li , Qingfu Zhu , Wanxiang Che

Transformer-based large language models (LLMs) demonstrate impressive performance in long context generation. Extending the context length has disproportionately shifted the memory footprint of LLMs during inference to the key-value cache…

Machine Learning · Computer Science 2025-02-19 Cheng Luo , Zefan Cai , Hanshi Sun , Jinqi Xiao , Bo Yuan , Wen Xiao , Junjie Hu , Jiawei Zhao , Beidi Chen , Anima Anandkumar

Large language models (LLMs) encode knowledge in parametric weights, making it costly to update or extend without retraining. Retrieval-augmented generation (RAG) mitigates this limitation by appending retrieved text to the input, but…

Computation and Language · Computer Science 2026-04-24 Bin Ju , Shenfeng Weng , Danying Zhou , Rongkai Xu , Kunkai Su

Decoding throughput improvements from larger inference batches are limited by GPU memory, which is largely consumed by the key-value (KV) cache. Prior training-free KV cache offloading alleviates this by keeping redundant context on the CPU…

Computation and Language · Computer Science 2026-01-30 Yuxiang Huang , Pengjie Wang , Jicheng Han , Weilin Zhao , Zhou Su , Ao Sun , Hongya Lyu , Hengyu Zhao , Yudong Wang , Chaojun Xiao , Xu Han , Zhiyuan Liu

Decentralized federated learning (DFL) based on low-rank adaptation (LoRA) enables mobile devices with multi-task datasets to collaboratively fine-tune a large language model (LLM) by exchanging locally updated parameters with a subset of…

Machine Learning · Computer Science 2026-02-25 Nuocheng Yang , Sihua Wang , Ouwen Huan , Mingzhe Chen , Tony Q. S. Quek , Changchuan Yin

The advent of pre-trained large language models (LLMs) has revolutionized various natural language processing tasks. These models predominantly employ an auto-regressive decoding mechanism that utilizes Key-Value (KV) caches to eliminate…

Computation and Language · Computer Science 2024-06-12 Hao Yu , Zelan Yang , Shen Li , Yong Li , Jianxin Wu

Large Language Models (LLMs) have demonstrated strong reasoning and memorization capabilities via pretraining on massive textual corpora. However, this poses risk of privacy and copyright violations, highlighting the need for efficient…

Machine Learning · Computer Science 2025-04-28 Sungmin Cha , Sungjun Cho , Dasol Hwang , Moontae Lee

Life-long learning aims at learning a sequence of tasks without forgetting the previously acquired knowledge. However, the involved training data may not be life-long legitimate due to privacy or copyright reasons. In practical scenarios,…

Computer Vision and Pattern Recognition · Computer Science 2022-07-19 Jingwen Ye , Yifang Fu , Jie Song , Xingyi Yang , Songhua Liu , Xin Jin , Mingli Song , Xinchao Wang

A fundamental advantage of neural models for NLP is their ability to learn representations from scratch. However, in practice this often means ignoring existing external linguistic resources, e.g., WordNet or domain specific ontologies such…

Computation and Language · Computer Science 2017-04-26 Ye Zhang , Matthew Lease , Byron C. Wallace

Data allocation plays a critical role in federated large language model (LLM) and small language models (SLMs) reasoning collaboration. Nevertheless, existing data allocation methods fail to address an under-explored challenge in…

Artificial Intelligence · Computer Science 2026-02-24 Wei Guo , Siyuan Lu , Xiangdong Ran , Yiqi Tong , Yikun Ban , Zelong Xu , Jing Fan , Zixuan Huang , Xiao Zhang , Zhaojun Hu , Fuzhen Zhuang

Large Language Models (LLMs) inevitably memorize sensitive information during training, posing significant privacy risks. Machine unlearning has emerged as a promising solution to selectively remove such information without full retraining.…

Machine Learning · Computer Science 2026-04-02 Yuze Wang , Yujia Tong , Xuan Liu , Junhao Dong

Recurrent LLM architectures have emerged as a promising approach for improving reasoning, as they enable multi-step computation in the embedding space without generating intermediate tokens. Models such as Ouro perform reasoning by…

The training of large models, involving fine-tuning, faces the scarcity of high-quality data. Compared to the solutions based on centralized data centers, updating large models in the Internet of Things (IoT) faces challenges in…

Machine Learning · Computer Science 2024-07-09 Shaoyuan Chen , Linlin You , Rui Liu , Shuo Yu , Ahmed M. Abdelmoniem

Transformers lack an explicit architectural mechanism for storing and organizing knowledge acquired during training. We introduce learnable sparse memory banks: a set of latent tokens, randomly initialized and trained end-to-end, that…

Machine Learning · Computer Science 2026-03-24 Tasmay Pankaj Tibrewal , Pritish Saha , Ankit Meda , Kunal Singh , Pradeep Moturi

Large language models (LLMs) exhibit a wide range of capabilities, including mathematical reasoning, code generation, and linguistic behaviors. We show that many capabilities are highly localized to small subsets of attention heads within…

Computation and Language · Computer Science 2026-03-05 Anna Bair , Yixuan Even Xu , Mingjie Sun , J. Zico Kolter

The conventional model aggregation-based federated learning (FL) approach requires all local models to have the same architecture, which fails to support practical scenarios with heterogeneous local models. Moreover, frequent model exchange…

Signal Processing · Electrical Eng. & Systems 2023-03-23 Zhixiong Chen , Wenqiang Yi , Yuanwei Liu , Arumugam Nallanathan

Scientific reasoning rarely stops at what is directly observable; it often requires uncovering hidden structure from data. From estimating reaction constants in chemistry to inferring demand elasticities in economics, this latent structure…

Artificial Intelligence · Computer Science 2026-05-01 Chaemin Jang , Woojin Park , Hyeok Yun , Dongman Lee , Jihee Kim

Large Language Models (LLMs) have demonstrated proficiency in their reasoning abilities, yet their large size presents scalability challenges and limits any further customization. In contrast, compact models offer customized training but…

Computation and Language · Computer Science 2024-02-06 Denis Tarasov , Kumar Shridhar