October DealsAmazon USOctober deal check: compare before you payAmazon US: current deals, useful picks and tech finds.Check DealsWindows FixRecommendedWindows errors stealing your time? Find the fix fastScan stability, cleanup and performance issues.Fix NowOctober DealsAmazon USDeal season is back - check today's better picksAmazon US: current deals, useful picks and tech finds.See Picks×
Skip to content
World desk3 min

How Discriminative Fine-Tuning Sets Learning Rates Across a Model

Discriminative fine-tuning assigns different learning rates across model layers. Learn why backbone and head rates may differ and how to test them for your task.
Special offer. See more information about Outbyte and uninstall instructions. Please review EULA and Privacy policy.

Yes, you can use different learning rates for a pretrained model’s backbone and its task-specific head. This approach—discriminative fine-tuning—lets the head adapt to new labels while changing pretrained layers at a different pace. It is a strategy to test, not a rule that the head must always learn faster: the right rates depend on the model, data, initialization, and training schedule.

What discriminative fine-tuning changes

A learning rate controls the scale of an optimizer’s parameter updates. Standard fine-tuning often assigns one rate to all trainable parameters. Discriminative fine-tuning instead assigns different rates to different layers or parameter groups.

As an Amazon Associate I earn from qualifying purchases.

The backbone is the pretrained feature extractor: its layers produce representations learned before the current task. The head is the task-specific output portion, such as a classifier that maps those representations to target labels. Because a newly added or adapted head may start with less task-specific information than the backbone, it can make sense to give the two different update scales.

Special offer. See more information about Outbyte and uninstall instructions. Please review EULA and Privacy policy.

Howard and Ruder define the approach this way: “Instead of using the same learning rate for all layers of the model, discriminative fine-tuning allows us to tune each layer with different learning rates.” Their ACL 2018 paper applies it to language-model fine-tuning for text classification.

#1 Best Overall
Sale
HPE NVIDIA Tesla V100 32GB HBM2 PCIe 3.0 x16 Passive GPU Computational Accelerator for AI Machine Learning HPC Deep Learning 699-2G500-0216-400 (Renewed)
  • NVIDIA Volta GV100 Architecture — 4,608 CUDA Cores, 640 1st-Gen Tensor Cores delivering 14 TFLOPS FP32 and 112 TFLOPS deep learning performance for AI training, inference, HPC, and scientific computing workloads
  • 32GB HBM2 ECC Memory — 900 GB/s Bandwidth — High-bandwidth memory on a 4096-bit bus with ECC error correction provides the memory capacity and throughput required for the largest AI models, simulations, and datasets
  • PCIe 3.0 x16 Interface — 250W TDP — Standard PCIe Gen3 connectivity with passive cooling designed for enterprise rack server deployment in HPE ProLiant, Dell PowerEdge, and Supermicro platforms with adequate chassis airflow
  • NVLink — Scale to 96GB Unified Memory — Connect two V100 GPUs via NVLink at 300 GB/s bi-directional bandwidth to scale GPU memory from 32GB to 96GB for larger AI training and HPC workloads
  • Multi-Precision Computing — Supports FP64 (7 TFLOPS), FP32 (14 TFLOPS), FP16 (112 TFLOPS) and INT8 precision modes for flexible deployment across training, inference, and scientific simulation workloads

Why the backbone and head may need different rates

The backbone already contains learned representations; large updates can alter them substantially. The head, by contrast, must map those representations to the new task’s outputs. Different rates provide a way to control these two kinds of adaptation separately.

This is an intuition for choosing a starting experiment, not a guarantee about which component should receive the larger rate. A well-initialized or previously task-adapted head may not need a high rate. A backbone facing a substantial domain shift may need more adaptation than a small, similar target dataset would suggest. The useful question is which rate assignment validates well on the target task, not whether one ordering is universally correct.

What the ULMFiT recipe did—and did not establish

