English
Related papers

Related papers: {\ae}net-PyTorch: a GPU-supported implementation f…

200 papers

Bayesian methods promise to fix many shortcomings of deep learning, but they are impractical and rarely match the performance of standard methods, let alone improve them. In this paper, we demonstrate practical training of deep networks…

Deep learning researchers and practitioners usually leverage GPUs to help train their deep neural networks (DNNs) faster. However, choosing which GPU to use is challenging both because (i) there are many options, and (ii) users grapple with…

Machine Learning · Computer Science 2021-06-09 Geoffrey X. Yu , Yubo Gao , Pavel Golikov , Gennady Pekhimenko

The demonstrated success of transfer learning has popularized approaches that involve pretraining models from massive data sources and subsequent finetuning towards a specific task. While such approaches have become the norm in fields such…

Neural network architectures are most often conceptually designed and described in visual terms, but are implemented by writing error-prone code. PrototypeML is a machine learning development environment that bridges the dichotomy between…

Machine Learning · Computer Science 2020-07-03 Daniel Reiss Harris

Training large neural network (NN) models requires extensive memory resources, and Activation Compressed Training (ACT) is a promising approach to reduce training memory footprint. This paper presents GACT, an ACT framework to support a…

As deep neural networks become more complex and input datasets grow larger, it can take days or even weeks to train a deep neural network to the desired accuracy. Therefore, distributed Deep Learning at a massive scale is a critical…

Distributed, Parallel, and Cluster Computing · Computer Science 2017-08-08 Minsik Cho , Ulrich Finkler , Sameer Kumar , David Kung , Vaibhav Saxena , Dheeraj Sreedhar

The objective of Information Extraction (IE) is to derive structured representations from unstructured or semi-structured documents. However, developing IE models is complex due to the need of integrating several subtasks. Additionally,…

Information Retrieval · Computer Science 2024-06-04 Arne Binder , Leonhard Hennig , Christoph Alt

In this work, we introduce BurTorch, a compact high-performance framework designed to optimize Deep Learning (DL) training on single-node workstations through an exceptionally efficient CPU-based backpropagation (Rumelhart et al., 1986;…

Machine Learning · Computer Science 2025-03-19 Konstantin Burlachenko , Peter Richtárik

Networks are ubiquitous in many real-world applications (e.g., social networks encoding trust/distrust relationships, correlation networks arising from time series data). While many networks are signed or directed, or both, there is a lack…

Machine Learning · Computer Science 2023-12-04 Yixuan He , Xitong Zhang , Junjie Huang , Benedek Rozemberczki , Mihai Cucuringu , Gesine Reinert

Most atomistic machine learning (ML) models rely on a locality ansatz, and decompose the energy into a sum of short-ranged, atom-centered contributions. This leads to clear limitations when trying to describe problems that are dominated by…

PyG (PyTorch Geometric) has evolved significantly since its initial release, establishing itself as a leading framework for Graph Neural Networks. In this paper, we present Pyg 2.0 (and its subsequent minor versions), a comprehensive update…

The ever-growing demand for further advances in artificial intelligence motivated research on unconventional computation based on analog physical devices. While such computation devices mimic brain-inspired analog information processing,…

Neural and Evolutionary Computing · Computer Science 2022-05-02 Mitsumasa Nakajima , Katsuma Inoue , Kenji Tanaka , Yasuo Kuniyoshi , Toshikazu Hashimoto , Kohei Nakajima

We develop a PyTorch-based Graph Network Simulator (GNS) that learns physics and predicts the flow behavior of particulate and fluid systems. GNS discretizes the domain with nodes representing a collection of material points and the links…

Machine Learning · Computer Science 2022-11-21 Krishna Kumar , Joseph Vantassel

This paper introduces an active learning approach to the fitting of machine learning interatomic potentials. Our approach is based on the D-optimality criterion for selecting atomic configurations on which the potential is fitted. It is…

Computational Physics · Physics 2017-09-19 Evgeny V. Podryabinkin , Alexander V. Shapeev

Spatiotemporal graph neural networks (ST-GNNs) are powerful tools for modeling spatial and temporal data dependencies. However, their applications have been limited primarily to small-scale datasets because of memory constraints. While…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-09-17 Seth Ockerman , Amal Gueroudji , Tanwi Mallick , Yixuan He , Line Pouchard , Robert Ross , Shivaram Venkataraman

Crafting neural-network interatomic potentials (NNIPs) remains a complex task, demanding specialized expertise in both machine learning and electronic-structure calculations. Here, we introduce AiiDA-TrainsPot, an automated, open-source,…

Computational Physics · Physics 2026-05-08 Davide Bidoggia , Nataliia Manko , Maria Peressi , Antimo Marrazzo

This work introduces CTorch, a PyTorch-compatible, GPU-accelerated, and auto-differentiable projector toolbox designed to handle various CT geometries with configurable projector algorithms. CTorch provides flexible scanner geometry…

Medical Physics · Physics 2025-12-15 Xiao Jiang , Grace J. Gang , J. Webster Stayman

Traditional rule-based cybersecurity systems have proven highly effective against known malware threats. However, they face challenges in detecting novel threats. To address this issue, emerging cybersecurity systems are incorporating AI…

Cryptography and Security · Computer Science 2024-12-18 Tobias Becher , Simon Torka

A neural network is used to train, predict, and evaluate a model to calculate the energies of 3-dimensional systems composed of Ti and O atoms. Python classes are implemented to quantify atomic interactions through symmetry functions and to…

Computational Physics · Physics 2024-04-30 James Paolo Rili

The ability to understand and engineer molecular structures relies on having accurate descriptions of the energy as a function of atomic coordinates. Here we outline a new paradigm for deriving energy functions of hyperdimensional molecular…