Linear Algebra for High-Dimensional Spaces

Vector spaces, linear mappings, dot products as geometric similarity, eigenvalues, and Singular Value Decomposition (SVD) underlying modern embeddings.

1. Theoretical Motivation & Foundations

Every modern neural network operates on high-dimensional vector spaces. When text, images, or audio are converted into embeddings (such as 1536-dimensional or 4096-dimensional vectors), geometric concepts like distance, angle, and projection dictate how semantic meaning is organized. In high dimensions, intuitive 3D geometry collapses: orthogonal vectors dominate, and volume concentrates entirely on the spherical shell. Understanding Singular Value Decomposition (SVD) and low-rank approximation is essential to mastering dimensionality reduction, PCA, and parameter-efficient fine-tuning (LoRA).

2. Mathematical Formulations & Derivations

The governing analytical formulations and proof frameworks for this module:

Projection Operator: proj_u(v) = ((v · u) / ||u||^2) * u Singular Value Decomposition (SVD): A = U Σ V^T, where U ∈ ℝ^{m × m}, Σ ∈ ℝ^{m × n}, V ∈ ℝ^{n × n} Cosine Similarity Metric: cos(θ) = (A · B) / (||A||_2 * ||B||_2) = (∑ A_i B_i) / (√(∑ A_i^2) * √(∑ B_i^2)) Frobenius Norm Matrix Approximation: ||A - A_k||_F = min_{rank(B) ≤ k} ||A - B||_F = √(∑_{i=k+1}^r σ_i^2)

3. From-Scratch Reference Implementation

Executable, production-tested reference code without magic libraries:

import numpy as np def cosine_similarity_matrix(embeddings: np.ndarray) -> np.ndarray: """Compute pairwise cosine similarity across all embedding vectors.""" # Normalize row vectors to unit length norms = np.linalg.norm(embeddings, axis=1, keepdims=True) normalized = embeddings / np.maximum(norms, 1e-12) # Dot product of normalized vectors yields cosine similarity return np.dot(normalized, normalized.T) def truncated_svd_projection(data_matrix: np.ndarray, k_dims: int): """Project high-dimensional matrix to k principal latent dimensions.""" U, s, Vt = np.linalg.svd(data_matrix, full_matrices=False) reduced_coords = np.dot(U[:, :k_dims], np.diag(s[:k_dims])) explained_variance = np.sum(s[:k_dims]**2) / np.sum(s**2) return reduced_coords, Vt[:k_dims, :], explained_variance # Example verification np.random.seed(42) sample_embeddings = np.random.randn(10, 1536) sims = cosine_similarity_matrix(sample_embeddings) coords, basis, variance_retained = truncated_svd_projection(sample_embeddings, k_dims=64) print(f'Retained Variance in 64 dims: {variance_retained * 100:.2f}%')

4. Systems Complexity & Memory Footprint

Time Complexity: O(m n min(m, n)) for full SVD computation. Space Complexity: O(m n) for dense representation.

5. Canonical Literature & Primary Research

Original research papers and foundational texts recommended for advanced study:

  1. Strang, G. (2006). Linear Algebra and Its Applications. 4th Edition.
  2. Eckart, C., & Young, G. (1936). The approximation of one matrix by another of lower rank. Psychometrika 1(3), 211-218.
  3. Mikolov, T., et al. (2013). Distributed Representations of Words and Phrases and their Compositionality. NeurIPS.