From Neurons to Layers: Implementing a Dense Layer in JavaScript

Introduction

Welcome back to our course "Neural Network Fundamentals: Neurons and Layers"! You've made excellent progress so far. In the previous lessons, we built a single artificial neuron and then enhanced it with the sigmoid activation function to introduce nonlinearity. Today, we're taking a significant step forward in our neural networks journey. Rather than working with individual neurons, we'll learn how to group neurons together into layers — the fundamental building blocks of neural network architectures. Specifically, we'll implement a Dense Layer (also called a fully connected layer), which is one of the most common types of layers in neural networks. By the end of this lesson, we'll have built a layer that can process multiple inputs through multiple neurons simultaneously, bringing us closer to implementing a complete neural network!

From Neurons to Layers

While a single neuron, as we've built, performs a basic computation, real-world problems demand more processing power. This is where layers come into play. A layer is essentially a group of neurons working in parallel, with each neuron in the layer processing the same input data independently. For instance, if a single neuron with 3 inputs produces 1 output, a layer of 5 such neurons, each receiving those same 3 inputs, would collectively produce 5 outputs. This layered approach offers significant advantages: Increased Computational Power : Multiple neurons can learn diverse patterns from the data. Parallelism : All neurons in a layer compute their outputs simultaneously. Efficiency : Enables the use of vectorized operations (like matrix math) for faster computations. Hierarchical Learning : When layers are stacked, the network can learn increasingly complex features from the input. This organization, inspired by how our brains process information, allows us to build more powerful and expressive neural network models.

Understanding Dense Layers

One of the most fundamental and common types of layers is the Dense Layer, also known as a fully connected layer. Its defining characteristic is that each neuron in the layer receives input from all features of the previous layer (or the initial input data, if it's the first layer). This "full" connectivity gives it its name. Key aspects of a dense layer include: Full Connectivity : Every input feature is connected to every neuron within the layer. Unique Parameters : Each of these connections has its own distinct weight, and each neuron in the layer has its own distinct bias. Shared Activation : Typically, all neurons within the same dense layer use the same activation function (like the sigmoid we implemented). To illustrate, consider a dense layer with 4 neurons that processes an input vector containing 3 features. This configuration would result in 3 (inputs) \times 4 (neurons) = 12 weight parameters and 4 bias parameters (one for each neuron in the dense layer). The layer would then produce 4 output values, one from each neuron. It's important to note that the 4-neuron layer shown in our example is not the final output layer. The 4-neuron layer produces 4 outputs, which are then fed into a final layer with 2 neurons, resulting in 2 output values. This multi-layer structure is typical for classification tasks like binary classification with 2 outputs (e.g., "cat" or "dog"). Alternatively, for binary classification, you can use just 1 output neuron with a sigmoid activation function, where values above 0.5 represent one class (e.g., "dog") and values below 0.5 represent the other class (e.g., "cat"). In practical terms, a dense layer performs a matrix multiplication between the input and a weight matrix, adds a bias vector, and then applies an activation function to these results.

Join the 1M+ learners on CodeSignal

Be a part of our community of 1M+ users who develop and demonstrate their skills on CodeSignal

Implementing the DenseLayer Class

Now that we understand the concept, let's implement our DenseLayer class in JavaScript. We'll use mathjs to handle matrix operations and random initialization. In this version, we'll store the biases as a row vector (a [1, nNeurons] matrix), which is a common convention in many neural network libraries.

JavaScript

const math = require('mathjs');

/**
 * DenseLayer class: Represents a fully connected layer of neurons.
 */
class DenseLayer {
    /**
     * Initialize a dense layer.
     * @param {number} nInputs - Number of input features.
     * @param {number} nNeurons - Number of neurons in this layer.
     */
    constructor(nInputs, nNeurons) {
        // Weights matrix: shape [nInputs, nNeurons]
        // Each column corresponds to the weights of one neuron.
        this.weights = math.multiply(math.random([nInputs, nNeurons]), 0.1);

        // Biases matrix: shape [1, nNeurons]
        // One bias for each neuron in the layer, stored as a row vector.
        this.biases = math.zeros(1, nNeurons);

        this.nInputs = nInputs;
        this.nNeurons = nNeurons;
    }
}

module.exports = { DenseLayer };

const math = require('mathjs');

/**
 * DenseLayer class: Represents a fully connected layer of neurons.
 */
class DenseLayer {
    /**
     * Initialize a dense layer.
     * @param {number} nInputs - Number of input features.
     * @param {number} nNeurons - Number of neurons in this layer.
     */
    constructor(nInputs, nNeurons) {
        // Weights matrix: shape [nInputs, nNeurons]
        // Each column corresponds to the weights of one neuron.
        this.weights = math.multiply(math.random([nInputs, nNeurons]), 0.1);

        // Biases matrix: shape [1, nNeurons]
        // One bias for each neuron in the layer, stored as a row vector.
        this.biases = math.zeros(1, nNeurons);

        this.nInputs = nInputs;
        this.nNeurons = nNeurons;
    }
}

module.exports = { DenseLayer };

Here's how the initialization process works:

The DenseLayer constructor initializes a single weight matrix and a single bias row vector. These structures collectively manage the parameters for all neurons within the layer, enabling efficient, vectorized operations.
The weights are initialized to small random values (e.g., math.random([nInputs, nNeurons]) * 0.1). This common practice helps break symmetry between neurons and is crucial for effective learning, preventing issues like all neurons learning the same features and aiding in the network's convergence during training.
The biases for all neurons in the layer are initialized to zero (e.g., math.zeros(1, nNeurons)). This provides a neutral starting point, allowing the network to learn the appropriate bias offset for each neuron based on the data during the training phase.
Notice that our layer stores information about its dimensions (nInputs and nNeurons), which will be useful for debugging and when connecting multiple layers together in the future.

Introduction

From Neurons to Layers

Understanding Dense Layers

Join the 1M+ learners on CodeSignal

Vectorized Implementation with mathjs

Implementing the DenseLayer Class

The Structure of Weights and Biases

Testing the Layer Structure

Conclusion and Next Steps