中文
相关论文

相关论文: Gradient-based Intra-attention Pruning on Pre-trai…

200 篇论文

Large Language Models (LLMs) have shown immense potential in enhancing various aspects of our daily lives, from conversational AI to search and AI assistants. However, their growing capabilities come at the cost of extremely large model…

机器学习 · 计算机科学 2025-02-27 Yingyu Liang , Jiangxuan Long , Zhenmei Shi , Zhao Song , Yufa Zhou

Quantization, knowledge distillation, and magnitude pruning are among the most popular methods for neural network compression in NLP. Independently, these methods reduce model size and can accelerate inference, but their relative benefit…

计算与语言 · 计算机科学 2022-08-23 Rajiv Movva , Jinhao Lei , Shayne Longpre , Ajay Gupta , Chris DuBois

Generative modeling has recently undergone remarkable advancements, primarily propelled by the transformative implications of Diffusion Probabilistic Models (DPMs). The impressive capability of these models, however, often entails…

机器学习 · 计算机科学 2023-10-03 Gongfan Fang , Xinyin Ma , Xinchao Wang

Large language models (LLMs) have garnered significant attention for their remarkable capabilities across various domains, whose vast parameter scales present challenges for practical deployment. Structured pruning is an effective method to…

人工智能 · 计算机科学 2024-12-25 Gui Ling , Ziyang Wang , Yuliang Yan , Qingwen Liu

This work introduces a novel approach to pruning deep learning models by using distilled data. Unlike conventional strategies which primarily focus on architectural or algorithmic optimization, our method reconsiders the role of data in…

机器学习 · 计算机科学 2023-08-10 Luke McDermott , Daniel Cummings

Diffusion Models (DMs) have exhibited superior performance in generating high-quality and diverse images. However, this exceptional performance comes at the cost of expensive architectural design, particularly due to the attention module…

计算机视觉与模式识别 · 计算机科学 2024-05-09 Hongjie Wang , Difan Liu , Yan Kang , Yijun Li , Zhe Lin , Niraj K. Jha , Yuchen Liu

Compressing convolutional neural networks (CNNs) by pruning and distillation has received ever-increasing focus in the community. In particular, designing a class-discrimination based approach would be desired as it fits seamlessly with the…

计算机视觉与模式识别 · 计算机科学 2021-10-22 Yuchen Liu , David Wentzlaff , S. Y. Kung

Network pruning and knowledge distillation are two widely-known model compression methods that efficiently reduce computation cost and model size. A common problem in both pruning and distillation is to determine compressed architecture,…

计算机视觉与模式识别 · 计算机科学 2022-02-23 Dongqi Wang , Shengyu Zhang , Zhipeng Di , Xin Lin , Weihua Zhou , Fei Wu

With the successful application of deep learning in communications systems, deep neural networks are becoming the preferred method for signal classification. Although these models yield impressive results, they often come with high…

机器学习 · 计算机科学 2024-06-13 Yao Lu , Yutao Zhu , Yuqi Li , Dongwei Xu , Yun Lin , Qi Xuan , Xiaoniu Yang

Driven by the need for more efficient and seamless integration of physical models and data, physics-informed neural networks (PINNs) have seen a surge of interest in recent years. However, ensuring the reliability of their convergence and…

The task of dataset distillation aims to find a small set of synthetic images such that training a model on them reproduces the performance of the same model trained on a much larger dataset of real samples. Existing distillation methods…

计算机视觉与模式识别 · 计算机科学 2025-11-21 George Cazenavette , Antonio Torralba , Vincent Sitzmann

Understanding and shaping the behaviour of Large Language Models (LLMs) is increasingly important as applications become more powerful and more frequently adopted. This paper introduces a machine unlearning method specifically designed for…

机器学习 · 计算机科学 2024-07-25 Nicholas Pochinkov , Nandi Schoots

Retrieval-augmented generation framework can address the limitations of large language models by enabling real-time knowledge updates for more accurate answers. An efficient way in the training phase of retrieval-augmented models is…

计算与语言 · 计算机科学 2024-02-20 Zizhong Li , Haopeng Zhang , Jiawei Zhang

With the development of transformer-based large language models (LLMs), they have been applied to many fields due to their remarkable utility, but this comes at a considerable computational cost at deployment. Fortunately, some methods such…

机器学习 · 计算机科学 2024-08-13 Harry Dong , Beidi Chen , Yuejie Chi

Pruning has recently been widely adopted to reduce the parameter scale and improve the inference efficiency of Large Language Models (LLMs). Mainstream pruning techniques often rely on uniform layerwise pruning strategies, which can lead to…

计算与语言 · 计算机科学 2025-06-04 Yuli Chen , Bo Cheng , Jiale Han , Yingying Zhang , Yingting Li , Shuhao Zhang

Transformer-based NLP models are trained using hundreds of millions or even billions of parameters, limiting their applicability in computationally constrained environments. While the number of parameters generally correlates with…

计算与语言 · 计算机科学 2022-08-16 Hassan Sajjad , Fahim Dalvi , Nadir Durrani , Preslav Nakov

Reasoning language models such as DeepSeek-R1 produce long chain-of-thought traces during inference time which make them costly to deploy at scale. We show that using compression techniques such as neural network pruning produces greater…

人工智能 · 计算机科学 2026-05-05 Ryan Lucas , Kayhan Behdin , Zhipeng Wang , Qingquan Song , Shao Tang , Rahul Mazumder

Overparametrized transformer networks are the state-of-the-art architecture for Large Language Models (LLMs). However, such models contain billions of parameters making large compute a necessity, while raising environmental concerns. To…

机器学习 · 计算机科学 2024-10-22 Yang Zhang , Yawei Li , Xinpeng Wang , Qianli Shen , Barbara Plank , Bernd Bischl , Mina Rezaei , Kenji Kawaguchi

Pruning is one of the most effective model reduction techniques. Deep networks require massive computation and such models need to be compressed to bring them on edge devices. Most existing pruning techniques are focused on vision-based…

机器学习 · 计算机科学 2020-04-30 Ramchalam Kinattinkara Ramakrishnan , Eyyüb Sari , Vahid Partovi Nia

Autoregressive Transformers adopted in Large Language Models (LLMs) are hard to scale to long sequences. Despite several works trying to reduce their computational cost, most of LLMs still adopt attention layers between all pairs of tokens…

计算与语言 · 计算机科学 2024-06-03 Sotiris Anagnostidis , Dario Pavllo , Luca Biggio , Lorenzo Noci , Aurelien Lucchi , Thomas Hofmann