연구자는 디퓨전 모델 파인튜닝 과정의 번거로움을 해결하기 위해 Bracket이라는 오픈 소스 하이퍼파라미터 검색 도구를 개발했습니다.
Bracket은 데이터셋과 모델을 입력받아 Optuna TPE를 활용한 짧은 트레이닝을 병렬로 실행하며, 트레이닝 손실 추이와 VLM을 활용한 이미지 품질을 평가합니다.
결과적으로 Bracket은 각 설정의 성능을 비교하고 통계적 신뢰도를 제공하여 주관적인 판단을 객관적인 데이터로 대체합니다.
SDXL, Z-Image, Flux 등 다양한 모델과 LoRA, 전체 FT를 지원하며, GPU 환경에 맞춰 검색 공간을 조정하고, 초기 단계에 유용한 설정값을 제공합니다.