中文
相关论文

相关论文: Lowering PyTorch's Memory Consumption for Selectiv…

200 篇论文

With the explosive growth of deep learning applications and increasing privacy concerns, the right to be forgotten has become a critical requirement in various AI industries. For example, given a facial recognition system, some individuals…

计算机视觉与模式识别 · 计算机科学 2024-12-20 Dasol Choi , Dongbin Na

Memory footprint is one of the main limiting factors for large neural network training. In backpropagation, one needs to store the input to each operation in the computational graph. Every modern neural network model has quite a few…

机器学习 · 计算机科学 2022-02-04 Georgii Novikov , Daniel Bershatsky , Julia Gusak , Alex Shonenkov , Denis Dimitrov , Ivan Oseledets

Neural network training is inherently sequential where the layers finish the forward propagation in succession, followed by the calculation and back-propagation of gradients (based on a loss function) starting from the last layer. The…

机器学习 · 计算机科学 2023-12-01 Vahid Janfaza , Shantanu Mandal , Farabi Mahmud , Abdullah Muzahid

Inspired by the recent success of deep neural networks and the recent efforts to develop multi-layer dictionary models, we propose a Deep Analysis dictionary Model (DeepAM) which is optimized to address a specific regression task known as…

机器学习 · 统计学 2021-02-03 Jun-Jie Huang , Pier Luigi Dragotti

How is knowledge stored in an LLM's weights? We study this via layer pruning: if removing a certain layer does not affect model performance in common question-answering benchmarks, then the weights in that layer are not necessary for…

计算与语言 · 计算机科学 2025-03-04 Andrey Gromov , Kushal Tirumala , Hassan Shapourian , Paolo Glorioso , Daniel A. Roberts

Memristive devices have shown great promise to facilitate the acceleration and improve the power efficiency of Deep Learning (DL) systems. Crossbar architectures constructed using these Resistive Random-Access Memory (RRAM) devices can be…

新兴技术 · 计算机科学 2025-01-30 Corey Lammie , Wei Xiang , Bernabé Linares-Barranco , Mostafa Rahimi Azghadi

Deep metric learning algorithms have a wide variety of applications, but implementing these algorithms can be tedious and time consuming. PyTorch Metric Learning is an open source library that aims to remove this barrier for both…

计算机视觉与模式识别 · 计算机科学 2020-08-24 Kevin Musgrave , Serge Belongie , Ser-Nam Lim

Standard deep learning relies on Backpropagation (BP), which is constrained by biologically implausible weight symmetry and suffers from significant gradient interference within dense representations. To mitigate these bottlenecks, we…

机器学习 · 计算机科学 2026-01-30 Fanping Liu , Hua Yang , Jiasi Zou

Edge computing for neural networks is getting important especially for low power applications and offline devices. TensorFlow Lite and PyTorch Mobile were released for this purpose. But they mainly support mobile devices instead of…

硬件体系结构 · 计算机科学 2020-07-06 Hasan Unlu

Reservoir computers (RCs) are among the fastest to train of all neural networks, especially when they are compared to other recurrent neural networks. RC has this advantage while still handling sequential data exceptionally well. However,…

机器学习 · 计算机科学 2022-07-14 Hayden Joy , Marios Mattheakis , Pavlos Protopapas

Deep convolutional neural networks (CNNs) are indispensable to state-of-the-art computer vision algorithms. However, they are still rarely deployed on battery-powered mobile devices, such as smartphones and wearable gadgets, where vision…

计算机视觉与模式识别 · 计算机科学 2017-04-20 Tien-Ju Yang , Yu-Hsin Chen , Vivienne Sze

Storing tabular data to balance storage and query efficiency is a long-standing research question in the database community. In this work, we argue and show that a novel DeepMapping abstraction, which relies on the impressive memorization…

数据库 · 计算机科学 2024-09-27 Lixi Zhou , K. Selçuk Candan , Jia Zou

We introduce a general formulation for automatic differentiation through direct form filters, yielding a closed-form backpropagation that includes initial condition gradients. The result is a single expression that can represent both the…

系统与控制 · 电气工程与系统科学 2025-11-25 Chin-Yun Yu , György Fazekas

Training convolutional neural networks at scale demands substantial memory, largely due to storing intermediate activations for backpropagation. Existing approaches -- such as checkpointing, invertible architectures, or gradient…

机器学习 · 计算机科学 2026-03-11 Anirudh Thatipelli , Jeffrey Sam , Mathias Louboutin , Ali Siahkoohi , Rongrong Wang , Felix J. Herrmann

The backpropagation of error algorithm (backprop) has been instrumental in the recent success of deep learning. However, a key question remains as to whether backprop can be formulated in a manner suitable for implementation in neural…

神经与进化计算 · 计算机科学 2020-10-13 Beren Millidge , Alexander Tschantz , Anil K Seth , Christopher L Buckley

The excellent performance of deep neural networks has enabled us to solve several automatization problems, opening an era of autonomous devices. However, current deep net architectures are heavy with millions of parameters and require…

计算机视觉与模式识别 · 计算机科学 2018-07-06 Dat Thanh Tran , Alexandros Iosifidis , Moncef Gabbouj

Learning deep representations to solve complex machine learning tasks has become the prominent trend in the past few years. Indeed, Deep Neural Networks are now the golden standard in domains as various as computer vision, natural language…

机器学习 · 计算机科学 2020-12-04 Vincent Gripon , Carlos Lassance , Ghouthi Boukli Hacene

Although deep learning has demonstrated remarkable capability in learning from unstructured data, modern tree-based ensemble models remain superior in extracting relevant information and learning from structured datasets. While several…

机器学习 · 计算机科学 2026-02-05 Yi-Chun Liao , Chieh-Lin Tsai , Yuan-Hao Chang , Camélia Slimani , Jalil Boukhobza , Tei-Wei Kuo

Dynamic Random Access Memory (DRAM) is the de-facto choice for main memory devices due to its cost-effectiveness. It offers a larger capacity and higher bandwidth compared to SRAM but is slower than the latter. With each passing generation,…

硬件体系结构 · 计算机科学 2022-01-19 Kaustav Goswami , Hemanta Kumar Mondal , Shirshendu Das , Dip Sankar Banerjee

The vast majority of natural sensory data is temporally redundant. Video frames or audio samples which are sampled at nearby points in time tend to have similar values. Typically, deep learning algorithms take no advantage of this…

神经与进化计算 · 计算机科学 2017-06-14 Peter O'Connor , Efstratios Gavves , Max Welling