Ridge Regression as Augmented OLS: The Data-Augmentation Trick
Recall that the ridge regression estimator for minimizes
Problem: Show that this is exactly equivalent to performing ordinary least squares (no penalty at all) on an augmented dataset , where
That is, show that .
Bonus reflection: What does this say conceptually about what ridge regression is doing to the data?
Answer: Ridge Regression as Augmented OLS
Key Idea / Intuition
Ridge regression penalizes large coefficients by adding . The augmentation trick makes this penalty literal: we append fake observations with input (the -th standard basis vector) and response . Predicting for these fake points forces the model to keep small — otherwise it pays a residual cost. The penalty term in ridge is thus reinterpreted as a genuine least-squares fit to artificial "zero-response" data.
Formal Proof / Solution
Step 1: Write the augmented OLS objective.
The OLS loss on the augmented data is
Expanding the block structure:
This is exactly the ridge objective .
Step 2: Compute the OLS normal equations for .
Step 3: Solve.
The OLS solution on the augmented data is
(Note: is always positive definite for , so the inverse exists even when is singular — a secondary benefit of ridge.)
Conceptual interpretation:
The fake data points each "observe" a single coefficient in isolation and expect it to be zero. Adding more such points (larger ) increases the pressure toward zero. Ridge shrinkage is literally the influence of these phantom zero-observations competing with the real data. This is an instance of the hints framework (Abu-Mostafa 1995): encode prior knowledge (here: "prefer small coefficients") as artificial training examples.
Source: The Elements of Statistical Learning, 2nd ed., Hastie, Tibshirani, Friedman — Ex. 3.12