Adding Activation Functions to Our Neuron (R Edition)

Introduction

Welcome back to our course "Neural Network Fundamentals: Neurons and Layers"! In this third lesson, we're building upon what you learned in our previous lesson about the basic artificial neuron. You've already implemented a simple neuron that computes a weighted sum of inputs plus a bias. Today, we're taking an important step forward by introducing activation functions — a crucial component that enables neural networks to learn complex patterns. In particular, we'll focus on the sigmoid activation function, one of the classical functions used in neural networks. In our previous lesson, our neuron could only produce linear outputs. While this is useful for some tasks, it severely limits what our neural networks can learn. Today, we'll overcome this limitation by adding non-linearity to our neurons, allowing them to model more complex relationships in data.

The Need for Non-Linearity

\text{Layer}_1^{out} = W_1 X + b_1

Understanding Activation Functions

Join the 1M+ learners on CodeSignal

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

Enhancing Our Neuron with Activation

Now that we have our sigmoid function, let's enhance our neuron implementation from the previous lesson by adding an activation function. We'll modify the implementation to apply the sigmoid function to the weighted sum:

Neuron <- function(n_inputs) {
  # Initialize a neuron.
  # n_inputs: Number of input features.
  
  # Create a list to store neuron properties
  neuron <- list()
  
  # Initialize weights and bias
  neuron$weights <- runif(n_inputs) * 0.1  # Small random weights
  neuron$bias <- 0.0  # Start with zero bias
  
  # Define the forward method
  neuron$forward <- function(inputs) {
    # Calculate the neuron's output:
    # 1. Weighted sum of inputs and weights, plus bias.
    # 2. Apply the sigmoid activation function.
    # inputs: A vector of input features.
    
    if (length(inputs) != length(neuron$weights)) {
      stop(paste("Input size", length(inputs), "does not match neuron's",
                "expected input size", length(neuron$weights)))
    }
    
    weighted_sum <- sum(inputs * neuron$weights)  # Calculate dot product
    raw_output <- weighted_sum + neuron$bias  # Add bias
    
    # Apply activation function to introduce non-linearity
    activated_output <- sigmoid(raw_output)
    
    return(list(activated_output = activated_output, raw_output = raw_output))  # Return both for demonstration
  }
  
  return(neuron)
}

Neuron <- function(n_inputs) {
  # Initialize a neuron.
  # n_inputs: Number of input features.
  
  # Create a list to store neuron properties
  neuron <- list()
  
  # Initialize weights and bias
  neuron$weights <- runif(n_inputs) * 0.1  # Small random weights
  neuron$bias <- 0.0  # Start with zero bias
  
  # Define the forward method
  neuron$forward <- function(inputs) {
    # Calculate the neuron's output:
    # 1. Weighted sum of inputs and weights, plus bias.
    # 2. Apply the sigmoid activation function.
    # inputs: A vector of input features.
    
    if (length(inputs) != length(neuron$weights)) {
      stop(paste("Input size", length(inputs), "does not match neuron's",
                "expected input size", length(neuron$weights)))
    }
    
    weighted_sum <- sum(inputs * neuron$weights)  # Calculate dot product
    raw_output <- weighted_sum + neuron$bias  # Add bias
    
    # Apply activation function to introduce non-linearity
    activated_output <- sigmoid(raw_output)
    
    return(list(activated_output = activated_output, raw_output = raw_output))  # Return both for demonstration
  }
  
  return(neuron)
}

The key changes in this updated version are:

We still calculate the weighted sum and add the bias, but now we call this intermediate result raw_output.
We apply the sigmoid function to the raw output, producing an activated_output.
We return both the activated output and the raw output as a list, which will help us understand the effect of the activation function.

This two-step process — first calculating the linear combination and then applying the non-linear activation — is fundamental to how neurons operate in neural networks.

Testing the Activated Neuron

Now let's test our enhanced neuron to see how the activation function affects its output:

# Initialize a neuron with 3 input features
num_input_features <- 3
neuron <- Neuron(num_input_features)
cat("Neuron weights:", paste(neuron$weights, collapse = ", "), 
    ", Bias:", neuron$bias, "\n")

# Create sample input and process it through the neuron
sample_inputs <- c(1.0, 2.0, 3.0)
cat("\nInput to neuron:", paste(sample_inputs, collapse = ", "), "\n")

# Get both raw and activated outputs
outputs <- neuron$forward(sample_inputs)
cat("Neuron's raw output (weighted sum + bias):", 
    sprintf("%.4f", outputs$raw_output), "\n")
cat("Neuron's activated output (Sigmoid):", 
    sprintf("%.4f", outputs$activated_output), "\n")

# Initialize a neuron with 3 input features
num_input_features <- 3
neuron <- Neuron(num_input_features)
cat("Neuron weights:", paste(neuron$weights, collapse = ", "), 
    ", Bias:", neuron$bias, "\n")

# Create sample input and process it through the neuron
sample_inputs <- c(1.0, 2.0, 3.0)
cat("\nInput to neuron:", paste(sample_inputs, collapse = ", "), "\n")

# Get both raw and activated outputs
outputs <- neuron$forward(sample_inputs)
cat("Neuron's raw output (weighted sum + bias):", 
    sprintf("%.4f", outputs$raw_output), "\n")
cat("Neuron's activated output (Sigmoid):", 
    sprintf("%.4f", outputs$activated_output), "\n")

When we run this code, we'll observe:

The random weights and bias of our initialized neuron.
The input values we're feeding into the neuron.
The raw output (weighted sum + bias) before applying the activation function.
The final activated output after applying the sigmoid function.

The key insight here is seeing how the sigmoid function transforms the raw output. If the raw output is a large positive number, the sigmoid output will be close to 1. If the raw output is a large negative number, the sigmoid output will be close to 0. And if the raw output is around 0, the sigmoid output will be around 0.5.

This non-linear transformation is what enables neural networks to model complex patterns and make decisions. The sigmoid function essentially "squashes" the raw output into a range between 0 and 1, which can be interpreted as a probability or a level of activation.

# Test with a few values cat("sigmoid(0) =", sigmoid(0), "\n") # Should be 0.5 cat("sigmoid(2) =", sprintf("%.4f", sigmoid(2)), "\n") # Should be around 0.88 cat("sigmoid(-2) =", sprintf("%.4f", sigmoid(-2)), "\n") # Should be around 0.12 cat("sigmoid(c(-2, 0, 2)) =", paste(sigmoid(c(-2, 0, 2)), collapse = " "), "\n")

Introduction

The Need for Non-Linearity

Understanding Activation Functions

Join the 1M+ learners on CodeSignal

The Sigmoid Activation Function

Implementing the Sigmoid Function

Enhancing Our Neuron with Activation

Testing the Activated Neuron

Conclusion and Next Steps