HyperAIHyperAI

Command Palette

Search for a command to run...

확산 모델의 일관성에 대한 랜덤 행렬 이론적 관점

Binxu Wang Jacob Zavatone-Veth Cengiz Pehlevan

초록

데이터셋의 서로 다른 비중첩 부분집합으로 훈련된 확산 모델들은 동일한 노이즈 시드를 입력받을 때 종종 매우 유사한 출력을 생성한다. 우리는 이러한 일관성이 간단한 선형 효과에서 비롯됨을 추적한다. 즉, 분할 간 공유된 가우시안 통계가 이미 생성 이미지의 상당 부분을 예측한다는 것이다. 이를 공식화하기 위해, 우리는 유한 데이터셋이 선형 설정에서 학습된 디노이저와 샘플링 맵의 기대값 및 분산을 어떻게 형성하는지 정량화하는 랜덤 행렬 이론(RMT) 프레임워크를 개발한다. 기대값의 경우, 샘플링 변동성은 자기 일관적 관계 σ2 7→ κ(σ2)를 통해 노이즈 수준의 재규격화(renormalization)로 작용하여, 제한된 데이터가 저분산 방향을 과도하게 축소시키고 샘플을 데이터셋 평균으로 끌어당기는 이유를 설명한다. 변동의 경우, 우리의 분산 공식은 분할 간 불일치의 세 가지 주요 요인, 즉 고유 모드 간 비등방성(anisotropy), 입력 간 비균질성(inhomogeneity), 그리고 데이터셋 크기에 따른 전반적인 스케일링을 밝힌다. 결정론적 동등성(deterministic–equivalence) 도구를 분수 행렬 거듭제곱으로 확장함으로써 전체 샘플링 궤적을 분석할 수 있게 된다. 이 이론은 선형 확산 모델의 거동을 날카롭게 예측하며, 우리는 비기억화(non-memorization) 영역에서 UNet 및 DiT 아키텍처에 대해 그 예측을 검증하여 훈련 데이터 분할에 따라 샘플이 어디에서, 어떻게 벗어나는지 식별한다. 이는 확산 훈련의 재현성을 위한 원칙적인 기준선을 제공하며, 데이터의 스펙트럼 특성을 샘플링 안정성과 연결한다.

One-sentence Summary

The authors develop a random matrix theory framework that attributes the consistency of diffusion models trained on disjoint data subsets to shared Gaussian statistics, uncovering a self-consistent noise renormalization σ2κ(σ2)\sigma^2 \mapsto \kappa(\sigma^2)σ2κ(σ2) that overshrinks low-variance directions and deriving variance formulas that predict cross-split disagreement by extending deterministic-equivalence tools to fractional matrix powers, with validation on UNet and DiT architectures.

Key Contributions

  • Diffusion models trained on nonoverlapping data splits produce strikingly similar outputs given the same noise seed; this consistency is largely explained by shared Gaussian statistics that already predict much of the generated images.
  • A random matrix theory framework quantifies how finite datasets shape the expectation and variance of the linear denoiser and sampling map. It reveals that sampling variability renormalizes the noise level to overshrink low-variance directions, and that cross-split variance decomposes into anisotropy across eigenmodes, inhomogeneity across inputs, and scaling with dataset size.
  • Deterministic-equivalence tools are extended to fractional matrix powers to derive closed-form predictions for entire sampling trajectories. The theory is validated on UNet and DiT architectures in their non-memorization regime, identifying where and how sample deviations emerge across data splits.

Introduction

Diffusion models exhibit a striking consistency across training runs: when trained on the same data distribution but with non-overlapping splits, different architectures, or repeated initializations, they often map the same noise seed to highly similar outputs. This contrasts with other generative frameworks like GANs and VAEs, where latent spaces are rotationally ambiguous. The phenomenon suggests that diffusion models recover universal statistical structure of the data manifold, raising fundamental questions about generalization and memorization. Prior work lacked a rigorous theoretical explanation for this consistency. The authors address this gap by leveraging random matrix theory (RMT). They show that a linear denoiser already predicts cross-split agreement, and that finite-sample variability enters through a renormalized noise scale that explains overshrinkage of low-variance modes. The analysis derives a variance law that factors deviation into anisotropic eigenmode contributions, input-dependent inhomogeneity, and a global scaling with dataset size. By extending deterministic equivalence to fractional matrix powers, the authors enable analysis of full sampling trajectories, and they validate that these RMT principles qualitatively govern consistency in deep CNN and DiT models beyond the linear regime.

Method

The authors leverage random matrix theory toanalyze the consistency of diffusion models trained on independent data splits. The core technical tool is deterministic equivalence, which allows the empirical covariance matrix to be replaced by a deterministic surrogate in the large-dimensional limit. This approach leads to a self-consistent equation for a renormalized noise scale κ(λ)\kappa(\lambda)κ(λ), where the stochastic effects of the sample covariance are absorbed into a scalar, leaving the population covariance unchanged.

In terms of expectation, the authors demonstrate that finite data effectively renormalize the noise scale in the population denoiser. This mechanism acts as an adaptive Ridge penalty, causing the finite-sample denoiser to shrink low-variance directions more aggressively toward the dataset mean. When extending this analysis to the full diffusion sampling map, which involves fractional powers of covariances, the authors utilize an integral representation to derive the deterministic equivalence. The resulting expectation reveals a systematic overshrinkage toward the dataset mean along lower eigenmodes, thereby reducing the generated variance in these directions.

To understand the fluctuation and consistency of the denoiser across different dataset realizations, the authors decompose the variance into three interpretable components: anisotropy, inhomogeneity, and global scaling with dataset size. The anisotropy factor shows that uncertainty is maximized along eigenmodes whose variance matches the renormalized noise. The inhomogeneity factor indicates that uncertainty is amplified for inputs displaced along high-variance modes.

The authors validate these theoretical predictions through numerical simulations of the linear diffusion sampling map. As illustrated in the figure, the expected scaling along eigenmodes confirms the overshrinkage effect on lower eigenmodes compared to the ideal scaling. The cross-split mean squared error demonstrates the anisotropy of consistency, showing larger deviations along the top eigenspaces, and the inhomogeneity effect, where samples displaced along high-variance modes exhibit greater disagreement. Furthermore, the decomposition of the mean squared error across eigenbands reveals that lower-variance modes require substantially larger dataset sizes before the cross-split deviation decays, highlighting that fine details need more data to achieve consistency across training splits.

Experiment

Diffusion models trained on independent data splits yield near-identical samples, a consistency largely captured by a linear Gaussian predictor and arising from shared covariance statistics. Finite-sample analysis shows that limited data renormalize noise scales, causing overshrinkage toward the mean and uneven denoiser agreement: uncertainty peaks along eigenmodes whose variance matches the renormalized noise and for inputs displaced along high-variance directions. These predictions extend to deep networks, where the same bias and spectral variance patterns emerge once the dataset size moves beyond the memorization regime, confirming that linear random matrix theory captures key aspects of diffusion model behavior under finite data.


AI로 AI 구축

아이디어에서 출시까지 — 무료 AI 코코딩, 즉시 사용 가능한 환경, 최적의 GPU 가격으로 AI 개발을 가속화하세요.

AI 협업 코딩
바로 사용 가능한 GPU
최적의 가격

HyperAI Newsletters

최신 정보 구독하기
한국 시간 매주 월요일 오전 9시 에 이번 주의 최신 업데이트를 메일로 발송합니다
이메일 서비스 제공: MailChimp
확산 모델의 일관성에 대한 랜덤 행렬 이론적 관점 | 문서 | HyperAI초신경