Ray Tune 및 YOLO26을 활용한 효율적인 하이퍼파라미터 튜닝#

하이퍼파라미터 튜닝은 최적의 하이퍼파라미터 조합을 찾아 모델 성능을 최고 수준으로 끌어올리는 데 필수적입니다. 이를 위해 다양한 하이퍼파라미터로 트라이얼을 실행하고 각 트라이얼의 성능을 평가합니다.

Ultralytics YOLO26 및 Ray Tune으로 튜닝 가속화#

Ultralytics YOLO26은 하이퍼파라미터 튜닝에 Ray Tune을 통합하여 YOLO26 모델 하이퍼파라미터의 최적화 과정을 간소화합니다. Ray Tune을 사용하면 고급 검색 전략, 병렬 처리 및 조기 중지 기능을 활용하여 튜닝 프로세스를 가속화할 수 있습니다.

Ray Tune#

Ray Tune은 효율성과 유연성을 위해 설계된 하이퍼파라미터 튜닝 라이브러리입니다. 다양한 검색 전략, 병렬 처리 및 조기 중지 전략을 지원하며, Ultralytics YOLO26을 비롯한 인기 있는 머신 러닝 프레임워크와 원활하게 통합됩니다.

Weights & Biases와의 통합#

YOLO26은 튜닝 프로세스를 모니터링하기 위해 Weights & Biases와 선택적으로 통합할 수도 있습니다.

설치#

필요한 패키지를 설치하려면 다음을 실행합니다:

# Install and update Ultralytics and Ray Tune packages
pip install -U ultralytics "ray[tune]" optuna

# Optionally install and enable W&B for per-trial logging
pip install wandb
yolo settings wandb=True

사용법#

from ultralytics import YOLO

# Load a YOLO26n model
model = YOLO("yolo26n.pt")

# Start tuning hyperparameters for YOLO26n training on the COCO8 dataset
result_grid = model.tune(data="coco8.yaml", use_ray=True)

tune() 메서드 매개변수#

YOLO26의 tune() 메서드는 Ray Tune을 사용한 하이퍼파라미터 튜닝을 위한 사용하기 쉬운 인터페이스를 제공합니다. 튜닝 프로세스를 사용자 지정할 수 있는 여러 인수를 허용합니다. 다음은 각 매개변수에 대한 자세한 설명입니다:

이러한 매개변수를 사용자 지정하면 특정 요구 사항과 사용 가능한 컴퓨팅 리소스에 맞게 하이퍼파라미터 최적화 프로세스를 세밀하게 조정할 수 있습니다.

기본 검색 공간 설명#

다음 표에는 Ray Tune을 사용한 YOLO26 하이퍼파라미터 튜닝의 기본 검색 공간 매개변수가 나와 있습니다. 각 매개변수에는 tune.uniform()으로 정의된 특정 값 범위가 있습니다.

사용자 지정 검색 공간 예제#

이 예제에서는 Ray Tune과 YOLO26을 사용한 하이퍼파라미터 튜닝에 사용자 지정 검색 공간을 사용하는 방법을 보여줍니다. 사용자 지정 검색 공간을 지정하면 관심 있는 특정 하이퍼파라미터에 튜닝 프로세스를 집중할 수 있습니다.

from ray import tune

from ultralytics import YOLO

# Define a YOLO model
model = YOLO("yolo26n.pt")

# Run Ray Tune on the model
result_grid = model.tune(
    data="coco8.yaml",
    space={"lr0": tune.uniform(1e-5, 1e-2)},
    epochs=50,
    use_ray=True,
)

위 코드 스니펫에서는 "yolo26n.pt" 사전 학습 가중치를 사용하여 YOLO 모델을 생성합니다. 그런 다음 "coco8.yaml"로 데이터셋 구성을 지정하여 tune() 메서드를 호출합니다. 키가 "lr0"이고 값이 tune.uniform(1e-5, 1e-2)인 딕셔너리를 사용하여 초기 학습률 lr0에 대한 사용자 지정 검색 공간을 제공합니다. 마지막으로 에포크 수와 같은 추가 학습 인수를 epochs=50으로 tune 메서드에 직접 전달합니다.

검색 알고리즘 예제#

이름으로 Ray Tune 검색 전략을 선택할 수 있습니다. 문자열 기반 검색 알고리즘의 경우 Ultralytics가 현재 작업의 기본 최적화 메트릭과 mode="max"을 자동으로 사용합니다. search_alg이 제공되지 않으면 Ultralytics는 병렬 처리를 고려한 제안 기능이 적용된 Optuna 다변량 TPE를 사용합니다.

