Forecasting time series with decomposition

December 2, 2023
Facebook logo.
Twitter logo.
LinkedIn logo.
Get this code in Google Colab

Forecasting time series with decomposition

In today’s newsletter, I’m going to show you how to forecast a time series of US unemployment data using decomposition.

Time series decomposition is breaking down a single time series into different parts. Each part represents a pattern that you can try to model and predict. The patterns usually fall into three categories: trend, seasonality, and noise.

Time series decomposition models are additive, multiplicative, or some combination of both. Additive models are used when the trend is not proportional to the level of the time series. Multiplicative models are used when the trend is proportional.

Decomposition is used by quants to forecast time series. For example, decomposing and forecasting macroeconomic time series helps portfolio managers and macro investors identify trends, rebalance portfolios to take advantage of seasonality, and manage risk.

Forecasting time series with decomposition

Most financial data is non-stationary so predictive models fit badly. Decomposing can help you remove components to make the time series more stationary. Once the stationary components are modeled, you can reassemble the time series.

Unfortunately, most people forget about decomposition as a way to predict time series. Especially when it comes to macroeconomic data.

By the end of this newsletter you will be able to:

Here's how to do it in Python, step by step.

Step 1: Import the libraries and get the data

First, import pandas for data manipulation, statsmodels for time series analysis, and the OpenBB SDK for data.

1import pandas as pd
2
3from statsmodels.tsa.seasonal import seasonal_decompose, STL
4from openbb_terminal.sdk import openbb

Use the OpenBB SDK to get the unemployment data.

1df = openbb.economy.unemp(2010)
2df = df.set_index("date")[:"2019-12-31"].sort_index()

I use 20 years of unemployment data. You can use whatever time series you want.

Step 2: Decompose the time series for seasonality and trend

Start with visualizing the data through time by plotting the time series.

df["rolling_mean"] = df["unemp"].rolling(window=12).mean()
df["rolling_std"] = df["unemp"].rolling(window=12).std()
df.plot(title="Unemployment rate")

Forecasting time series with decomposition. I’m going to show you how to decompose a time series of US unemployment data.

There’s a clear downward trend in the unemployment rate. There also appears to be some consistent spikes. Time series decomposition should pick up these patterns.

Run the model and plot the results.

1decomposition_results = seasonal_decompose(
2    df["unemp"], 
3    model="additive"
4).plot()
Forecasting time series with decomposition. I’m going to show you how to decompose a time series of US unemployment data.

The code extracted the trend, seasonal, and noise components from the time series. Take a look at the noise component and inspect if it looks random. If there was a strong pattern, it would tell you the time series is serially auto-correlated and the model fit is suspect.

The additive model is basic and comes with caveats. First, there are missing data points at the beginning and end because the algorithm uses a moving average. The model assumes there’s the same seasonal pattern every year. Finally, the model is not robust to outliers.

Time to try a more robust method.

Step 3: Decompose the time series with LOESS

Seasonal-trend decomposition (STL) uses locally estimated scatterplot smoothing (LOESS) to extract seasonality and trend from a time series. It improves on the basic additive model by handling any kind of seasonality and being more robust to outliers.

Run the model and plot the results.

1stl_decomposition = STL(df[["unemp"]]).fit()
2stl_decomposition.plot().suptitle("STL Decomposition");
Forecasting time series with decomposition. I’m going to show you how to decompose a time series of US unemployment data.

The results are like the additive model. But, there are no missing values, and the seasonality component changes slowly over time.

In today’s newsletter, I showed you how to decompose a time series of US unemployment data. Now you can inspect the trend and seasonality of a data series. By inspecting the noise component, you have an idea of the “goodness of fit” of the model, too. By decomposing time series you can model each component and better predict the combined time series.

Man with glasses and a wristwatch, wearing a white shirt, looking thoughtfully at a laptop with a data screen in the background.