Kalman Filter for Online Classification of Non-Stationary Data

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Titsias, Michalis K., Galashov, Alexandre, Rannen-Triki, Amal, Pascanu, Razvan, Teh, Yee Whye, Bornschein, Jorg
Format: Preprint
Publié: 2023
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866909380061954048
author Titsias, Michalis K.
Galashov, Alexandre
Rannen-Triki, Amal
Pascanu, Razvan
Teh, Yee Whye
Bornschein, Jorg
author_facet Titsias, Michalis K.
Galashov, Alexandre
Rannen-Triki, Amal
Pascanu, Razvan
Teh, Yee Whye
Bornschein, Jorg
contents In Online Continual Learning (OCL) a learning system receives a stream of data and sequentially performs prediction and training steps. Important challenges in OCL are concerned with automatic adaptation to the particular non-stationary structure of the data, and with quantification of predictive uncertainty. Motivated by these challenges we introduce a probabilistic Bayesian online learning model by using a (possibly pretrained) neural representation and a state space model over the linear predictor weights. Non-stationarity over the linear predictor weights is modelled using a parameter drift transition density, parametrized by a coefficient that quantifies forgetting. Inference in the model is implemented with efficient Kalman filter recursions which track the posterior distribution over the linear weights, while online SGD updates over the transition dynamics coefficient allows to adapt to the non-stationarity seen in data. While the framework is developed assuming a linear Gaussian model, we also extend it to deal with classification problems and for fine-tuning the deep learning representation. In a set of experiments in multi-class classification using data sets such as CIFAR-100 and CLOC we demonstrate the predictive ability of the model and its flexibility to capture non-stationarity.
format Preprint
id arxiv_https___arxiv_org_abs_2306_08448
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Kalman Filter for Online Classification of Non-Stationary Data
Titsias, Michalis K.
Galashov, Alexandre
Rannen-Triki, Amal
Pascanu, Razvan
Teh, Yee Whye
Bornschein, Jorg
Machine Learning
Artificial Intelligence
In Online Continual Learning (OCL) a learning system receives a stream of data and sequentially performs prediction and training steps. Important challenges in OCL are concerned with automatic adaptation to the particular non-stationary structure of the data, and with quantification of predictive uncertainty. Motivated by these challenges we introduce a probabilistic Bayesian online learning model by using a (possibly pretrained) neural representation and a state space model over the linear predictor weights. Non-stationarity over the linear predictor weights is modelled using a parameter drift transition density, parametrized by a coefficient that quantifies forgetting. Inference in the model is implemented with efficient Kalman filter recursions which track the posterior distribution over the linear weights, while online SGD updates over the transition dynamics coefficient allows to adapt to the non-stationarity seen in data. While the framework is developed assuming a linear Gaussian model, we also extend it to deal with classification problems and for fine-tuning the deep learning representation. In a set of experiments in multi-class classification using data sets such as CIFAR-100 and CLOC we demonstrate the predictive ability of the model and its flexibility to capture non-stationarity.
title Kalman Filter for Online Classification of Non-Stationary Data
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2306.08448