딥러닝 클러스터의 혁신: 자원 활용 극대화 스케줄링 알고리즘 'Hadar'와 'HadarE'


Abeda Sultana 등 연구진이 개발한 'Hadar'와 'HadarE' 알고리즘은 이기종 하드웨어의 특성을 고려한 딥러닝 클러스터 스케줄링을 통해 자원 활용률을 높이고 처리 속도를 획기적으로 향상시켰습니다. 'Hadar'는 기존 방식보다 1.2배 빠르며, 'HadarE'는 클러스터 자원 활용률을 1.45배 향상시키고 처리 속도를 최대 80%까지 단축시킵니다.

related iamge

GPU, TPU와 같은 고성능 가속기를 갖춘 강력한 클러스터에서 딥러닝 모델을 학습시키는 것은 여전히 큰 과제입니다. 기존의 스케줄링 방식은 이기종 하드웨어의 특성을 정교하게 고려하지 못하거나 자원 활용률이 낮다는 문제점을 가지고 있습니다.

하지만 최근, Abeda Sultana 등 연구진이 개발한 'Hadar' 와 'HadarE' 알고리즘은 이러한 문제점을 극복하고 딥러닝 클러스터의 효율성을 획기적으로 높였습니다.

'Hadar': 이기종 하드웨어의 특성을 고려한 최적화 프레임워크 기반의 작업 단위 스케줄링 알고리즘입니다. 공간적 및 시간적 차원에서 스케줄링 결정을 내림으로써 평균 작업 완료 시간을 단축하는 데 초점을 맞추고 있습니다. 기존 최첨단 이기종 인식 스케줄러인 Gavel과 비교했을 때, 총 처리 시간을 1.2배 단축시키는 놀라운 성능을 보였습니다.

'HadarE': 'Hadar'를 더욱 발전시킨 알고리즘으로, 각 작업을 여러 복사본으로 분할하여 서로 다른 노드에 있는 이기종 GPU에서 동시에 학습시키는 방식입니다. 이를 통해 클러스터 자원 활용률을 1.45배 향상시켰습니다. Amazon AWS 클러스터에서는 총 처리 시간을 50%, 연구실 클러스터에서는 무려 80%까지 단축시키는 성과를 거두었습니다. 뿐만 아니라, 'HadarE'는 'Hadar'보다 더욱 우수한 추론 성능을 갖춘 딥러닝 모델을 생성합니다.

이러한 연구 결과는 딥러닝 모델 학습의 속도와 효율성을 크게 향상시킬 뿐만 아니라, 클라우드 기반 딥러닝 서비스의 경제성을 높이는 데에도 크게 기여할 것으로 기대됩니다. 'Hadar'와 'HadarE'는 딥러닝 분야의 발전에 중요한 이정표를 세운 혁신적인 알고리즘으로 평가받고 있습니다. 앞으로 이러한 기술이 더욱 발전하여 더욱 빠르고 효율적인 딥러닝 환경을 구축하는 데 기여할 것으로 예상됩니다.


*이 기사는 AI가 생성한 내용으로, 일부 정보가 실제와 다를 수 있습니다. 정확한 확인을 위해 추가적인 검증을 권장드립니다.

Reference

[arxiv] Resource Heterogeneity-Aware and Utilization-Enhanced Scheduling for Deep Learning Clusters

Published:  (Updated: )

Author: Abeda Sultana, Nabin Pakka, Fei Xu, Xu Yuan, Li Chen, Nian-Feng Tzeng

http://arxiv.org/abs/2503.10918v1