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]"

# Optionally install W&B for logging
pip install wandb

사용법

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을 사용한 하이퍼파라미터 튜닝을 위한 사용하기 쉬운 인터페이스를 제공합니다. 튜닝 과정을 사용자 정의할 수 있는 여러 인수를 허용합니다. 각 매개변수에 대한 상세 설명은 다음과 같습니다:

이러한 매개변수를 사용자 정의함으로써 특정 요구 사항과 가용한 컴퓨팅 자원에 맞춰 하이퍼파라미터 최적화 과정을 미세 조정할 수 있습니다.

기본 탐색 공간(Search Space) 설명

다음 표는 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,
)

In the code snippet above, we create a YOLO model with the "yolo26n.pt" pretrained weights. Then, we call the tune() method, specifying the dataset configuration with "coco8.yaml". We provide a custom search space for the initial learning rate lr0 using a dictionary with the key "lr0" and the value tune.uniform(1e-5, 1e-2). Finally, we pass additional training arguments, such as the number of epochs directly to the tune method as epochs=50.

검색 알고리즘 예제

이름으로 Ray Tune 검색 전략을 선택할 수 있습니다. 문자열 기반 검색 알고리즘의 경우, Ultralytics는 현재 태스크의 기본 최적화 메트릭과 mode="max"를 자동으로 사용합니다. search_alg가 제공되지 않으면 Ray Tune은 기본 무작위 검색 전략인 BasicVariantGenerator를 사용합니다.

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,
)

검색기의 샘플러를 사용자 정의하거나 메트릭 및 최적화 모드를 재정의해야 하는 경우, 문자열 대신 인스턴스화된 Ray Tune 검색기 객체를 전달하십시오.

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으로 중단된 하이퍼파라미터 튜닝 세션 재개하기

You can resume an interrupted Ray Tune session by passing resume=True. You can optionally pass the directory name used by Ray Tune under runs/{task} to resume. Otherwise, it would resume the last interrupted session. You don't need to provide the iterations and space again, but you need to provide the rest of the training arguments again including data and 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을 사용하여 하이퍼파라미터 튜닝 실험을 실행한 후, 얻은 결과에 대해 다양한 분석을 수행할 수 있습니다. 이 가이드에서는 이러한 결과를 처리하고 분석하기 위한 일반적인 워크플로우를 안내합니다.

디렉토리에서 튜닝 실험 결과 불러오기

tuner.fit()으로 튜닝 실험을 실행한 후, 디렉토리에서 결과를 불러올 수 있습니다. 이는 특히 초기 학습 스크립트가 종료된 후 분석을 수행할 때 유용합니다.

experiment_path = f"{storage_path}/{exp_name}"
print(f"Loading results from {experiment_path}...")

restored_tuner = tune.Tuner.restore(experiment_path, trainable=train_mnist)
result_grid = restored_tuner.get_results()

기본 실험 수준 분석

시도가 어떻게 수행되었는지에 대한 개요를 확인합니다. 시도 중에 오류가 있었는지 빠르게 확인할 수 있습니다.

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["mean_accuracy"],
        label=f"Trial {i}",
    )

plt.xlabel("Training Iterations")
plt.ylabel("Mean Accuracy")
plt.legend()
plt.show()

요약

이 가이드에서는 Ultralytics를 사용하여 Ray Tune으로 실행된 실험 결과를 분석하는 일반적인 워크플로우를 다루었습니다. 주요 단계는 디렉토리에서 실험 결과 불러오기, 기본 실험 수준 및 시도 수준 분석 수행, 메트릭 플로팅하기를 포함합니다.

하이퍼파라미터 튜닝 실험을 최대한 활용하려면 Ray Tune의 결과 분석 문서 페이지를 참조하여 자세히 알아보십시오.

FAQ

Ray Tune을 사용하여 YOLO26 모델의 하이퍼파라미터를 튜닝하려면 어떻게 해야 하나요?

