Custom Training Loops

Customize deep learning training loops and loss functions for image networks

If the trainingOptions function does not provide the training options that you need for your task, or you have a loss function that the trainnet function does not support, then you can define a custom training loop. For models that cannot be specified as networks of layers, you can define the model as a function. To learn more, see Define Custom Training Loops, Loss Functions, and Networks.

Functions

expand all

Custom Training Loops

`dlnetwork`	Deep learning neural network
`trainingProgressMonitor`	Monitor and plot training progress for deep learning custom training loops (Since R2022b)
`minibatchqueue`	Create mini-batches for deep learning (Since R2020b)
`dlarray`	Deep learning array for customization
`dlgradient`	Compute gradients for custom training loops using automatic differentiation
`dljacobian`	Jacobian matrix deep learning operation (Since R2024b)
`dldivergence`	Divergence of deep learning data (Since R2024b)
`dllaplacian`	Laplacian of deep learning data (Since R2024b)
`dlfeval`	Evaluate deep learning model for custom training loops

Loss Operations

`crossentropy`	Cross-entropy loss for classification tasks
`indexcrossentropy`	Index cross-entropy loss for classification tasks (Since R2024b)
`l1loss`	L₁ loss for regression tasks (Since R2021b)
`l2loss`	L₂ loss for regression tasks (Since R2021b)
`huber`	Huber loss for regression tasks (Since R2021a)
`mse`	Half mean squared error

Deep Learning Operations

`dlconv`	Deep learning convolution
`dltranspconv`	Deep learning transposed convolution
`fullyconnect`	Sum all weighted input data and apply a bias
`batchnorm`	Normalize data across all observations for each channel independently
`crosschannelnorm`	Cross channel square-normalize using local responses (Since R2020a)
`groupnorm`	Normalize data across grouped subsets of channels for each observation independently (Since R2020b)
`instancenorm`	Normalize across each channel for each observation independently (Since R2021a)
`layernorm`	Normalize data across all channels for each observation independently (Since R2021a)
`avgpool`	Pool data to average values over spatial dimensions
`maxpool`	Pool data to maximum value
`maxunpool`	Unpool the output of a maximum pooling operation

Activations

`relu`	Apply rectified linear unit activation
`leakyrelu`	Apply leaky rectified linear unit activation
`gelu`	Apply Gaussian error linear unit (GELU) activation (Since R2022b)
`softmax`	Apply softmax activation to channel dimension
`sigmoid`	Apply sigmoid activation

Topics

Custom Training Loops

Train Deep Learning Model in MATLAB
Learn how to training deep learning models in MATLAB^®.
Define Custom Training Loops, Loss Functions, and Networks
Learn how to define and customize deep learning training loops, loss functions, and models.
Train Network Using Custom Training Loop
This example shows how to train a network that classifies handwritten digits with a custom learning rate schedule.
Monitor Custom Training Loop Progress
Track and plot custom training loop progress.
Train Network with Multiple Outputs
This example shows how to train a deep learning network with multiple outputs that predict both labels and angles of rotations of handwritten digits.
Classify Videos Using Deep Learning with Custom Training Loop
This example shows how to create a network for video classification by combining a pretrained image classification model and a sequence classification network.
Train Image Classification Network Robust to Adversarial Examples
This example shows how to train a neural network that is robust to adversarial examples using fast gradient sign method (FGSM) adversarial training.
Speed Up Deep Neural Network Training
Learn how to accelerate deep neural network training.