All challenges
Easy 0 solved 0% acceptance

Xavier and He Weight Initialization

Compute the range or standard deviation used to initialize the weights of a layer.

What you'll learn

  • Compute initialization ranges for a layer
  • Choose between Xavier and He
  • Understand why initialization matters

Problem

Weights that start too big or too small make deep networks hard to train. Two standard schemes pick the scale from the layer size, where fan_in is the number of inputs and fan_out the number of outputs:

methodreturns
"xavier_uniform"limit √(6 / (fan_in + fan_out)), weights drawn from [-limit, limit]
"xavier_normal"standard deviation √(2 / (fan_in + fan_out))
"he_uniform"limit √(6 / fan_in)
"he_normal"standard deviation √(2 / fan_in)

Xavier (Glorot) suits sigmoid and tanh layers; He suits ReLU layers.

Implement init_bound(method, fan_in, fan_out).

Input

method: One of "xavier_uniform", "xavier_normal", "he_uniform" or "he_normal".

fan_in, fan_out: The layer's number of inputs and outputs.

Output

Return the limit or standard deviation as a decimal number.

Constraints

1 ≤ fan_in, fan_out ≤ 100,000

Examples

Example 1
Input:
fan_in = 3
method = "xavier_uniform"
fan_out = 3

Expected output:
expected = 1.0
Input:
long long fan_in = 3;
std::string method = "xavier_uniform";
long long fan_out = 3;

Expected output:
double expected = 1.0;
Input:
long fan_in = 3;
string method = "xavier_uniform";
long fan_out = 3;

Expected output:
double expected = 1.0;

Explanation

√(6 / (3 + 3)) = √1 = 1, so weights are drawn from [-1, 1].

Example 2
Input:
fan_in = 50
method = "he_normal"
fan_out = 10

Expected output:
expected = 0.2
Input:
long long fan_in = 50;
std::string method = "he_normal";
long long fan_out = 10;

Expected output:
double expected = 0.2;
Input:
long fan_in = 50;
string method = "he_normal";
long fan_out = 10;

Expected output:
double expected = 0.2;

Explanation

He only uses fan_in: √(2 / 50) = √0.04 = 0.2.

References

This challenge is adapted for CodeUtility. Follow the statement and constraints above.

Learn from others

Community solutions

No published solutions yet. Solve this challenge to share the first one.