Regresión con máquinas de soporte vectorial (SVR) | Predicciones robustas con un tubo ε-insensible

Creado: 2019-07-10 Última actualización: 2020-06-03 Tiempo de lectura: 3 min

まとめ

SVR extiende las SVM a regresión, tratando los errores dentro de un tubo ε-insensible como cero para reducir el impacto de los valores atípicos.
Gracias a los kernels puede capturar relaciones no lineales mientras mantiene el modelo compacto mediante vectores de soporte.
Los hiperparámetros C, epsilon y gamma controlan el equilibrio entre generalización y suavidad.
La estandarización de las características es esencial; encapsular el preprocesado y el aprendizaje en un pipeline garantiza transformaciones consistentes.

Intuición #

SVR ajusta una función rodeada por un tubo de ancho ε: los puntos dentro del tubo no incurren en pérdida, mientras que los que lo atraviesan pagan una penalización. Solo los puntos que tocan o salen del tubo —los vectores de soporte— influyen en el modelo final, lo que produce aproximaciones suaves y resistentes al ruido.

Formulación matemática #

El problema de optimización es

$$ \min_{\mathbf{w}, b, \boldsymbol{\xi}, \boldsymbol{\xi}^} \frac{1}{2} |\mathbf{w}|^2 + C \sum_{i=1}^{n} (\xi_i + \xi_i^) $$

sujeto a

$$ \begin{aligned} y_i - (\mathbf{w}^\top \phi(\mathbf{x}_i) + b) &\le \epsilon + \xi_i, \ (\mathbf{w}^\top \phi(\mathbf{x}_i) + b) - y_i &\le \epsilon + \xi_i^, \ \xi_i, \xi_i^ &\ge 0, \end{aligned} $$

donde $\phi$ es la transformación inducida por el kernel elegido. Resolver el dual revela los vectores de soporte y sus coeficientes.

Experimentos con Python #

Ejemplo de SVR combinado con StandardScaler dentro de un pipeline.

from __future__ import annotations

import japanize_matplotlib
import matplotlib.pyplot as plt
import numpy as np
from sklearn.metrics import mean_squared_error
from sklearn.model_selection import train_test_split
from sklearn.pipeline import make_pipeline
from sklearn.preprocessing import StandardScaler
from sklearn.svm import SVR


def run_svr_demo(
    *,
    n_samples: int = 300,
    train_size: float = 0.75,
    xlabel: str = "input x",
    ylabel: str = "output y",
    label_train: str = "train samples",
    label_test: str = "test samples",
    label_pred: str = "SVR prediction",
    label_truth: str = "ground truth",
    title: str | None = None,
) -> dict[str, float]:
    """Train SVR on synthetic nonlinear data, plot fit, and report metrics.

    Args:
        n_samples: Number of data points sampled from the underlying function.
        train_size: Fraction of data used for training.
        xlabel: X-axis label for the plot.
        ylabel: Y-axis label for the plot.
        label_train: Legend label for training samples.
        label_test: Legend label for test samples.
        label_pred: Legend label for the SVR prediction line.
        label_truth: Legend label for the ground-truth curve.
        title: Optional title for the plot.

    Returns:
        Dictionary containing training and test RMSE values.
    """
    japanize_matplotlib.japanize()
    rng = np.random.default_rng(seed=42)

    X = np.linspace(0.0, 6.0, n_samples, dtype=float)
    y_true = np.sin(X) * 1.5 + 0.3 * np.cos(2 * X)
    y_noisy = y_true + rng.normal(scale=0.2, size=X.shape)

    X_train, X_test, y_train, y_test, y_true_train, y_true_test = train_test_split(
        X[:, np.newaxis],
        y_noisy,
        y_true,
        train_size=train_size,
        random_state=42,
        shuffle=True,
    )

    svr = make_pipeline(
        StandardScaler(),
        SVR(kernel="rbf", C=10.0, epsilon=0.1, gamma="scale"),
    )
    svr.fit(X_train, y_train)

    train_pred = svr.predict(X_train)
    test_pred = svr.predict(X_test)

    grid = np.linspace(0.0, 6.0, 400, dtype=float)[:, np.newaxis]
    grid_truth = np.sin(grid.ravel()) * 1.5 + 0.3 * np.cos(2 * grid.ravel())
    grid_pred = svr.predict(grid)

    fig, ax = plt.subplots(figsize=(10, 5))
    ax.scatter(X_train, y_train, color="#1f77b4", alpha=0.6, label=label_train)
    ax.scatter(X_test, y_test, color="#ff7f0e", alpha=0.6, label=label_test)
    ax.plot(grid, grid_truth, color="#2ca02c", linewidth=2, label=label_truth)
    ax.plot(grid, grid_pred, color="#d62728", linewidth=2, linestyle="--", label=label_pred)
    ax.set_xlabel(xlabel)
    ax.set_ylabel(ylabel)
    if title:
        ax.set_title(title)
    ax.legend()
    fig.tight_layout()
    plt.show()

    return {
        "train_rmse": float(mean_squared_error(y_train, train_pred, squared=False)),
        "test_rmse": float(mean_squared_error(y_test, test_pred, squared=False)),
    }



metrics = run_svr_demo(
    xlabel="entrada x",
    ylabel="salida y",
    label_train="muestras de entrenamiento",
    label_test="muestras de prueba",
    label_pred="predicción SVR",
    label_truth="curva real",
    title="SVR en un conjunto no lineal",
)
print(f"RMSE de entrenamiento: {metrics['train_rmse']:.3f}")
print(f"RMSE de prueba: {metrics['test_rmse']:.3f}")

Interpretación de los resultados #

El pipeline escala los datos de entrenamiento con su media y varianza y aplica la misma transformación al conjunto de prueba.
pred contiene las predicciones; ajustar epsilon y C modifica el equilibrio entre sobreajuste y subajuste.
Aumentar gamma en el kernel RBF enfatiza patrones locales, mientras que valores pequeños producen funciones más suaves.

Referencias #

Smola, A. J., & Schölkopf, B. (2004). A Tutorial on Support Vector Regression. Statistics and Computing, 14(3), 199–222.
Vapnik, V. (1995). The Nature of Statistical Learning Theory. Springer.