Data Augmentation and Class Imbalance Compensation Using CTGAN to Improve Gas Detection Systems

作者：Shima Mahinnezhad, Shirin Mahinnezhad, Kuljeet Kaur, A. Shih · 发表于：International Instrumentation and Measurement Technology Conference · 年份：2024 · DOI：10.1109/I2MTC60896.2024.10561121 · 被引用次数：5 · 研究领域：Computer Science

The use of sensors in gas detection systems for environmental monitoring is largely affected by sensor drift over time which reduces accurate classification. This drift can be minimized by using machine learning models trained on sensor data. Here, two different machine learning models are trained on the Gas Sensor Array Drift Dataset. However, this dataset, which has been collected over three years, suffers not only from drift but also from class imbalance. As a result, machine learning models cannot perform properly on this dataset. To address these problems, this paper introduces an innovative methodology for data compensation and augmentation using Conditional Tabular Generative Adversarial Networks (CTGAN). By employing this methodology, we can counteract the class imbalance and limit drift by bringing diversity to the dataset, which in turn improves the accuracy of machine learning models for gas detection systems. With class imbalance compensation, Multi-Layer Perceptron (MLP) and Support Vector Machines (SVM) achieved an improvement in classification accuracy in five batches, up to 20% for certain batches. Through data augmentation, they reached higher accuracy across six batches, with certain batches exceeding a 10% improvement. These achievements highlight the effectiveness and reliability of the use of synthetic data generation in tabular data for sensors.