An Ensemble of Transformer and LSTM Approach for Multivariate Time Series Data Classification

Neha Tarannum Pendari,Bodhi Satwa Mishra,P. G. S. Hiremath,Aryan Narayan,Shankar Gangisetty
DOI: https://doi.org/10.1109/BigData52589.2021.9671632
2021-12-15
Abstract:Wafer manufacturing is a complex and time taking process. The multivariate time-series data collected from many soft sensors in the process are highly noisy and imbalanced. Thus, wafer classification is a challenging task. To overcome this challenge, we propose an effective ensemble approach with transformer and long short term memory (LSTM) based deep learning techniques for wafer classification. Though deep learning is a promising technique to analyze the data and make effective predictions, but not widely integrated in manufacture industries for soft sensing due to insufficient research. Also the research community has not been exposed to accessing the real and large scale wafer data that is highly noisy and imbalanced. Our proposed approach is an ensemble of four models, namely, multilayer LSTM, multilayer perceptron classifier, transformer, and feed forward neural network. We finally ensemble all of these models using ROC-AUC scores by adjusting the weights based on skewness of the models to obtain effective performance. We perform an exhaustive empirical analysis of the proposed approach and obtain a best ROC score of 0.748 that is significantly better compared to the baseline models.
Engineering,Materials Science,Computer Science
What problem does this paper attempt to address?