from ultralytics import YOLO

# Define a YOLO model
model = YOLO("yolo26n.pt")

result_grid = model.tune(
    data="coco8.yaml",
    epochs=50,
    search_alg="optuna",
    use_ray=True,
)

searcher의 sampler를 사용자 지정하거나 메트릭 및 최적화 모드를 재정의해야 하는 경우 문자열 대신 인스턴스화된 Ray Tune searcher 객체를 전달합니다.

from ray.tune.search.optuna import OptunaSearch

from ultralytics import YOLO

# Define a YOLO model
model = YOLO("yolo26n.pt")

# Define a searcher
searcher = OptunaSearch(
    metric="metrics/mAP50-95(B)",
    mode="max",
)

result_grid = model.tune(
    data="coco8.yaml",
    epochs=50,
    search_alg=searcher,
    use_ray=True,
)

Ray Tune으로 중단된 하이퍼파라미터 튜닝 세션 재개#

resume=True을 전달하여 중단된 Ray Tune 세션을 재개할 수 있습니다. 선택적으로 Ray Tune이 runs/{task} 아래에서 사용하는 디렉터리 name을 전달하여 재개할 수 있습니다. 그렇지 않으면 마지막으로 중단된 세션이 재개됩니다. iterations 및 space를 다시 제공할 필요는 없지만, data 및 epochs을 포함한 나머지 학습 인수는 다시 제공해야 합니다.

from ultralytics import YOLO

# Define a YOLO model
model = YOLO("yolo26n.pt")

# Resume previous run
results = model.tune(use_ray=True, data="coco8.yaml", epochs=50, resume=True)

# Resume Ray Tune run with name 'tune_exp_2'
results = model.tune(use_ray=True, data="coco8.yaml", epochs=50, name="tune_exp_2", resume=True)

Ray Tune 결과 처리#

Ray Tune으로 하이퍼파라미터 튜닝 실험을 실행한 후에는 얻은 결과를 다양한 방식으로 분석할 수 있습니다. 이 가이드에서는 이러한 결과를 처리하고 분석하는 일반적인 워크플로를 안내합니다.

디렉터리에서 Tune 실험 결과 로드#

model.tune(use_ray=True)은(는) Ray ResultGrid을(를) 반환하고 실험을 runs/{task}/tune 아래에 저장하거나 사용자가 전달한 name 아래에 저장합니다. 원래 스크립트가 종료된 후 해당 결과를 로드하려면 세션을 재개하여 저장된 시도를 복원하고 동일한 ResultGrid을(를) 반환합니다.

from ultralytics import YOLO

model = YOLO("yolo26n.pt")
result_grid = model.tune(data="coco8.yaml", epochs=50, use_ray=True, resume=True)

기본 실험 수준 분석#

트라이얼의 성능에 대한 개요를 확인합니다. 트라이얼 중 오류가 있었는지 빠르게 확인할 수 있습니다.

if result_grid.errors:
    print("One or more trials failed!")
else:
    print("No errors!")

기본 트라이얼 수준 분석#

개별 트라이얼의 하이퍼파라미터 구성과 마지막으로 보고된 메트릭에 액세스합니다.

for i, result in enumerate(result_grid):
    print(f"Trial #{i}: Configuration: {result.config}, Last Reported Metrics: {result.metrics}")

트라이얼에서 보고된 메트릭 전체 기록 플로팅#

각 트라이얼에서 보고된 메트릭의 기록을 플로팅하여 시간에 따라 메트릭이 어떻게 변화했는지 확인할 수 있습니다.

import matplotlib.pyplot as plt

for i, result in enumerate(result_grid):
    plt.plot(
        result.metrics_dataframe["training_iteration"],
        result.metrics_dataframe["metrics/mAP50-95(B)"],
        label=f"Trial {i}",
    )

plt.xlabel("Training Iterations")
plt.ylabel("mAP50-95")
plt.legend()
plt.show()

요약#

이 가이드에서는 Ultralytics를 사용하여 Ray Tune으로 실행한 실험 결과를 분석하는 일반적인 워크플로를 살펴보았습니다. 주요 단계는 디렉터리에서 실험 결과를 로드하고, 기본적인 실험 수준 및 트라이얼 수준 분석을 수행하며, 메트릭을 플로팅하는 것입니다.

하이퍼파라미터 튜닝 실험을 최대한 활용하려면 Ray Tune의 결과 분석 문서 페이지를 확인해 보세요.

FAQ#