In their ULMFiT experiments, Howard and Ruder first selected a rate for the last layer, then set each lower layer’s rate to the rate of the layer above divided by 2.6. This creates a progressively smaller rate farther down the model. It is a specific empirical recipe from that paper, not a default ratio for every architecture or modern fine-tuning task. The paper PDF describes the method.

Special offer. See more information about Outbyte and uninstall instructions. Please review EULA and Privacy policy.

The ACL Anthology record reports error reductions of 18–24% on the majority of six text-classification datasets. That is the reported result for ULMFiT’s approach in those experiments; it is not an isolated estimate of the effect of discriminative learning rates alone, nor a promised improvement for other tasks. See the ACL Anthology record.

Keep learning rates separate from freezing choices

Freezing and discriminative learning rates answer different questions:

  • Freezing prevents selected parameters from updating.
  • Fine-tuning allows selected parameters to update.
  • Discriminative rates set different update scales for trainable parameter groups.
  • Gradual unfreezing changes which layers are trainable over the course of training.

These choices can be combined. For example, a training procedure can initially freeze some layers, unfreeze them in stages, and assign different rates once they are trainable. But a layer that is frozen does not receive updates, regardless of its assigned learning rate.

Independent reader supportYour contribution helps us test, update, and keep practical guides available for everyone.Support on Ko-Fi

How to compare rate and unfreezing strategies

Compare alternatives on the same model and target-task data so the result reflects the training choices rather than a changed evaluation setup. Keep the data split, optimizer, schedule, and evaluation metric consistent where possible. Record:

Special offer. See more information about Outbyte and uninstall instructions. Please review EULA and Privacy policy.
  • Which parameters are trainable at each stage.
  • How rates are assigned, including any head-to-backbone ratio or per-layer decay.
  • Whether layers are unfrozen all at once or gradually.
  • Validation performance and whether it is stable across training.
  • Compute requirements and how much target-task data is available.

Do not assume gradual unfreezing will compensate for a single shared rate. An ICLR 2024 study reports that gradual unfreezing with a single rate or cosine schedule was insufficient in its experimental settings. That finding is evidence about those settings, not a universal verdict on the schedules.

A practical decision rule

Start by deciding which layers should update at all. If the head is new and the backbone is already useful for the target task, testing a faster head rate and more cautious backbone rates is a reasonable starting point. If the task or domain differs substantially from the pretraining data, test alternatives that allow more backbone adaptation too. Where compute permits, compare those choices against a shared-rate baseline using held-out target-task validation.

For background on the relationship between discriminative fine-tuning and progressive unfreezing, see Sebastian Ruder’s overview of transfer learning in NLP.

Product prices and availability are accurate as of the date/time indicated and are subject to change. Any price and availability information displayed on Amazon at the time of purchase will apply.

Free tools Windows power users keep installed

One-click scans. No signup required.

Special offer. See more information about Outbyte and uninstall instructions. Please review EULA and Privacy policy.

Leave a Reply

Your email address will not be published. Required fields are marked *

Special offer. See more information about Outbyte and uninstall instructions. Please review EULA and Privacy policy.

More from the Wire

  1. World desk4 min
    How to Spot an AI Voice Scam Before Sending MoneyDon’t rely on how a caller sounds. Pause, call back through a known number, and verify the emergency with another trusted person before sending money.
  2. Mountain View desk4 min
    Google’s SynthID Detector: How to Check AI-Generated Images, Video and AudioGoogle’s SynthID Detector looks for an embedded watermark in supported images, video and audio. Here is what its results do—and do not—show.
  3. Redmond desk20 min
    How to create a link to File or Folder in Windows 11Windows 11 gives you several ways to point to a file or folder without moving or duplicating it. You can create a desktop shortcut,…
Recommended PC Tool
Recommended PC Tool
PC Slower Than It Used to Be?Free scan - under a minute
Crashes, No Sound, or Screen Glitches?Free driver scan

Two free Windows tools

One Free Minute Could Fix That PC

Before you go - each of these free tools takes about a minute and tackles what quietly slows a Windows PC down.

Special offer. View Outbyte info, uninstall instructions, EULA, and Privacy Policy.