English
Related papers

Related papers: Deep Kernel Fusion for Transformers

200 papers

In recent years, transformer-based deep learning networks have gained popularity in Hyperspectral (HS) unmixing applications due to their superior performance. The attention mechanism within transformers facilitates input-dependent…

Deep convolutional neural networks (ConvNets) of 3-dimensional kernels allow joint modeling of spatiotemporal features. These networks have improved performance of video and volumetric image analysis, but have been limited in size due to…

Computer Vision and Pattern Recognition · Computer Science 2017-06-13 David Budden , Alexander Matveev , Shibani Santurkar , Shraman Ray Chaudhuri , Nir Shavit

In this paper, we introduce LightVLM, a simple but effective method that can be seamlessly deployed upon existing Vision-Language Models (VLMs) to greatly accelerate the inference process in a training-free manner. We divide the inference…

Computer Vision and Pattern Recognition · Computer Science 2025-09-03 Lianyu Hu , Fanhua Shang , Wei Feng , Liang Wan

Inference accounts for the majority of latency and energy consumption in large language model (LLM) deployments, often exceeding 90% of total cost. While training-time efficiency has seen extensive progress, runtime optimization remains a…

Autoregressive decoding with generative Large Language Models (LLMs) on accelerators (GPUs/TPUs) is often memory-bound where most of the time is spent on transferring model parameters from high bandwidth memory (HBM) to cache. On the other…

Machine Learning · Computer Science 2024-02-15 Yashas Samaga B L , Varun Yerram , Chong You , Srinadh Bhojanapalli , Sanjiv Kumar , Prateek Jain , Praneeth Netrapalli

Spectral-domain CNNs have been shown to be more efficient than traditional spatial CNNs in terms of reducing computation complexity. However they come with a `kernel explosion' problem that, even after compression (pruning), imposes a high…

Hardware Architecture · Computer Science 2023-10-18 Yue Niu , Rajgopal Kannan , Ajitesh Srivastava , Viktor Prasanna

Accelerating deep model training and inference is crucial in practice. Existing deep learning frameworks usually concentrate on optimizing training speed and pay fewer attentions to inference-specific optimizations. Actually, model…

Distributed, Parallel, and Cluster Computing · Computer Science 2020-08-12 Yongchao Liu , Yue Jin , Yong Chen , Teng Teng , Hang Ou , Rui Zhao , Yao Zhang

The past few years has witnessed specialized large language model (LLM) inference systems, such as vLLM, SGLang, Mooncake, and DeepFlow, alongside rapid LLM adoption via services like ChatGPT. Driving these system design efforts is the…

Databases · Computer Science 2025-06-30 James Pan , Guoliang Li

While long-context inference is crucial for advancing large language model (LLM) applications, its prefill speed remains a significant bottleneck. Current approaches, including sequence parallelism strategies and compute reduction through…

Machine Learning · Computer Science 2025-05-27 Yuxiang Huang , Mingye Li , Xu Han , Chaojun Xiao , Weilin Zhao , Sun Ao , Hao Zhou , Jie Zhou , Zhiyuan Liu , Maosong Sun

Faster inference of deep learning models is highly demanded on edge devices and even servers, for both financial and environmental reasons. To address this issue, we propose SoftNeuro, a novel, high-performance inference framework with…

Machine Learning · Computer Science 2021-10-13 Masaki Hilaga , Yasuhiro Kuroda , Hitoshi Matsuo , Tatsuya Kawaguchi , Gabriel Ogawa , Hiroshi Miyake , Yusuke Kozawa

Gated Linear Units (GLUs) have become essential components in the feed-forward networks of state-of-the-art Large Language Models (LLMs). However, they require twice as many memory reads compared to feed-forward layers without gating, due…

Machine Learning · Computer Science 2025-07-01 Yukito Tajima , Nakamasa Inoue , Yusuke Sekikawa , Ikuro Sato , Rio Yokota

The deep learning (DL) has been penetrating daily life in many domains, how to keep the DL model inference secure and sample privacy in an encrypted environment has become an urgent and increasingly important issue for various…

Cryptography and Security · Computer Science 2025-12-01 Wenbo Song , Xinxin Fan , Quanliang Jing , Shaoye Luo , Wenqi Wei , Chi Lin , Yunfeng Lu , Ling Liu

As the Large Language Model (LLM) becomes increasingly important in various domains. However, the following challenges still remain unsolved in accelerating LLM inference: (1) Synchronized partial softmax update. The softmax operation…

Machine Learning · Computer Science 2024-01-08 Ke Hong , Guohao Dai , Jiaming Xu , Qiuli Mao , Xiuhong Li , Jun Liu , Kangdi Chen , Yuhan Dong , Yu Wang

Large language models (LLMs) have grown beyond the memory capacity of single GPU devices, necessitating quantization techniques for practical deployment. While NF4 (4-bit NormalFloat) quantization enables 4$\times$ memory reduction,…

Machine Learning · Computer Science 2026-04-06 Xiangbo Qi , Chaoyi Jiang , Murali Annavaram

Unified multimodal models have recently shown remarkable gains in both capability and versatility, yet most leading systems are still trained from scratch and require substantial computational resources. In this paper, we show that…

Computer Vision and Pattern Recognition · Computer Science 2025-11-21 Zeyu Wang , Zilong Chen , Chenhui Gou , Feng Li , Chaorui Deng , Deyao Zhu , Kunchang Li , Weihao Yu , Haoqin Tu , Haoqi Fan , Cihang Xie

In order to achieve better performance for point cloud analysis, many researchers apply deeper neural networks using stacked Multi-Layer-Perceptron (MLP) convolutions over irregular point cloud. However, applying dense MLP convolutions over…

Computer Vision and Pattern Recognition · Computer Science 2019-09-24 Can Chen , Luca Zanotti Fragonara , Antonios Tsourdos

To address increasing compute demand from recent multi-model workloads with heavy models like large language models, we propose to deploy heterogeneous chiplet-based multi-chip module (MCM)-based accelerators. We develop an advanced…

Hardware Architecture · Computer Science 2023-12-18 Mohanad Odema , Hyoukjun Kwon , Mohammad Abdullah Al Faruque

Channel estimation is crucial in 5G communication networks for optimizing transmission parameters and ensuring reliable, high-speed communication. However, the use of multiple-input and multiple-output (MIMO) and millimeter-wave (mmWave) in…

Information Theory · Computer Science 2026-05-05 Shengzhe Lyu , Yuhan She , Di Duan , Tao Ni , Yu Hin Chan , Chengwen Luo , Ray C. C. Cheung , Weitao Xu

We present automatic horizontal fusion, a novel optimization technique that complements the standard kernel fusion techniques for GPU programs. Unlike the standard fusion, whose goal is to eliminate intermediate data round trips, our…

Distributed, Parallel, and Cluster Computing · Computer Science 2020-07-03 Ao Li , Bojian Zheng , Gennady Pekhimenko , Fan Long

Although Large Language Models (LLMs) have demonstrated remarkable capabilities, their massive parameter counts and associated extensive computing make LLMs' deployment the main part of carbon emission from nowadays AI applications.…

Machine Learning · Computer Science 2024-10-24 Jie Peng , Zhang Cao , Huaizhi Qu , Zhengyu Zhang , Chang Guo , Yanyong Zhang , Zhichao Cao , Tianlong Chen