Daily Arxiv

전 세계에서 발간되는 인공지능 관련 논문을 정리하는 페이지 입니다.
본 페이지는 Google Gemini를 활용해 요약 정리하며, 비영리로 운영 됩니다.
논문에 대한 저작권은 저자 및 해당 기관에 있으며, 공유 시 출처만 명기하면 됩니다.

Unlocking Dataset Distillation with Diffusion Models

Created by
  • Haebom

저자

Brian B. Moser, Federico Raue, Sebastian Palacio, Stanislav Frolov, Andreas Dengel

개요

Latent Dataset Distillation with Diffusion Models (LD3M)은 데이터셋을 작은 합성 샘플로 축소하는 기술인 데이터셋 증류를 위한 새로운 방법론입니다. 기존 증류 방식이 GAN이나 오토인코더에 의존하거나 고정된 확산 prior에서 샘플링하는 것과 달리, LD3M은 사전 훈련된 잠재 확산 모델을 통해 gradient 기반의 증류된 잠재 및 클래스 임베딩을 end-to-end 방식으로 학습합니다. LD3M은 선형적으로 감소하는 skip connection을 사용하여 확산 모델 가중치 미세 조정 없이 수십 단계에 걸쳐 gradient 신호를 유지하며, ImageNet subsets에서 기존 SOTA 대비 최대 4.8%p (1 IPC) 및 4.2%p (10 IPC) 향상을 보입니다.

시사점, 한계점

확산 모델을 활용한 데이터셋 증류를 위한 새로운 접근 방식 제시
기존 증류 방법론 대비 성능 향상 (ImageNet subset 기준)
선형적으로 감소하는 skip connection을 통해 gradient vanishing 문제를 해결
확산 모델 가중치 미세 조정 불필요
확산 모델을 사용하기 때문에, 확산 모델 학습에 필요한 계산 비용이 존재할 수 있음
ImageNet subset에 대한 실험 결과만 제시되어, 다른 데이터셋에 대한 일반화 성능은 추가 검증 필요
👍