Skip to main content
Linear Algebra is one of the foundations of Artificial Intelligence and Machine Learning because datasets, images, embeddings, and neural network calculations are represented using vectors and matrices.

Why Linear Algebra in AI/ML?

Linear Algebra is used in:
  • Linear Regression
  • Neural Networks
  • Deep Learning
  • Computer Vision
  • Recommendation Systems
  • NLP Embeddings
  • PCA (Principal Component Analysis)
  • Clustering
  • Search and Similarity Systems
Example: House prediction dataset: Represented as:
Rows → data points
Columns → features

1. Vectors

What is a Vector?

A vector is a one-dimensional array representing a data point. Example:
Output:

AI/ML Usage

Vectors represent:
  • User embeddings
  • Text embeddings
  • Image pixels
  • Features
Example: House features:

Dot Product

What it does

Multiplies corresponding elements and sums them. Formula:
Code:
Output:
Calculation:

AI/ML Usage

Used in:
  • Neural networks
  • Linear regression
  • Recommendation systems
  • Similarity search
Prediction equation:

Vector Norms

What it does

Measures vector magnitude (length).

L2 Norm (Euclidean Distance)

Formula:
Code:
Output:
Calculation:

L1 Norm (Manhattan Distance)

Formula:
Code:
Output:
Calculation:

AI/ML Usage

Used in:
  • Regularization
  • Clustering
  • Similarity measurement
  • Feature selection

Cosine Similarity

What it does

Measures similarity between vectors using angle instead of distance. Formula:
Code:
Output:
Interpretation:

AI/ML Usage

Used heavily in:
  • ChatGPT embeddings
  • Semantic search
  • Recommendation systems
  • NLP
Example:
Embeddings close together → high cosine similarity

2. Matrices

What is Matrix?

Collection of vectors arranged in rows and columns. Example:
Output:

AI/ML Usage

Represents:
  • Datasets
  • Images
  • Neural network weights

Matrix Multiplication

What it does

Rows multiply columns. Condition:
Code:
Output:
Calculation:

AI/ML Usage

Used in:
  • Neural networks
  • Attention mechanisms
  • Transformations

Matrix Transpose

What it does

Converts rows into columns. Code:
Output:

AI/ML Usage

Used in:
  • Linear regression
  • PCA
  • Matrix multiplication

Matrix Inverse

What it does

Finds matrix that reverses transformation. Condition:
Code:
Output:
Verification:
Output:

AI/ML Usage

Used in:
  • Solving equations
  • Linear Regression

Matrix Rank

What it does

Determines how many independent rows or columns exist. Code:
Output:
Interpretation:

AI/ML Usage

Used in:
  • Removing redundant features
  • Detecting multicollinearity

Eigenvalues and Eigenvectors

What it does

Finds:
  • Stretch amount → Eigenvalue
  • Stretch direction → Eigenvector
Formula:
Code:
Output:

AI/ML Usage

Used in:
  • PCA
  • Face recognition
  • Dimensionality reduction
Intuition: Imagine stretching a rubber sheet:
  • Eigenvector → direction
  • Eigenvalue → stretch amount

Singular Value Decomposition (SVD)

What it does

Breaks matrix into three matrices. Formula:
Where:
Code:
Output:

AI/ML Usage

Used in:
  • Recommendation systems
  • Image compression
  • NLP
  • PCA

PCA (Principal Component Analysis) Intuition

What it does

Reduces dimensions while keeping maximum information. Example: Dataset:
Maybe:
contain similar information. PCA converts:
into:
using eigenvectors. Steps:

AI/ML Usage

Used in:
  • Data compression
  • Noise reduction
  • Visualization
  • Faster training

Quick Revision Notes

Vector

Dot Product

Norm

Cosine Similarity

Matrix

Transpose

Inverse

Rank

Eigenvalues

Eigenvectors

SVD

PCA


Applications