中文
相关论文

相关论文: GPU Memory Requirement Prediction for Deep Learnin…

200 篇论文

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…

机器学习 · 计算机科学 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…

机器学习 · 计算机科学 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…

人机交互 · 计算机科学 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…

机器学习 · 计算机科学 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…

机器学习 · 计算机科学 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…

分布式、并行与集群计算 · 计算机科学 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…

机器学习 · 计算机科学 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…

音频与语音处理 · 电气工程与系统科学 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…

机器学习 · 计算机科学 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…

机器学习 · 统计学 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…

神经与进化计算 · 计算机科学 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…

计算机视觉与模式识别 · 计算机科学 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…

机器学习 · 计算机科学 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…

机器学习 · 计算机科学 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…

计算与语言 · 计算机科学 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…

机器学习 · 计算机科学 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…

音频与语音处理 · 电气工程与系统科学 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…

分布式、并行与集群计算 · 计算机科学 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…

硬件体系结构 · 计算机科学 2019-11-19 Bongjoon Hyun , Youngeun Kwon , Yujeong Choi , John Kim , Minsoo Rhu