Early time series classification (eTSC) is the problem of classifying a time series after as few measurements as possible with the highest possible accuracy. The most critical issue of any eTSC method is to decide when enough data of a time series has been seen to take a decision: Waiting for more data points usually makes the classification problem easier but delays the time in which a classification is made; in contrast, earlier classification has to cope with less input data, often leading to inferior accuracy.
This notebook gives a quick guide to get you started with running eTSC algorithms in aeon.
The UCR/UEA time series classification archive contains a large number of example TSC problems that have been used thousands of times in the literature to assess TSC algorithms. Read the data loading documentation and notebooks for details on the aeon data formats and loading data for aeon.
# Imports used in this notebook
import numpy as np
from aeon.classification.early_classification._teaser import TEASER
from aeon.classification.interval_based import TimeSeriesForestClassifier
from aeon.datasets import load_arrow_head
# Load default train/test splits from aeon/datasets/data
arrow_train_X, arrow_train_y = load_arrow_head(split="train")
arrow_test_X, arrow_test_y = load_arrow_head(split="test")
arrow_test_X.shape
(175, 1, 251)
TEASER [1] is a two-tier model using a base classifier to make predictions and a decision making estimator to decide whether these predictions are safe. As a first tier, TEASER requires a TSC algorithm, such as WEASEL, which produces class probabilities as output. As a second tier an anomaly detector is required, such as a one-class SVM.
teaser = TEASER(
random_state=0,
classification_points=[25, 50, 75, 100, 125, 150, 175, 200, 251],
estimator=TimeSeriesForestClassifier(n_estimators=10, random_state=0),
)
teaser.fit(arrow_train_X, arrow_train_y)
TEASER(classification_points=[25, 50, 75, 100, 125, 150, 175, 200, 251], estimator=TimeSeriesForestClassifier(n_estimators=10, random_state=0), random_state=0)In a Jupyter environment, please rerun this cell to show the HTML representation or trust the notebook.
TEASER(classification_points=[25, 50, 75, 100, 125, 150, 175, 200, 251], estimator=TimeSeriesForestClassifier(n_estimators=10, random_state=0), random_state=0)
TimeSeriesForestClassifier(n_estimators=10, random_state=0)
TimeSeriesForestClassifier(n_estimators=10, random_state=0)
Commonly accuracy is used to determine the correctness of the predictions, while earliness is used to determine how much of the series is required on average to obtain said accuracy. I.e. for the below values, using just 34% of the full test data, we were able to get an accuracy of 65%.
hm, acc, earl = teaser.score(arrow_test_X, arrow_test_y)
print("Earliness on Test Data %2.2f" % earl)
print("Accuracy on Test Data %2.2f" % acc)
print("Harmonic Mean on Test Data %2.2f" % hm)
Earliness on Test Data 0.34 Accuracy on Test Data 0.65 Harmonic Mean on Test Data 0.65
print("Earliness on Train Data %2.2f" % teaser._train_earliness)
print("Accuracy on Train Data %2.2f" % teaser._train_accuracy)
Earliness on Train Data 0.28 Accuracy on Train Data 0.72
With the full test data, we would obtain 67% accuracy with the same classifier.
accuracy = (
TimeSeriesForestClassifier(n_estimators=10, random_state=0)
.fit(arrow_train_X, arrow_train_y)
.score(arrow_test_X, arrow_test_y)
)
print("Accuracy on the full Test Data %2.2f" % accuracy)
Accuracy on the full Test Data 0.67
The main draw of eTSC is the capabilility to make classifications with incomplete time series. aeon eTSC algorithms accept inputs with less time points than the full series length, and output two items: The prediction made and whether the algorithm thinks the prediction is safe. Information about the decision such as the time stamp it was made at can be obtained from the state_info attribute.
X = arrow_test_X[:, :, :50]
probas, _ = teaser.predict_proba(X)
idx = (probas >= 0).all(axis=1)
print("First 10 Finished prediction\n", np.argwhere(idx).flatten()[:10])
print("First 10 Probabilities of finished predictions\n", probas[idx][:10])
First 10 Finished prediction [ 0 4 5 8 9 16 18 21 22 30] First 10 Probabilities of finished predictions [[0.8 0. 0.2] [0.8 0. 0.2] [0.6 0.1 0.3] [0.2 0.2 0.6] [0.6 0.3 0.1] [0. 0.3 0.7] [0.3 0.1 0.6] [0.1 0.3 0.6] [0.8 0.1 0.1] [0.8 0. 0.2]]
_, acc, _ = teaser.score(X, arrow_test_y)
print("Accuracy with 50 points on Test Data %2.2f" % acc)
Accuracy with 50 points on Test Data 0.65
The rationale is to keep the state info from the previous predictions in the TEASER object and use it whenever new data is available.
test_points = [25, 50, 75, 100, 125, 150, 175, 200, 251]
final_states = np.zeros((arrow_test_X.shape[0], 4), dtype=int)
final_decisions = np.zeros(arrow_test_X.shape[0], dtype=int)
open_idx = np.arange(0, arrow_test_X.shape[0])
teaser.reset_state_info()
for i in test_points:
probas, decisions = teaser.update_predict_proba(arrow_test_X[:, :, :i])
final_states[open_idx] = teaser.get_state_info()
arrow_test_X, open_idx, final_idx = teaser.split_indices_and_filter(
arrow_test_X, open_idx, decisions
)
final_decisions[final_idx] = i
(
hm,
acc,
earliness,
) = teaser.compute_harmonic_mean(final_states, arrow_test_y)
print("Earliness on length %2i is %2.2f" % (i, earliness))
print("Accuracy on length %2i is %2.2f" % (i, acc))
print("Harmonic Mean on length %2i is %2.2f" % (i, hm))
print("...........")
print("Time Stamp of final decisions", final_decisions)
Earliness on length 25 is 0.10 Accuracy on length 25 is 0.43 Harmonic Mean on length 25 is 0.59 ........... Earliness on length 50 is 0.20 Accuracy on length 50 is 0.67 Harmonic Mean on length 50 is 0.73 ........... Earliness on length 75 is 0.26 Accuracy on length 75 is 0.62 Harmonic Mean on length 75 is 0.67 ........... Earliness on length 100 is 0.29 Accuracy on length 100 is 0.64 Harmonic Mean on length 100 is 0.67 ........... Earliness on length 125 is 0.30 Accuracy on length 125 is 0.64 Harmonic Mean on length 125 is 0.67 ........... Earliness on length 150 is 0.32 Accuracy on length 150 is 0.64 Harmonic Mean on length 150 is 0.66 ........... Earliness on length 175 is 0.33 Accuracy on length 175 is 0.63 Harmonic Mean on length 175 is 0.65 ........... Earliness on length 200 is 0.34 Accuracy on length 200 is 0.64 Harmonic Mean on length 200 is 0.65 ........... Earliness on length 251 is 0.34 Accuracy on length 251 is 0.66 Harmonic Mean on length 251 is 0.66 ........... Time Stamp of final decisions [ 50 150 75 75 50 50 251 125 50 50 75 75 75 75 75 150 50 75 50 75 125 50 50 251 75 75 75 75 75 75 50 175 251 50 175 50 50 75 75 75 75 50 75 251 75 75 50 50 75 175 75 75 125 50 75 50 50 100 175 75 150 50 75 50 75 75 75 75 50 75 50 75 150 50 125 50 100 75 50 50 175 75 251 75 50 75 75 175 50 50 75 50 50 50 75 50 75 100 75 100 50 50 50 50 50 50 50 150 75 50 50 50 251 100 125 75 125 100 75 50 75 50 50 75 200 50 50 100 50 50 75 75 50 150 50 75 50 75 200 50 75 75 200 200 75 50 75 200 75 75 50 200 75 75 75 100 200 75 75 50 100 50 50 75 50 75 75 75 75 50 75 75 50 100 75]
[1] Schäfer, P., & Leser, U. (2020). TEASER: early and accurate time series classification. Data mining and knowledge discovery, 34(5), 1336-1362