Machine learning models can be categorized into different types based on their learning approach, function, and application. Below is a comprehensive list of ML models:
1. Supervised Learning Models
These models learn from labeled data.
Regression Models (Predict Continuous Values)
- Linear Regression
- Polynomial Regression
- Ridge Regression
- Lasso Regression
- ElasticNet Regression
- Bayesian Regression
- Quantile Regression
- Support Vector Regression (SVR)
- Decision Tree Regression
- Random Forest Regression
- Gradient Boosting Regression (GBR)
- AdaBoost Regression
- XGBoost Regression
- LightGBM Regression
- CatBoost Regression
- Gaussian Process Regression (GPR)
- Huber Regression
- Theil-Sen Estimator Regression
- Tweedie Regression
Classification Models (Predict Discrete Classes)
- Logistic Regression
- Na?ve Bayes (Gaussian, Multinomial, Bernoulli)
- K-Nearest Neighbors (KNN)
- Support Vector Machines (SVM) - Linear & Non-linear
- Decision Tree Classifier
- Random Forest Classifier
- Gradient Boosting Classifier
- AdaBoost Classifier
- XGBoost Classifier
- LightGBM Classifier
- CatBoost Classifier
- Bagging Classifier
- Extra Trees Classifier
- Quadratic Discriminant Analysis (QDA)
- Linear Discriminant Analysis (LDA)
- Perceptron
- Ridge Classifier
- Passive Aggressive Classifier
2. Unsupervised Learning Models
These models learn patterns from unlabeled data.
Clustering Models
- K-Means Clustering
- Hierarchical Clustering (Agglomerative & Divisive)
- DBSCAN (Density-Based Spatial Clustering)
- OPTICS (Ordering Points to Identify Clustering Structure)
- Mean Shift Clustering
- Gaussian Mixture Model (GMM)
- BIRCH (Balanced Iterative Reducing and Clustering using Hierarchies)
- Affinity Propagation
- Spectral Clustering
Dimensionality Reduction Models
- Principal Component Analysis (PCA)
- Kernel PCA
- Incremental PCA
- Truncated SVD (Singular Value Decomposition)
- t-SNE (t-Distributed Stochastic Neighbor Embedding)
- UMAP (Uniform Manifold Approximation and Projection)
- LDA (Latent Dirichlet Allocation)
- Factor Analysis
- Independent Component Analysis (ICA)
- Autoencoders (for Representation Learning)
Anomaly Detection Models
- Isolation Forest
- Local Outlier Factor (LOF)
- One-Class SVM
- Elliptic Envelope
- Autoencoder-based Anomaly Detection
3. Semi-Supervised Learning Models
These models use a small amount of labeled data with a large amount of unlabeled data.
- Self-training Models
- Label Propagation
- Label Spreading
- Graph-Based Semi-Supervised Learning
4. Reinforcement Learning Models
These models learn from interaction with an environment.
- Markov Decision Process (MDP)
- Q-Learning
- Deep Q Networks (DQN)
- SARSA (State-Action-Reward-State-Action)
- Actor-Critic Methods
- Policy Gradient Methods
- Proximal Policy Optimization (PPO)
- Trust Region Policy Optimization (TRPO)
- Deep Deterministic Policy Gradient (DDPG)
- Twin Delayed DDPG (TD3)
- Soft Actor-Critic (SAC)
- Monte Carlo Control
- Evolutionary Strategies
5. Deep Learning Models
These models use neural networks for feature extraction and learning.
Feedforward Neural Networks (FNN)
- Multi-Layer Perceptron (MLP)
Convolutional Neural Networks (CNN)
- LeNet
- AlexNet
- VGGNet
- GoogLeNet (Inception Networks)
- ResNet (Residual Networks)
- DenseNet
- EfficientNet
- MobileNet
- Vision Transformers (ViTs)
Recurrent Neural Networks (RNN)
- Simple RNN
- Long Short-Term Memory (LSTM)
- Gated Recurrent Unit (GRU)
- Bidirectional LSTM/GRU
Transformer-Based Models
- Transformer (Original by Vaswani et al.)
- BERT (Bidirectional Encoder Representations from Transformers)
- GPT (Generative Pre-trained Transformer, GPT-1, GPT-2, GPT-3, GPT-4, etc.)
- T5 (Text-to-Text Transfer Transformer)
- XLNet
- RoBERTa
- ALBERT
- DistilBERT
- BART (Bidirectional and Auto-Regressive Transformers)
- Whisper (Speech-to-Text by OpenAI)
Generative Models
- Autoencoders (Vanilla, Variational, Denoising)
- Generative Adversarial Networks (GANs) Vanilla GAN Deep Convolutional GAN (DCGAN) Conditional GAN (cGAN) StyleGAN CycleGAN Pix2Pix
- Normalizing Flows
- Diffusion Models (DALL-E, Stable Diffusion, Imagen, etc.)
Graph Neural Networks (GNN)
- Graph Convolutional Networks (GCN)
- Graph Attention Networks (GAT)
- GraphSAGE
- ChebNet (Chebyshev Graph CNNs)
- GNN Explainers
Self-Supervised Learning Models
- SimCLR (Simple Framework for Contrastive Learning)
- BYOL (Bootstrap Your Own Latent)
- MoCo (Momentum Contrast)
- DINO (Self-Supervised Transformers)
- MAE (Masked Autoencoders for Vision Tasks)
6. Hybrid Models & Meta-Learning
- Stacking Models
- Blending Models
- Bayesian Optimization-Based Learning
- Neural Architecture Search (NAS)
- Few-Shot Learning Models (Siamese Networks, Prototypical Networks, Matching Networks, etc.)
- Federated Learning Models