23M 파라미터 소형 확산 모델 'Talus', 게임 지형 생성 시연

요약

Talus는 2300만 개의 파라미터를 가진 게임 지형 생성용 확산 모델로, 웹 브라우저의 WebGPU 환경에서 구동 가능합니다. 이 모델은 지형 유형과 평균 고도, 경사도, 수분 함량 등 5가지 속성 조건을 바탕으로 64x64 크기의 높이맵을 생성합니다. 개발자는 4만 5천 개의 자체 생성 데이터를 학습에 활용했으며, RTX 5060(8GB) GPU를 사용하여 약 4.5시간 만에 모델을 완성했습니다. 픽셀 공간 U-Net 구조를 채택하고 v-prediction 및 50단계 DDIM 샘플러 등을 적용해 효율성을 높였습니다. 특히 각 속성에 학습된 '알 수 없음(unknown)' 임베딩을 부여하여 추론 시 사용자가 원하는 속성 조합만으로도 지형 생성이 가능합니다.

AI가 원문을 요약한 내용으로, 부정확할 수 있습니다.