atlas

Research map / Architectures and efficiency / RNNs and normalization

Efficient Precision and Recall Metrics for Assessing Generative Models using Hubness-aware Sampling

ICML 2024 · Architectures and efficiency › RNNs and normalization · approach: Neural scaling laws

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes efficient precision and recall (eP&R) metrics for generative model evaluation that use hubness-aware sampling and approximate k-NN to approximate the original P&R at much lower time and memory cost.

Problem

Original precision and recall metrics for generative models require O(n^2 log n) time and O(n^2) space due to pairwise k-NN distances, becoming infeasible on commodity hardware for large datasets.

Approach

Identifies two redundancies in P&R (ratio computation and inside/outside manifold identification) and shows that a small set of high-hubness samples can replace the full sets, with approximate k-NN (IVF-PQ) used for hub identification since hub selection is insensitive to exact k-NN results.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use hubness-aware sampling and approximate k-NN for eP&R instead of full-sample original P&RPrecision and recall evaluation · FFHQ, LSUN-Car, LSUN-Church, LSUN-Cat, LSUN-Horse; StyleGAN3, Projected-GAN, VQ-VAE-2, Latent Diffusion on FFHQprecision and recall: Error(%): 0.4%–2.3%No clear effect
Use eP&R instead of original P&RPrecision and recall evaluation on FFHQ (70k images) · FFHQparallel runtime: 144s -> 75sHelped
Use eP&R instead of original P&RPrecision and recall evaluation on FFHQ (70k images) · FFHQpeak memory: 19.90 GB -> 14.24 GBHelped
Remove representative-sample ratio estimation (Observation 4.1) (ablation)eP&R · FFHQprecision: 0.719±0.002 -> 0.715±0.005No clear effect
Remove representative-sample ratio estimation (Observation 4.1) (ablation)eP&R · FFHQrecall: 0.501±0.001 -> 0.497±0.004No clear effect
Remove manifold inside/outside redundancy reduction (Observation 4.2) (ablation)eP&R · FFHQprecision: 0.719±0.002 -> 0.708±0.005No clear effect
Remove manifold inside/outside redundancy reduction (Observation 4.2) (ablation)eP&R · FFHQrecall: 0.501±0.001 -> 0.501±0.005No clear effect
Remove approximate k-NN for hub identification (ablation)eP&R with Observations 4.1 and 4.2 · FFHQprecision: 0.719±0.002 -> 0.719±0.002No clear effect
Remove approximate k-NN for hub identification (ablation)eP&R with Observations 4.1 and 4.2 · FFHQrecall: 0.501±0.001 -> 0.494±0.001No clear effect
Use k=5 instead of k=3 (ablation)eP&R · FFHQprecision: 0.719±0.001 -> 0.813±0.002No clear effect
Use k=5 instead of k=3 (ablation)eP&R · FFHQrecall: 0.494±0.001 -> 0.615±0.002No clear effect
Increase hub threshold from t=3 to t=5 (ablation)eP&R · FFHQprecision: 0.719±0.002 -> 0.730±0.001Mixed

Surprising findings

Limitations

Experimental setup

Datasets: FFHQ LSUN-Car LSUN-Church LSUN-Cat LSUN-Horse

Benchmarks: FFHQ LSUN-Car LSUN-Church LSUN-Cat LSUN-Horse

Metrics: precision recall relative approximation error time (s) memory (GB)

Baselines: original precision and recall (Kynkäänniemi et al., 2019) Sajjadi et al. 2018 P&R Simon et al. 2019 P&R

Builds on: Kynkäänniemi et al. 2019 improved precision and recall hubness phenomenon IVF-PQ

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy