Skip to content
Tutoriales
Python
nn.Linear in PyTorch: Clearly Explained

Introducción a nn.Linear en PyTorch: Explicación clara

El aprendizaje profundo ha revolucionado el campo de la inteligencia artificial, permitiendo a las máquinas imitar la inteligencia humana de una manera sin precedentes. En el corazón de esta revolución se encuentra PyTorch, una popular biblioteca de aprendizaje automático de código abierto que proporciona dos características de alto nivel: cálculo de tensores con una fuerte aceleración de la GPU y redes neuronales profundas basadas en un sistema de autograduación basado en cinta. Uno de los componentes fundamentales de PyTorch es nn.Linear, un módulo que aplica una transformación lineal a los datos de entrada. Este artículo proporciona una guía exhaustiva para comprender nn.Linear en PyTorch, su papel en las redes neuronales y cómo se compara con otros métodos de transformación lineal.

nn.Linear es una capa lineal utilizada en redes neuronales que aplica una transformación lineal a los datos de entrada utilizando pesos y sesgos. Es un componente crítico en la arquitectura de muchos modelos de aprendizaje profundo. Esta guía profundizará en los detalles de nn.Linear, incluyendo su definición, cómo funciona y sus aplicaciones en el aprendizaje profundo. También abordaremos preguntas frecuentes y consultas relacionadas, proporcionando una comprensión completa de este módulo esencial de PyTorch.

¿Desea crear rápidamente visualizaciones de datos a partir de dataframes de Python Pandas sin escribir código?

PyGWalker es una biblioteca de Python para el análisis exploratorio de datos con visualización. PyGWalker (opens in a new tab) puede simplificar el flujo de trabajo de análisis de datos y visualización de datos en su cuaderno Jupyter, convirtiendo su dataframe de pandas (y dataframe de polars) en una interfaz de usuario de exploración visual al estilo Tableau.

PyGWalker para visualización de datos (opens in a new tab)

Comprendiendo nn.Linear en PyTorch

¿Qué es nn.Linear?

En el contexto de las redes neuronales, nn.Linear es un módulo proporcionado por PyTorch que aplica una transformación lineal a los datos de entrada. Esta transformación se representa mediante la fórmula y = xA^T + b, donde x es la entrada, A es el peso, b es el sesgo, e y es la salida.

El módulo nn.Linear toma dos parámetros: in_features y out_features, que representan el número de características de entrada y salida, respectivamente. Cuando se crea un objeto nn.Linear, inicializa aleatoriamente una matriz de pesos y un vector de sesgos. El tamaño de la matriz de pesos es out_features x in_features, y el tamaño del vector de sesgos es out_features.

import torch
from torch import nn
 
## Creando un objeto para la clase lineal
linear_layer = nn.Linear(in_features=3, out_features=1)

En el fragmento de código anterior, creamos una instancia de nn.Linear con tres características de entrada y una característica de salida. Esto resulta en una matriz de pesos de tamaño 3x1 y un vector de sesgos de tamaño 1x1.

¿Cómo funciona nn.Linear?

nn.Linear funciona realizando una multiplicación de matrices de los datos de entrada con la matriz de pesos y agregando el término de sesgo. Esta operación se aplica a cada capa en una red neuronal de alimentación directa.

## Pasando la entrada a la capa lineal
output = linear_layer(torch.tensor([1,2,3], dtype=torch.float32))
print(output)

En el fragmento de código anterior, pasamos un tensor de tamaño 3 (correspondiente al número de características de entrada) a la linear_layer. La salida es un tensor de tamaño 1 (correspondiente al número de características de salida), que es el resultado de la transformación lineal.

Inicializando pesos y sesgos

Los pesos y sesgos en nn.Linear son parámetros que el modelo aprende durante el entrenamiento. Inicialmente, se establecen en valores aleatorios. Puede ver los pesos y sesgos utilizando los atributos weight y bias.

## Para ver los pesos y sesgos
print(linear_layer.weight)
print(linear_layer.bias)

El fragmento de código anterior imprime la matriz de pesos y el vector de sesgos de la capa nn.Linear.

Si bien PyTorch inicializa estos parámetros de forma aleatoria, también puede establecerlos manualmente o utilizar diferentes métodos de inicialización. Por ejemplo, puede usar el módulo torch.nn.init para aplicar métodos de inicialización específicos a los pesos y sesgos. Aquí hay un ejemplo de cómo utilizar la inicialización uniforme de Xavier:

import torch.nn.init as init
 
## Inicializar los pesos utilizando la inicialización uniforme de Xavier
init.xavier_uniform_(linear_layer.weight)
 
## Inicializar los sesgos a cero
init.zeros_(linear_layer.bias)

En el fragmento de código anterior, utilizamos la función xavier_uniform_ de torch.nn.init para inicializar los pesos de nuestra linear_layer. El sesgo se inicializa a cero utilizando la función zeros_. Estos métodos de inicialización pueden ayudar a mejorar el proceso de aprendizaje de la red neuronal.

Comparando nn.Linear y nn.Conv2d

nn.Linear y nn.Conv2d son ambos módulos fundamentales en PyTorch utilizados para diferentes propósitos. Mientras que nn.Linear aplica una transformación lineal a los datos de entrada, nn.Conv2d aplica una convolución 2D sobre una señal de entrada compuesta por varios planos de entrada.

La principal diferencia entre nn.Linear y nn.Conv2d radica en su aplicación. nn.Linear se utiliza típicamente en capas completamente conectadas donde cada neurona de entrada está conectada a cada neurona de salida. Por otro lado, nn.Conv2d se utiliza en capas convolucionales, que se utilizan principalmente en redes neuronales convolucionales (CNN) para tareas como el procesamiento de imágenes.

