Logo image
Back
A Machine Learning Approach for Air Quality Prediction: Model Regularization and Optimization
Journal article   Open access   Peer reviewed

A Machine Learning Approach for Air Quality Prediction: Model Regularization and Optimization

Dixian Zhu, Changjie Cai, Tianbao Yang and Xun Zhou
Big data and cognitive computing, Vol.2(1), p.5
02/24/2018
DOI: 10.3390/bdcc2010005
url
https://doi.org/10.3390/bdcc2010005View
Published (Version of record) Open Access

Abstract

In this paper, we tackle air quality forecasting by using machine learning approaches to predict the hourly concentration of air pollutants (e.g., ozone, particle matter ( PM2.5 ) and sulfur dioxide). Machine learning, as one of the most popular techniques, is able to efficiently train a model on big data by using large-scale optimization algorithms. Although there exist some works applying machine learning to air quality prediction, most of the prior studies are restricted to several-year data and simply train standard regression models (linear or nonlinear) to predict the hourly air pollution concentration. In this work, we propose refined models to predict the hourly air pollution concentration on the basis of meteorological data of previous days by formulating the prediction over 24 h as a multi-task learning (MTL) problem. This enables us to select a good model with different regularization techniques. We propose a useful regularization by enforcing the prediction models of consecutive hours to be close to each other and compare it with several typical regularizations for MTL, including standard Frobenius norm regularization, nuclear norm regularization, and ℓ2,1 -norm regularization. Our experiments have showed that the proposed parameter-reducing formulations and consecutive-hour-related regularizations achieve better performance than existing standard regression models and existing regularizations.

Details

Metrics

42 Record Views
Logo image