English
Related papers

Related papers: GPU Memory Requirement Prediction for Deep Learnin…

200 papers

This paper introduces a distributed, GPU-centric experience replay system, GEAR, designed to perform scalable reinforcement learning (RL) with large sequence models (such as transformers). With such models, existing systems such as Reverb…

Machine Learning · Computer Science 2023-10-10 Hanjing Wang , Man-Kit Sit , Congjie He , Ying Wen , Weinan Zhang , Jun Wang , Yaodong Yang , Luo Mai

There are many ways to represent a molecule as input to a machine learning model and each is associated with loss and retention of certain kinds of information. In the interest of preserving three-dimensional spatial information, including…

Machine Learning · Computer Science 2019-12-11 Jocelyn Sunseri , David Ryan Koes

In this study, we introduce an innovative EEG signal reconstruction sub-module designed to enhance the performance of deep learning models on EEG eye-tracking tasks. This sub-module can integrate with all Encoder-Classifier-based deep…

Human-Computer Interaction · Computer Science 2024-08-13 Weigeng Li , Neng Zhou , Xiaodong Qu

GPU kernels have come to the forefront of computing due to their utility in varied fields, from high-performance computing to machine learning. A typical GPU compute kernel is invoked millions, if not billions of times in a typical…

Machine Learning · Computer Science 2024-04-18 Khawir Mahmood , Jehandad Khan , Hammad Afzal

Gated recurrent units (GRUs) are specialized memory elements for building recurrent neural networks. Despite their incredible success on various tasks, including extracting dynamics underlying neural data, little is understood about the…

Machine Learning · Computer Science 2021-07-30 Ian D. Jordan , Piotr Aleksander Sokol , Il Memming Park

This paper presents GPU performance optimization and scaling results for inference models of the Sparse Deep Neural Network Challenge 2020. Demands for network quality have increased rapidly, pushing the size and thus the memory…

Distributed, Parallel, and Cluster Computing · Computer Science 2020-09-04 Mert Hidayetoglu , Carl Pearson , Vikram Sharma Mailthody , Eiman Ebrahimi , Jinjun Xiong , Rakesh Nagi , Wen-Mei Hwu

Proactive alert prediction in computer networks is critical for mitigating evolving cyber threats and enabling timely defensive actions. Temporal Graph Neural Networks (TGNs) provide a principled framework for modeling time-evolving…

Machine Learning · Computer Science 2026-04-28 Zahra Makki Nayeri , Mohsen Rezvani

This paper introduces a new Dynamic Gated Recurrent Neural Network (DG-RNN) for compute-efficient speech enhancement models running on resource-constrained hardware platforms. It leverages the slow evolution characteristic of RNN hidden…

Audio and Speech Processing · Electrical Eng. & Systems 2024-09-25 Longbiao Cheng , Ashutosh Pandey , Buye Xu , Tobi Delbruck , Shih-Chii Liu

Circuit representation learning has become pivotal in electronic design automation, enabling critical tasks such as testability analysis, logic reasoning, power estimation, and SAT solving. However, existing models face significant…

Machine Learning · Computer Science 2025-05-21 Ziyang Zheng , Shan Huang , Jianyuan Zhong , Zhengyuan Shi , Guohao Dai , Ningyi Xu , Qiang Xu

Recent work on deep learning for tabular data demonstrates the strong performance of deep tabular models, often bridging the gap between gradient boosted decision trees and neural networks. Accuracy aside, a major advantage of neural models…

In a variety of application domains the content to be recommended to users is associated with text. This includes research papers, movies with associated plot summaries, news articles, blog posts, etc. Recommendation approaches based on…

Machine Learning · Statistics 2016-09-12 Trapit Bansal , David Belanger , Andrew McCallum

Current AI training infrastructure is dominated by single instruction multiple data (SIMD) and systolic array architectures, such as Graphics Processing Units (GPUs) and Tensor Processing Units (TPUs), that excel at accelerating parallel…

Neural and Evolutionary Computing · Computer Science 2023-11-09 Jan Finkbeiner , Thomas Gmeinder , Mark Pupilli , Alexander Titterton , Emre Neftci

Diffusion models have recently gained state of the art performance on many image generation tasks. However, most models require significant computational resources to achieve this. This becomes apparent in the application of medical image…

Computer Vision and Pattern Recognition · Computer Science 2025-04-16 Karan Jain , Mohammad Nayeem Teli

Effective models for analysing and predicting pedestrian flow are important to ensure the safety of both pedestrians and other road users. These tools also play a key role in optimising infrastructure design and geometry and supporting the…

Machine Learning · Computer Science 2024-11-07 Yiwei Dong , Tingjin Chu , Lele Zhang , Hadi Ghaderi , Hanfang Yang

This paper presents a deep learning framework for analyzing on board vibration response signals in infrastructure health monitoring. The proposed WaveletInception-BiGRU network uses a Learnable Wavelet Packet Transform (LWPT) for early…

Machine Learning · Computer Science 2026-05-19 Reza Riahi Samani , Alfredo Nunez , Bart De Schutter

In this work, we present a compact, modular framework for constructing novel recurrent neural architectures. Our basic module is a new generic unit, the Transition Based Recurrent Unit (TBRU). In addition to hidden layer activations, TBRUs…

Computation and Language · Computer Science 2017-03-14 Lingpeng Kong , Chris Alberti , Daniel Andor , Ivan Bogatyy , David Weiss

With the evolution of power systems as it is becoming more intelligent and interactive system while increasing in flexibility with a larger penetration of renewable energy sources, demand prediction on a short-term resolution will…

Machine Learning · Computer Science 2022-12-20 Saad Emshagin , Wayes Koroni Halim , Rasha Kashef

A field that has directly benefited from the recent advances in deep learning is Automatic Speech Recognition (ASR). Despite the great achievements of the past decades, however, a natural and robust human-machine speech interaction still…

Audio and Speech Processing · Electrical Eng. & Systems 2018-03-29 Mirco Ravanelli , Philemon Brakel , Maurizio Omologo , Yoshua Bengio

The deployment of Machine Learning models in the cloud has grown among tech companies. Hardware requirements are higher when these models involve Deep Learning techniques, and the cloud providers' costs may be a barrier. We explore…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-01-13 Elayne Lemos , Rodrigo Oliveira , Jairson Rodrigues , Rosalvo F. Oliveira Neto

To satisfy the compute and memory demands of deep neural networks, neural processing units (NPUs) are widely being utilized for accelerating deep learning algorithms. Similar to how GPUs have evolved from a slave device into a mainstream…

Hardware Architecture · Computer Science 2019-11-19 Bongjoon Hyun , Youngeun Kwon , Yujeong Choi , John Kim , Minsoo Rhu