COSCO: A Sharpness-Aware Training Framework for Few-shot Multivariate Time Series Classification

Jesus Barreda,Ashley Gomez,Ruben Puga,Kaixiong Zhou,Li Zhang
DOI: https://doi.org/10.1145/3627673.3679891
2024-09-15
Abstract:Multivariate time series classification is an important task with widespread domains of applications. Recently, deep neural networks (DNN) have achieved state-of-the-art performance in time series classification. However, they often require large expert-labeled training datasets which can be infeasible in practice. In few-shot settings, i.e. only a limited number of samples per class are available in training data, DNNs show a significant drop in testing accuracy and poor generalization ability. In this paper, we propose to address these problems from an optimization and a loss function perspective. Specifically, we propose a new learning framework named COSCO consisting of a sharpness-aware minimization (SAM) optimization and a Prototypical loss function to improve the generalization ability of DNN for multivariate time series classification problems under few-shot setting. Our experiments demonstrate our proposed method outperforms the existing baseline methods. Our source code is available at: <a class="link-external link-https" href="https://github.com/JRB9/COSCO" rel="external noopener nofollow">this https URL</a>.
Machine Learning,Artificial Intelligence,Neural and Evolutionary Computing
What problem does this paper attempt to address?
The paper aims to address the problem of Multivariate Time Series Classification (MTSC) in a few-shot setting. Specifically, the paper proposes a new framework called COSCO, which combines Sharpness-Aware Minimization (SAM) optimization technique and Prototypical Loss to improve the generalization ability of Deep Neural Networks (DNN) for multivariate time series classification tasks under few-shot settings. The study finds that traditional deep learning models often exhibit poor generalization performance, especially in terms of test accuracy, when the amount of data is limited. COSCO addresses this issue by optimizing the loss function landscape and smoothing local minima, thereby enhancing the model's classification performance under few-shot conditions. Experimental results show that COSCO outperforms existing baseline methods on various datasets.