Ray Tune을 사용하여 Ultralytics YOLO26 모델의 하이퍼파라미터를 튜닝하려면 다음 단계를 따르십시오:

1. 필수 패키지 설치:

   pip install -U ultralytics "ray[tune]"
   pip install wandb # optional for logging

2. YOLO26 모델 불러오기 및 튜닝 시작:

   from ultralytics import YOLO
   
   # Load a YOLO26 model
   model = YOLO("yolo26n.pt")
   
   # Start tuning with the COCO8 dataset
   result_grid = model.tune(data="coco8.yaml", use_ray=True)

이는 Ray Tune의 고급 검색 전략과 병렬 처리를 활용하여 모델의 하이퍼파라미터를 효율적으로 최적화합니다. 자세한 내용은 Ray Tune 문서를 확인하십시오.

Ray Tune을 이용한 YOLO26 튜닝의 기본 하이퍼파라미터는 무엇인가요?

Ultralytics YOLO26은 Ray Tune을 이용한 튜닝에 대해 다음과 같은 기본 하이퍼파라미터를 사용합니다:

이러한 하이퍼파라미터는 사용자의 특정 요구 사항에 맞게 조정할 수 있습니다. 전체 목록 및 자세한 내용은 Hyperparameter Tuning 가이드를 참조하십시오.

YOLO26 모델 튜닝에 Weights & Biases를 어떻게 통합할 수 있습니까?

Ultralytics YOLO26 튜닝 프로세스에 Weights & Biases(W&B)를 통합하려면 다음 단계를 따르십시오:

1. W&B 설치:

   pip install wandb

2. 튜닝 스크립트 수정:

   import wandb
   
   from ultralytics import YOLO
   
   wandb.init(project="YOLO-Tuning", entity="your-entity")
   
   # Load YOLO model
   model = YOLO("yolo26n.pt")
   
   # Tune hyperparameters
   result_grid = model.tune(data="coco8.yaml", use_ray=True)

이 설정을 사용하면 튜닝 프로세스를 모니터링하고, 하이퍼파라미터 구성을 추적하며, W&B에서 결과를 시각화할 수 있습니다.

YOLO26 하이퍼파라미터 최적화를 위해 Ray Tune을 사용해야 하는 이유는 무엇입니까?

Ray Tune은 하이퍼파라미터 최적화를 위해 다음과 같은 수많은 이점을 제공합니다:

• 고급 탐색 전략: Bayesian Optimization 및 HyperOpt와 같은 알고리즘을 활용하여 효율적인 파라미터 탐색을 수행합니다.
• 병렬 처리: 여러 시도의 병렬 실행을 지원하여 튜닝 프로세스 속도를 크게 높입니다.
• 조기 종료: ASHA와 같은 전략을 사용하여 성능이 낮은 시도를 조기에 종료함으로써 계산 자원을 절약합니다.

Ray Tune은 Ultralytics YOLO26과 원활하게 통합되어 하이퍼파라미터를 효과적으로 튜닝할 수 있는 사용하기 쉬운 인터페이스를 제공합니다. 시작하려면 Hyperparameter Tuning 가이드를 확인하십시오.

YOLO26 하이퍼파라미터 튜닝을 위한 사용자 지정 검색 공간은 어떻게 정의할 수 있습니까?

Ray Tune을 사용하여 YOLO26 하이퍼파라미터 튜닝을 위한 사용자 지정 검색 공간을 정의하려면 다음을 수행하십시오:

from ray import tune

from ultralytics import YOLO

model = YOLO("yolo26n.pt")
search_space = {"lr0": tune.uniform(1e-5, 1e-2), "momentum": tune.uniform(0.7, 0.98)}
result_grid = model.tune(data="coco8.yaml", space=search_space, use_ray=True)

이를 통해 튜닝 프로세스 중에 탐색할 초기 학습률 및 모멘텀과 같은 하이퍼파라미터 범위를 사용자 지정할 수 있습니다. 고급 구성에 대해서는 Custom Search Space Example 섹션을 참조하십시오.