녹색 확인
링크가 클립보드에 복사됨

텐서보드 통합으로 훈련 메트릭 시각화하기

강력한 시각화 및 실험 추적을 통해 Ultralytics YOLO11 워크플로우를 개선하여 모델 성능을 최적화하는 TensorBoard 통합에 대해 알아보세요.

신뢰할 수 있는 컴퓨터 비전 모델을 개발하려면 데이터 수집, 모델 훈련, 잠재적인 문제를 해결하고 성능을 개선하기 위한 반복적인 미세 조정 프로세스 등 여러 단계를 거쳐야 합니다. 이러한 단계 중 모델 훈련이 가장 중요한 단계로 간주되는 경우가 많습니다.

교육 과정을 시각화하면 이 단계를 보다 명확하게 파악하는 데 도움이 될 수 있습니다. 그러나 상세한 그래프를 만들고, 시각적 데이터를 분석하고, 차트를 생성하려면 많은 시간과 노력이 필요할 수 있습니다. Ultralytics 에서 지원하는 텐서보드 통합과 같은 도구는 간단한 시각 자료와 심층 분석을 제공하여 이 과정을 간소화합니다.

텐서보드는 모델의 학습 진행 상황에 대한 실시간 인사이트를 제공하는 신뢰할 수 있는 시각화 도구입니다. Ultralytics YOLO 모델과 함께 사용할 경우 Ultralytics YOLO11과 같이 객체 감지 및 인스턴스 분할과 같은 컴퓨터 비전 작업에서 정확도로 유명한 모델과 함께 사용하면 TensorBoard는 훈련 진행 상황을 추적할 수 있는 시각적 대시보드를 제공합니다. 이 통합을 통해 주요 지표를 추적하고, 훈련 성과를 모니터링하고, 실행 가능한 인사이트를 확보하여 모델을 미세 조정하고 원하는 결과를 얻을 수 있습니다.

이 글에서는 실시간 시각화, 실행 가능한 인사이트, 성능 최적화를 위한 실용적인 팁을 통해 Ultralytics YOLO11 모델 트레이닝을 개선하는 방법을 살펴봅니다.

텐서보드란 무엇인가요?

텐서보드는 TensorFlow 에서 개발한 오픈 소스 시각화 도구입니다. 머신 러닝 및 컴퓨터 비전 모델의 개발과 학습을 지원하는 필수 메트릭과 시각화를 제공합니다. 이 툴킷의 대시보드는 그래프, 이미지, 텍스트, 오디오 등 다양한 형식의 데이터를 표시하여 모델 동작에 대한 심층적인 이해를 제공합니다. 이러한 시각화를 통해 데이터에 기반한 더 나은 의사 결정을 내려 모델 성능을 개선할 수 있습니다.

그림 1. 모델 학습 그래프가 있는 TensorBoard 대시보드.

텐서보드의 주요 기능

텐서보드는 모델 워크플로우의 여러 측면을 향상시키는 다양한 기능을 제공합니다. 예를 들어 정확도, 학습률, 손실과 같은 성능 메트릭을 실시간으로 시각화하여 모델이 학습하는 방식에 대한 귀중한 인사이트를 제공하고 학습 중 과적합 또는 과소적합과 같은 문제를 강조 표시할 수 있습니다. 

또 다른 흥미로운 기능은 '그래프' 도구로, 데이터가 모델을 통해 어떻게 흐르는지 시각적으로 매핑합니다. 이 그래픽 표현을 사용하면 모델의 아키텍처와 복잡성을 한 눈에 쉽게 이해할 수 있습니다.

다음은 텐서보드 통합의 다른 주요 기능입니다:

  • 데이터 분포를 분석합니다: 가중치, 편향, 활성화와 같은 모델의 내부 값에 대한 자세한 분포를 제공합니다. 이를 통해 데이터가 모델 네트워크를 통해 어떻게 흐르는지 파악하고 개선이 필요한 잠재적 영역을 식별할 수 있습니다.
  • 데이터 패턴을 평가합니다: '히스토그램' 기능을 사용하면 가중치, 편향도, 기울기 등 모델 매개변수의 시간 경과에 따른 분포를 시각화할 수 있습니다. 이러한 패턴을 읽음으로써 모델의 잠재적인 편향과 병목 현상을 파악할 수 있습니다.
  • 고차원 데이터를 탐색하세요: '프로젝터' 기능은 복잡한 고차원 데이터를 저차원 공간으로 변환할 수 있습니다. 이를 통해 모델이 서로 다른 객체를 어떻게 그룹화하는지를 더 쉽게 시각화할 수 있습니다.
  • 모델 예측을 시각화하세요: TensorBoard를 사용하면 입력 이미지, 정확한 레이블(기준값), 모델의 예측을 나란히 비교할 수 있습니다. 이렇게 하면 모델이 무언가를 잘못 식별하거나(오탐) 중요한 것을 놓치는 경우(오탐)와 같은 실수를 쉽게 발견할 수 있습니다. 

