The Intrinsic Dimension of Images and Its Impact on Learning

1. The Intrinsic Dimension of Images and Its Impact on Learning

Intrinsic dimension (ID) means the effective number of degrees of freedom needed to describe the variability of the data.

Formally: If your dataset lives in a high-dimensional ambient space 𝑅𝐷 (e.g., an image with 𝐷=150,528 pixels), but all the points lie close to a low-dimensional manifold 𝑀 of dimension 𝑑, then 𝑑 is the intrinsic dimension.

So the goal is: estimate 𝑑 directly from samples.

1.1. Estimating Intrinsic Dimension

The paper adopts a Maximum Likelihood Estimator (MLE) for local intrinsic dimension. This method uses nearest-neighbor distances.

1.1.1. Step 1: Local density assumption

They assume that around any point π‘₯𝑖, data points are locally uniformly distributed in a small ball of radius π‘Ÿ in a 𝑑-dimensional manifold. Under this assumption, the distances to the nearest neighbors follow a known statistical distribution that depends only on 𝑑. Compute nearest-neighbor distances

1.1.2. Step 2: Compute nearest-neighbor distances

For each data point π‘₯𝑖:

  1. Compute its π‘˜ nearest neighbors in the dataset.
  2. Denote 𝑇𝑗(π‘₯𝑖) as the distance from π‘₯𝑖 to its 𝑗-th nearest neighbor (sorted ascendingly). So 𝑇1(π‘₯𝑖)≀𝑇2(π‘₯𝑖)β‰€β‹―β‰€π‘‡π‘˜(π‘₯𝑖).

They typically choose π‘˜ between 5 and 20 for stability.

1.1.3. Step 3: Derive the local dimension formula

Under the local uniformity assumption, the MLE for the local intrinsic dimension at point π‘₯𝑖 is:

𝑑̂(π‘₯𝑖)=[1π‘˜βˆ’1βˆ‘π‘—=1π‘˜βˆ’1log(π‘‡π‘˜(π‘₯𝑖)𝑇𝑗(π‘₯𝑖))]βˆ’1

Let's break that down:

  • The term log(π‘‡π‘˜π‘‡π‘—) measures how fast the neighbor distances grow as you include more neighbors.
  • If the data lie in a low-dimensional manifold, neighbor distances grow faster with 𝑗; in a higher-dimensional one, they grow slower.
  • The average of those ratios (in log-space) encodes the "expansion rate" of the local neighborhood β€” which relates directly to the manifold's dimensionality.

1.1.4. Step 4: Global intrinsic dimension

They then average the local estimates across many (or all) points:

𝑑̂globalΒ =1π‘βˆ‘π‘–=1𝑁𝑑̂(π‘₯𝑖)

This gives one single number representing the dataset’s overall intrinsic dimension.

1.1.5. Step 5: Choice of π‘˜

  • π‘˜ is a sensitivity parameter.

    • If π‘˜ is too small β†’ noisy (high variance) MLE.
    • If π‘˜ is too large β†’ local uniformity assumption breaks (high bias).
  • In practice, they like π‘˜ values like (5, 10, 15, 20) and report a range.
  • For example, for ImageNet, ID "approx" 26–43 depending on π‘˜.

1.2. Findings

  • Real-world image datasets lie on surprisingly low-dimensional manifolds β€” the intrinsic dimension (ID) is tiny compared to pixel count.
  • Intrinsic dimension strongly correlates with how hard a task is to learn β€” higher ID β†’ harder learning, more samples required, weaker generalization.
  • The intrinsic dimension β€” not the raw pixel (ambient) dimension β€” determines learning complexity.

References

  1. The Intrinsic Dimension of Images and Its Impact on Learning