GOGH: Correlation-Guided Orchestration of GPUs in Heterogeneous Clusters

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Raeisi, Ahmad, Dolati, Mahdi, Darabi, Sina, Talebi, Sadegh, Eugster, Patrick, Khonsari, Ahmad
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866914098649759744
author Raeisi, Ahmad
Dolati, Mahdi
Darabi, Sina
Talebi, Sadegh
Eugster, Patrick
Khonsari, Ahmad
author_facet Raeisi, Ahmad
Dolati, Mahdi
Darabi, Sina
Talebi, Sadegh
Eugster, Patrick
Khonsari, Ahmad
contents The growing demand for computational resources in machine learning has made efficient resource allocation a critical challenge, especially in heterogeneous hardware clusters where devices vary in capability, age, and energy efficiency. Upgrading to the latest hardware is often infeasible, making sustainable use of existing, mixed-generation resources essential. In this paper, we propose a learning-based architecture for managing machine learning workloads in heterogeneous clusters. The system operates online, allocating resources to incoming training or inference requests while minimizing energy consumption and meeting performance requirements. It uses two neural networks: the first provides initial estimates of how well a new model will utilize different hardware types and how it will affect co-located models. An optimizer then allocates resources based on these estimates. After deployment, the system monitors real performance and uses this data to refine its predictions via a second neural network. This updated model improves estimates not only for the current hardware but also for hardware not initially allocated and for co-location scenarios not yet observed. The result is an adaptive, iterative approach that learns over time to make more effective resource allocation decisions in heterogeneous deep learning clusters.
format Preprint
id arxiv_https___arxiv_org_abs_2510_15652
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle GOGH: Correlation-Guided Orchestration of GPUs in Heterogeneous Clusters
Raeisi, Ahmad
Dolati, Mahdi
Darabi, Sina
Talebi, Sadegh
Eugster, Patrick
Khonsari, Ahmad
Distributed, Parallel, and Cluster Computing
Machine Learning
The growing demand for computational resources in machine learning has made efficient resource allocation a critical challenge, especially in heterogeneous hardware clusters where devices vary in capability, age, and energy efficiency. Upgrading to the latest hardware is often infeasible, making sustainable use of existing, mixed-generation resources essential. In this paper, we propose a learning-based architecture for managing machine learning workloads in heterogeneous clusters. The system operates online, allocating resources to incoming training or inference requests while minimizing energy consumption and meeting performance requirements. It uses two neural networks: the first provides initial estimates of how well a new model will utilize different hardware types and how it will affect co-located models. An optimizer then allocates resources based on these estimates. After deployment, the system monitors real performance and uses this data to refine its predictions via a second neural network. This updated model improves estimates not only for the current hardware but also for hardware not initially allocated and for co-location scenarios not yet observed. The result is an adaptive, iterative approach that learns over time to make more effective resource allocation decisions in heterogeneous deep learning clusters.
title GOGH: Correlation-Guided Orchestration of GPUs in Heterogeneous Clusters
topic Distributed, Parallel, and Cluster Computing
Machine Learning
url https://arxiv.org/abs/2510.15652