Ultralytics YOLO 모델 개요

Ultralytics YOLO (You Only Look Once) 모델은 오늘날 가장 인기 있고 널리 사용되는 컴퓨터 비전 모델 중 하나입니다. 주로 객체 감지 및 인스턴스 분할과 같은 고성능 컴퓨터 비전 작업에 사용됩니다. 속도, 정확성, 사용 편의성으로 널리 알려진 YOLO 모델은 농업, 제조, 의료 등 다양한 산업 분야에서 채택되고 있습니다. 

이 모든 것은 Ultralytics YOLOv5와 같은 도구로 Vision AI 모델을 더 쉽게 사용할 수 있게 해준 PyTorch 에서 시작되었습니다. Next, Ultralytics YOLOv8 에는 포즈 추정 및 이미지 분류와 같은 기능이 추가되었습니다. 

이제 YOLO11 에서 더 나은 성능을 제공합니다. 실제로 YOLO11m은 YOLOv8m 보다 22% 적은 매개변수를 사용하면서도 COCO 데이터 세트에서 더 높은 평균 정밀도(mAP)를 달성하여 물체를 더 정확하고 효율적으로 감지할 수 있습니다.

그림 2. 물체 감지를 위해 YOLO11 을 사용한 예시.

텐서보드 통합을 통해 주요 메트릭을 추적 및 모니터링하고, 심층 분석을 수행하고, YOLO11 의 맞춤형 교육 및 개발 프로세스를 간소화할 수 있습니다. 실시간 시각화 기능을 통해 YOLO11 을 보다 효율적으로 구축, 미세 조정 및 최적화하여 개발자와 AI 연구자가 적은 노력으로 더 나은 결과를 얻을 수 있도록 지원합니다.

TensorBoard 통합 사용

사용자 지정 교육( Ultralytics YOLO11 )을 하면서 TensorBoard 통합을 사용하는 것은 쉽습니다. TensorBoard는 Ultralytics Python 패키지와 원활하게 통합되므로 추가 설치나 설정 단계가 필요하지 않습니다. 

훈련이 시작되면 패키지는 손실, 정확도, 학습률, 평균 정밀도(mAP)와 같은 주요 지표를 지정된 디렉토리에 자동으로 기록하여 상세한 성능 분석을 가능하게 합니다. 출력 메시지를 통해 TensorBoard가 트레이닝 세션을 적극적으로 모니터링하고 있음을 확인할 수 있으며, 'http://localhost:6006/'과 같은 URL에서 대시보드를 볼 수 있습니다.  

기록된 데이터에 액세스하려면 URL을 사용하여 TensorBoard를 실행하고 손실, 정확도, 학습률, 맵과 같은 메트릭의 실시간 시각화와 함께 그래프, 스칼라, 히스토그램과 같은 도구를 사용하여 더 심층적인 분석을 할 수 있습니다. 

이러한 역동적인 대화형 비주얼을 통해 교육 진행 상황을 쉽게 모니터링하고, 문제를 발견하고, 개선이 필요한 부분을 정확히 파악할 수 있습니다. 이러한 기능을 활용하여 TensorBoard를 통합하면 YOLO11 교육 프로세스를 투명하고 체계적이며 이해하기 쉽게 유지할 수 있습니다.

Google Colab에서 작업하는 사용자의 경우, 구성 명령이 실행되는 노트북 셀 내에 TensorBoard가 직접 통합되어 교육 인사이트에 원활하게 액세스할 수 있습니다. 

설치에 대한 단계별 안내와 모범 사례는 YOLO11 설치 가이드를 참조하세요. 필수 패키지를 설정하는 동안 문제가 발생하는 경우 일반적인 문제 가이드에서 유용한 솔루션과 문제 해결 팁을 확인할 수 있습니다. 

교육 지표 분석

모델 성능을 평가하려면 주요 학습 메트릭을 이해하는 것이 필수적이며, TensorBoard 통합은 이를 위한 심층적인 시각화를 제공합니다. 하지만 어떻게 작동할까요?

훈련이 진행됨에 따라 유효성 검사 데이터에 대한 모델의 정확도가 어떻게 향상되는지 보여주는 그래프인 평가 정확도 곡선을 관찰하고 있다고 가정해 보겠습니다. 처음에는 정확도가 급격히 증가하는 것을 볼 수 있으며, 이는 모델이 빠르게 학습하고 성능이 향상되고 있음을 나타냅니다. 

