Daily Arxiv

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

Influence Functions for Scalable Data Attribution in Diffusion Models

Created by
  • Haebom
Category
Empty

저자

Bruno Mlodozeniec, Runa Eschenhagen, Juhan Bae, Alexander Immer, David Krueger, Richard Turner

개요

본 논문은 확산 모델의 데이터 귀속 및 해석성 문제 해결을 위해 영향 함수 프레임워크를 개발합니다. 기존 지도 학습에서의 영향 함수는 특정 데이터 제거 시 손실 변화 예측에 사용되었지만, 본 논문에서는 확산 모델에 적용하여 특정 예시 생성 확률 변화를 여러 근사 측정값을 통해 예측하는 방법을 제시합니다. 여러 기존 방법들을 본 프레임워크의 특정 설계 선택으로 재해석하고, 확산 모델에 특화된 일반화된 Gauss-Newton 행렬 기반 K-FAC 근사를 통해 Hessian 계산의 확장성을 확보합니다. 제안된 방법이 Linear Data-modelling Score (LDS)나 상위 영향 제거 재학습과 같은 일반적인 평가에서 기존 데이터 귀속 방법보다 우수한 성능을 보이며, 별도의 하이퍼파라미터 튜닝이 필요 없음을 보입니다.

시사점, 한계점

시사점:
확산 모델의 데이터 귀속 및 해석성 문제에 대한 새로운 해결책 제시.
기존 방법들을 통합하는 일반적인 영향 함수 프레임워크 제공.
효율적인 Hessian 계산을 위한 K-FAC 근사 기법 개발.
제안된 방법의 우수한 성능 및 하이퍼파라미터 튜닝 불필요성 입증.
한계점:
K-FAC 근사의 정확도 및 계산 비용에 대한 추가적인 분석 필요.
다양한 유형의 확산 모델에 대한 일반화 가능성 검증 필요.
실제 응용 분야에서의 효용성 및 제한점에 대한 추가 연구 필요.
👍