En cuanto a sus parámetros, nn.Linear requiere el número de características de entrada y el número de características de salida. nn.Conv2d requiere el número de canales de entrada (o la profundidad de la entrada), el número de canales de salida y el tamaño del kernel.

Aplicaciones de nn.Linear en el aprendizaje profundo

nn.Linear es un módulo versátil en PyTorch y tiene numerosas aplicaciones en el aprendizaje profundo. Aquí hay algunos ejemplos:

  1. Perceptrón Multicapa (MLP): Los MLP son un tipo de red neuronal de alimentación directa que consta de al menos tres capas de nodos: una capa de entrada, una capa oculta y una capa de salida. Cada capa está completamente conectada a la siguiente, y nn.Linear se utiliza para implementar estas conexiones.

  2. Regresión Lineal: En tareas de regresión lineal, nn.Linear se puede utilizar para implementar la ecuación lineal que aprende el modelo.

  3. Transformación de Datos: nn.Linear se puede utilizar para transformar los datos de entrada en una dimensión superior para tareas más complejas.

  4. Modelos de Aprendizaje Profundo: Muchos modelos de aprendizaje profundo, como los autoencoders, utilizan nn.Linear en su arquitectura.

En el siguiente segmento, profundizaremos en más detalles sobre el uso de nn.Linear en un modelo PyTorch, incluyendo cómo inicializar pesos y sesgos, y cómo utilizarlo en un modelo. También proporcionaremos ejemplos de sus aplicaciones en el aprendizaje profundo.

Uso de nn.Linear en un Modelo PyTorch

Incorporar nn.Linear en un modelo PyTorch implica definir la capa en el constructor del modelo y luego aplicarla a los datos de entrada en el método forward. Aquí hay un ejemplo de una red neuronal de alimentación directa simple que utiliza nn.Linear:

class Net(nn.Module):
    def __init__(self):
        super(Net, self).__init__()
        self.fc1 = nn.Linear(10, 5)
        self.fc2 = nn.Linear(5, 2)
 
    def forward(self, x):
        x = F.relu(self.fc1(x))
        x = self.fc2(x)
        return x
 
# Crear una instancia de la red
net = Net()

En el fragmento de código anterior, definimos una red Net con dos capas lineales (fc1 y fc2). El método forward define el pase hacia adelante de la entrada a través de la red. La función F.relu aplica la función de activación ReLU a la salida de la primera capa lineal antes de pasarla a la segunda capa lineal.

Errores Comunes y Soluciones para nn.Linear en PyTorch

Al usar nn.Linear, es posible que encuentres algunos errores comunes. Aquí tienes algunos de ellos junto con sus soluciones:

  1. Tamaño de Entrada Desigual: El tamaño de entrada debe coincidir con el parámetro in_features de nn.Linear. Si no coinciden, se producirá un error en tiempo de ejecución. Para solucionar esto, asegúrate de que el tamaño del tensor de entrada coincida con el parámetro in_features.

  2. Tamaños Incorrectos de Pesos y Sesgos: La matriz de pesos y el vector de sesgos deben tener tamaños que coincidan con los parámetros in_features y out_features. Si no coinciden, se producirá un error en tiempo de ejecución. Para solucionar esto, asegúrate de que los tamaños de la matriz de pesos y el vector de sesgos sean correctos.

  3. Uso de nn.Linear con Entrada 3D: nn.Linear espera una entrada 2D, pero a veces puedes pasar accidentalmente una entrada 3D (por ejemplo, desde una capa de convolución en una CNN). Esto dará como resultado un error en tiempo de ejecución. Para solucionar esto, puedes usar torch.flatten o view para remodelar la entrada a 2D.

Conclusión

En conclusión, nn.Linear es un componente fundamental en PyTorch y el aprendizaje profundo. Juega un papel crucial en la implementación de transformaciones lineales en redes neuronales, y comprenderlo puede ayudar significativamente en la construcción y solución de problemas de modelos de aprendizaje profundo. Ya sea que seas un principiante que recién comienza con PyTorch o un practicante experimentado, dominar nn.Linear es una habilidad valiosa en tu conjunto de herramientas de aprendizaje profundo.

Preguntas Frecuentes

¿Cuál es el propósito de un vector de sesgos en nn.Linear?

El vector de sesgos en nn.Linear permite que el modelo desplace la salida de la transformación lineal a lo largo del eje y. Esto puede ser crucial para ajustar el modelo a los datos, especialmente cuando los datos no están centrados alrededor del origen. Sin el sesgo, el modelo siempre pasaría por el origen, lo que podría limitar su capacidad para ajustarse a los datos.

¿Cómo se inicializan los pesos y sesgos en una capa lineal en PyTorch?

Los pesos y sesgos para una capa lineal en PyTorch se pueden inicializar cuando se crea un objeto nn.Linear. De forma predeterminada, se inicializan con valores aleatorios. Sin embargo, puedes establecerlos manualmente o utilizar diferentes métodos de inicialización proporcionados por el módulo torch.nn.init.

¿Cuál es la diferencia entre nn.Linear y nn.Conv2d en PyTorch?

nn.Linear y nn.Conv2d se utilizan ambos para implementar capas en redes neuronales, pero sirven para propósitos diferentes. nn.Linear aplica una transformación lineal a los datos de entrada y se utiliza principalmente en capas completamente conectadas. Por otro lado, nn.Conv2d aplica una convolución 2D a una señal de entrada y se utiliza principalmente en capas convolucionales para tareas como el procesamiento de imágenes.