그러나 학습이 계속되면 개선 속도가 느려지고 곡선이 평평해지기 시작할 수 있습니다. 이러한 평탄화는 모델이 최적의 상태에 가까워지고 있음을 의미합니다. 이 시점 이후에도 학습을 계속하면 큰 개선 효과를 기대하기 어렵고 과적합으로 이어질 수 있습니다. 

아래 그림과 같이 텐서보드 통합을 통해 이러한 추세를 시각화하면 모델의 최적 상태를 파악하고 훈련 프로세스에 필요한 조정을 수행할 수 있습니다.

그림 3. 텐서보드 그래프의 예시. (이미지 제공: 저자).

텐서보드 통합의 이점

텐서보드 통합은 YOLO11 모델 학습 및 성능 최적화를 개선하는 다양한 이점을 제공합니다. 몇 가지 주요 이점은 다음과 같습니다:

  • 실험 비교: 여러 훈련 실행을 쉽게 비교하여 가장 성능이 좋은 모델 구성을 파악할 수 있습니다.
  • 시간과 노력을 절약하세요: 이 통합으로 교육 지표를 모니터링하고 분석하는 프로세스가 간소화되어 수작업이 줄어들고 모델 개발 속도가 빨라집니다.
  • 사용자 지정 지표 추적: 애플리케이션과 관련된 특정 메트릭을 모니터링하도록 로깅을 구성하여 모델에 맞는 심층적인 인사이트를 제공할 수 있습니다.
  • 효율적인 리소스 활용: 교육 지표 외에도 사용자 지정 로깅을 통해 GPU 사용량, 메모리 할당 및 계산 시간을 모니터링하여 하드웨어 성능을 최적화할 수 있습니다.
그림 3. 텐서보드 통합 사용의 이점. 작성자 이미지.

텐서보드 통합을 사용하기 위한 모범 사례

이제 텐서보드 통합이 무엇이고 어떻게 사용하는지 이해했으니, 이 통합을 사용하는 몇 가지 모범 사례를 살펴보겠습니다: 

  • 명확한 명명 규칙을 사용하세요: 모델 유형, 데이터 세트 및 주요 매개 변수를 포함하는 실험의 구조화된 이름을 만들어 혼란을 피하고 비교를 쉽게 하세요.
  • 최적의 로깅 빈도를 설정하세요: YOLO11 모델 학습 프로세스의 속도를 저하시키지 않으면서 유용한 인사이트를 제공하는 간격으로 데이터를 로깅합니다.
  • 재현성과 호환성을 보장하세요: 새로운 기능, 버그 수정, 진화하는 데이터 요구 사항과의 호환성을 보장하기 위해 TensorBoard, Ultralytics, 데이터 세트와 같은 패키지를 정기적으로 업데이트하세요.

이러한 모범 사례를 따르면 YOLO11 개발 프로세스를 보다 효율적이고 체계적이며 생산적으로 만들 수 있습니다. 컴퓨터 비전 워크플로를 개선하고 모델의 잠재력을 극대화하기 위해 사용 가능한 다른 통합 기능을 살펴보세요.

주요 요점

Ultralytics 에서 지원하는 텐서보드 통합을 사용하면 모델 개발 프로세스를 더 쉽게 모니터링하고 추적할 수 있어 전반적인 성능이 향상됩니다. 직관적인 시각화 기능을 갖춘 TensorBoard는 학습 지표에 대한 인사이트를 제공하고, 손실과 정확도의 추세를 추적하며, 실험 전반을 원활하게 비교할 수 있게 해줍니다.

데이터 준비, 설정 미세 조정, 메트릭 분석을 간소화하여 모델 성능을 최적화함으로써 의사 결정을 간소화합니다. 또한 이러한 기능은 컴퓨터 비전 애플리케이션의 출시 기간을 단축하고 개발 비용을 절감하는 등 상당한 비즈니스 이점을 제공합니다. 개발자는 명확한 이름 지정 및 업데이트 유지와 같은 모범 사례를 사용하여 교육을 더 쉽게 진행할 수 있습니다. 개발자는 YOLO11 과 같은 고급 컴퓨터 비전 모델을 사용하여 보다 효율적으로 작업하고 새로운 옵션을 탐색할 수 있습니다.

유니티 커뮤니티의 일원이 되어 GitHub 리포지토리를 살펴보고 AI에 대해 자세히 알아보세요. 솔루션 페이지를 방문하여 제조 분야의 컴퓨터 비전과 의료 분야의 AI가 어떻게 혁신을 주도하고 있는지 알아보세요. 지금 바로 라이선스 옵션을 확인하여 비전 AI 여정을 시작하는 것도 잊지 마세요!

Facebook 로고트위터 로고LinkedIn 로고복사 링크 기호

이 카테고리에서 자세히 보기

인공지능의 미래
를 함께 만들어 갑시다!

머신 러닝의 미래와 함께하는 여정 시작하기