엔비디아(NVIDIA)는 대규모 언어모델(LLM) 학습의 효율성을 획기적으로 높이는 새로운 기술인 '비균일 텐서 병렬화(Nonuniform Tensor Parallelism)'를 공개했습니다. 이 기술은 LLM 학습 과정에서 GPU 자원 활용도를 극대화하여, AI 모델 개발자들이 더 빠르고 경제적으로 모델을 훈련할 수 있도록 돕는 것을 목표로 합니다.
기존의 텐서 병렬화(Tensor Parallelism)는 모델의 모든 레이어에 동일한 병렬화 전략을 적용했습니다. 하지만 엔비디아의 비균일 텐서 병렬화는 LLM의 각기 다른 레이어 특성을 고려하여, 레이어별로 최적화된 병렬화 방식을 동적으로 적용하는 것이 핵심입니다. 예를 들어, 어텐션(attention) 레이어와 피드포워드(feed-forward) 레이어는 계산 및 통신 패턴이 다르므로, 각 레이어에 가장 적합한 병렬화 크기를 선택해 GPU 간 통신 오버헤드를 줄이고 계산 효율을 높이는 방식입니다. 엔비디아는 이 기술을 통해 학습 처리량(goodput)을 최대 20%까지 향상시킬 수 있다고 밝혔습니다.
이 기술은 특히 수십억, 수조 개의 매개변수를 가진 초대규모 LLM을 훈련할 때 그 진가를 발휘할 것으로 보입니다. LLM 학습은 막대한 컴퓨팅 자원과 시간이 소요되는 작업이므로, 학습 효율 20% 증가는 개발 비용을 크게 절감하고 모델 개발 주기를 단축하는 효과를 가져올 수 있습니다. 이는 AI 연구 개발의 속도를 높이고, 더 복잡하고 강력한 AI 모델의 등장을 가속화하는 중요한 진전으로 평가됩니다.