NVIDIA, TAO 에이전트로 자동화된 LoRA 미세조정 실험 공개
에이전트가 데이터 정제부터 하이퍼파라미터 튜닝까지 수행하여 효율적인 학습이 가능함을 입증. MLOps 자동화 워크플로우 예시로 참고 가능.
요약
NVIDIA는 Codex를 활용해 Cosmos 3 Nano 모델의 성능을 하루 만에 정확도 54.41%에서 93.35%로 향상시켰다. Toyota의 Woven Traffic Safety 데이터셋을 기반으로 한 4지선다형 비디오 추론 실험에서 NVIDIA TAO 에이전트 기술이 메타데이터 수정부터 LoRA 구성, 학습 및 평가까지 자율적으로 수행했다. 첫 번째 프롬프트 실행 시 8개의 A100 GPU에서 약 30분 만에 87.14%의 정확도를 달성했고, 두 번째 프롬프트로 43개의 AutoML 병렬 실험을 수행해 19.5시간 만에 93.35%에 도달했다. 특히 LoRA 방식은 전체 파라미터 학습 대비 약 7배 적은 GPU 시간을 소요했다. 이번 실험은 에이전트 기술이 전문적인 머신러닝 인프라를 운영하는 핵심 인터페이스로 진화하고 있음을 보여준다.
AI가 원문을 요약한 내용으로, 부정확할 수 있습니다.
원문 제목 @kimmonismus: NVIDIA says Codex post-trained Cosmos 3 Nano from 54.41% to 93.35% accuracy in one day - with two prompts. The experiment used Toy
원문 보기 ↗