Profiling and optimization of multi-card GPU machine learning jobs

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Lawenda, Marcin, Khloponin, Kyrylo, Samborski, Krzesimir, Szustak, Łukasz
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908383411437568
author Lawenda, Marcin
Khloponin, Kyrylo
Samborski, Krzesimir
Szustak, Łukasz
author_facet Lawenda, Marcin
Khloponin, Kyrylo
Samborski, Krzesimir
Szustak, Łukasz
contents The effectiveness and efficiency of machine learning methodologies are crucial, especially with respect to the quality of results and computational cost. This paper discusses different model optimization techniques, providing a comprehensive analysis of key performance indicators. Several parallelization strategies for image recognition, adapted to different hardware and software configurations, including distributed data parallelism and distributed hardware processing, are analyzed. Selected optimization strategies are studied in detail, highlighting the related challenges and advantages of their implementation. Furthermore, the impact of different performance improvement techniques (DPO, LoRA, QLoRA, and QAT) on the tuning process of large language models is investigated. Experimental results illustrate how the nature of the task affects the iteration time in a multiprocessor environment, VRAM utilization, and overall memory transfers. Test scenarios are evaluated on the modern NVIDIA H100 GPU architecture.
format Preprint
id arxiv_https___arxiv_org_abs_2505_22905
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Profiling and optimization of multi-card GPU machine learning jobs
Lawenda, Marcin
Khloponin, Kyrylo
Samborski, Krzesimir
Szustak, Łukasz
Distributed, Parallel, and Cluster Computing
Performance
The effectiveness and efficiency of machine learning methodologies are crucial, especially with respect to the quality of results and computational cost. This paper discusses different model optimization techniques, providing a comprehensive analysis of key performance indicators. Several parallelization strategies for image recognition, adapted to different hardware and software configurations, including distributed data parallelism and distributed hardware processing, are analyzed. Selected optimization strategies are studied in detail, highlighting the related challenges and advantages of their implementation. Furthermore, the impact of different performance improvement techniques (DPO, LoRA, QLoRA, and QAT) on the tuning process of large language models is investigated. Experimental results illustrate how the nature of the task affects the iteration time in a multiprocessor environment, VRAM utilization, and overall memory transfers. Test scenarios are evaluated on the modern NVIDIA H100 GPU architecture.
title Profiling and optimization of multi-card GPU machine learning jobs
topic Distributed, Parallel, and Cluster Computing
Performance
url https://arxiv.org/abs/2505.22905