• Home
  • Search
  • XLA-NDP: Efficient Scheduling and Code Generation for Deep Learning Model Training on Near-Data Processing Memory
  • Cite Icon6
  • https://doi.org/10.1109/lca.2023.3261136Copy DOI Icon

XLA-NDP: Efficient Scheduling and Code Generation for Deep Learning Model Training on Near-Data Processing Memory

Show More
  • Abstract
  • Literature Map
  • References
  • Citations
  • Similar Papers
Abstract

Deep learning (DL) model training must address the memory bottleneck to continue scaling. Processing-in-memory approaches can be a viable solution as they move computations near or into the memory, reducing substantial data movement. However, to deploy applications on such hardware, end-to-end software support is crucial for efficient computation mapping and scheduling as well as extensible code generation, but no consideration has been made for DL training workloads. In this paper, we propose XLA-NDP, a compiler and runtime solution for NDPX, a near-data processing (NDP) architecture integrated with an existing DL training framework. XLA-NDP offloads NDPX kernels and schedules them to overlap with GPU kernels to maximize parallelism based on GPU and NDPX costs, while providing a template-based code generator with low-level optimizations. The experiments showed that XLA-NDP provides up to 1.41x speedup (1.24x on average) over the GPU baseline for four DL model training.

Similar Papers
  • Video Transcripts

Deep Learning-Based Computer Vision Application with Multiple Built-In Data Science-Oriented Capabilities

  • Jun 08, 2020
  • Underline Science Inc.
  • Flavius Opritoiu +2
  • Research Article
  • Citations8

Development of training environment for deep learning with medical images on supercomputer system based on asynchronous parallel Bayesian optimization

  • Jan 20, 2020
  • The Journal of Supercomputing
  • Yukihiro Nomura +11
  • Research Article
  • Citations17

Leveraging Deep Learning and Generative AI for Predicting Rheological Properties and Material Compositions of 3D Printed Polyacrylamide Hydrogels

  • Oct 15, 2024
  • Gels
  • Sakib Mohammad +4
  • Research Article
  • Citations5

Unsupervised semantic label generation in agricultural fields.

  • Feb 25, 2025
  • Frontiers in robotics and AI
  • Gianmarco Roggiolani +4
  • Conference Article

Optimizing Deep Learning Frameworks Using Bio-Inspired Algorithms

  • Nov 28, 2025
  • Rachna Rathore +5
  • Research Article
  • Citations58

Adaptive Assessment of Power System Transient Stability Based on Active Transfer Learning With Deep Belief Network

  • Apr 01, 2023
  • IEEE Transactions on Automation Science and Engineering
  • Baoqin Li +1
  • Research Article
  • Citations1

Metadata-guided feature disentanglement for functional genomics.

  • Sep 01, 2024
  • Bioinformatics (Oxford, England)
  • Alexander Rakowski +5
  • Research Article
  • Citations13

Curious Containers: A framework for computational reproducibility in life sciences with support for Deep Learning applications

  • May 18, 2020
  • Future Generation Computer Systems
  • Christoph Jansen +8
  • Research Article
  • Citations5

BO-CLAHE enhancing neonatal chest X-ray image quality for improved lesion classification

  • Feb 10, 2025
  • Scientific Reports
  • Jiwon Han +3
  • PDF
  • Research Article
  • Citations38

Improving deep learning performance by using Explainable Artificial Intelligence (XAI) approaches

  • Oct 06, 2021
  • Discover Artificial Intelligence
  • Vitor Bento +4
  • Research Article
  • Citations126

A novel deep metric learning model for imbalanced fault diagnosis and toward open-set classification

  • Mar 05, 2021
  • Knowledge-Based Systems
  • Cunjun Wang +2
  • Conference Article

Converging Deep Learning and Cloud Computing: A Scalable and Efficient Approach for Modern AI Infrastructure

  • Nov 21, 2025
  • Priyanka Gupta +3
  • Conference Article

Towards Better Coverage of Dataset with Software Product Line Engineering

  • Dec 01, 2021
  • Lei Shi +3
  • Conference Article
  • Citations1

EvoStore: Towards Scalable Storage of Evolving Learning Models

  • Jun 03, 2024
  • Robert Underwood +3
  • Preprint Article

Deep Learning Models on CPUs: A Methodology for Efficient Training

  • Jun 20, 2022
  • arXiv (Cornell University)
  • Quchen Fu +7
Cactus Communications logo

Copyright 2026 Cactus Communications. All rights reserved.