GreenEyes: An Air Quality Evaluating Model based on WaveNet

Kan Huang,Kai Zhang,Ming Liu
DOI: https://doi.org/10.48550/arXiv.2212.04175
2022-12-08
Abstract:Accompanying rapid industrialization, humans are suffering from serious air pollution problems. The demand for air quality prediction is becoming more and more important to the government's policy-making and people's daily life. In this paper, We propose GreenEyes -- a deep neural network model, which consists of a WaveNet-based backbone block for learning representations of sequences and an LSTM with a Temporal Attention module for capturing the hidden interactions between features of multi-channel inputs. To evaluate the effectiveness of our proposed method, we carry out several experiments including an ablation study on our collected and preprocessed air quality data near HKUST. The experimental results show our model can effectively predict the air quality level of the next timestamp given any segment of the air quality data from the data set. We have also released our standalone dataset at <a class="link-external link-https" href="https://github.com/AI-Huang/IAQI_Dataset" rel="external noopener nofollow">this https URL</a> The model and code for this paper are publicly available at <a class="link-external link-https" href="https://github.com/AI-Huang/AirEvaluation" rel="external noopener nofollow">this https URL</a>
Machine Learning,Signal Processing
What problem does this paper attempt to address?
The problem that this paper attempts to solve is the air quality prediction problem in air quality management. With the acceleration of the global industrialization process, air pollution poses a serious threat to human health. Accurate air quality prediction is of great significance for government policy - making and residents' daily lives. Specifically, the paper proposes a deep neural network model based on WaveNet - GreenEyes, aiming to effectively predict the air quality index (AQI) at a certain point in the future, especially the IAQI (Individual Air Quality Index) of PM2.5. ### Specific description of the problem 1. **Hazards of air pollution**: According to the data of the World Health Organization (WHO), air pollution is one of the largest environmental health risks in the world, which can lead to diseases such as respiratory infections, heart diseases, chronic obstructive pulmonary disease (COPD), strokes and lung cancers. 2. **Limitations of existing systems**: Although existing Internet of Things (IoT) systems can monitor air quality, most systems only perform quality detection tasks and cannot predict future air quality, so it is difficult to achieve intelligent control to save energy. 3. **Data processing challenges**: The original air quality data has high - frequency noise, and directly using these data for training will cause difficulties for the model to learn, especially in areas with large data fluctuations. ### Solutions To address the above problems, the paper proposes the following solutions: - **GreenEyes model**: This model combines the WaveNet architecture to learn sequence representations and uses LSTM with a time - attention mechanism to capture the hidden interactions between multi - channel input features. - **Data pre - processing**: Convert the original IAQI levels into polygon functions by means of manual annotation, thereby eliminating high - frequency noise in the data and maintaining trend features. - **Multi - sensor fusion**: Use multiple sensors of the same model at the same location to collect data to improve the prediction performance of the model. ### Main contributions 1. **Innovative model architecture**: Use the residual layers of WaveNet as feature blocks, stack multiple WaveNet blocks to build the main body of the model, and add an attention mechanism and LSTM at the end. 2. **Data processing method**: Propose a method based on manual annotation to convert the original IAQI levels into polygon functions, which solves the data fluctuation problem. 3. **Experimental verification**: Verify the effectiveness of the model through a series of experiments, including ablation experiments and performance comparisons on different datasets. In conclusion, this paper aims to provide a more accurate and stable air quality prediction method by introducing the GreenEyes model, thereby helping to improve air pollution management and residents' quality of life.