EventsThe 1st International Online Conference on Forecasting
Published
This submission belongs to the session S4. Weather and Climate Forecasting of the event The 1st International Online Conference on Forecasting
Published date
16 Sep, 2026
Academic Editor
author-avatarSonia Leva
Citation
KATIKALA JYOTHI, NIDAMANURI SRINU, DASARI BUJJI BABU, RAJASEKHAR MANDA, An Optimized Ensemble Machine Learning Framework for Multi-City AQI Forecasting in India: A Comparative Study of Feature Engineering and Hyperparameter Tuning, in Proceedings of The 1st International Online Conference on Forecasting, 21 September–22 September 2026, MDPI: Basel, Switzerland
Share
Email
Facebook
Twitter
LinkedIn

An Optimized Ensemble Machine Learning Framework for Multi-City AQI Forecasting in India: A Comparative Study of Feature Engineering and Hyperparameter Tuning

NIDAMANURI SRINU 2
DASARI BUJJI BABU 1
1. Department of Computer Science and Engineering, QIS College of Engineering and Technology (A), Vengamukkapalem, Ongole-523272, Andhra Pradesh, India
2. Department of Computer Science and Engineering, QIS College of Engineering and Technology (A), Vengamukkapalem, Ongole-523272, Andhra Pradesh, India
3. Department of ECE, QIS College of Engineering and Technology (A), Vengamukkapalem, Ongole-523272, Andhra Pradesh, India
Abstract

Air pollution has become one of the most pressing environmental challenges confronting India's rapidly expanding smart cities, with hazardous pollutant levels directly impacting public health, urban livability, and long-term sustainability. Accurate and timely forecasting of the Air Quality Index (AQI) is therefore not merely an academic pursuit but a public health necessity, enabling proactive interventions and informed policy decisions. This study develops and rigorously validates an optimized machine learning framework for multi-city AQI prediction, leveraging historical pollutant concentration data alongside key meteorological parameters such as temperature and wind speed. The proposed framework implements a comprehensive comparative analysis of four regression models, Linear Regression, Support Vector Machines, Random Forest, and Gradient Boosting, across four Indian cities representing diverse pollution profiles: Delhi, Visakhapatnam, Hyderabad, and Kolkata. A critical contribution of this work lies in its systematic approach to feature engineering, incorporating lag variables and seasonal indicators, coupled with extensive hyperparameter tuning using grid search optimization. This optimization process significantly enhanced model performance, with the Gradient Boosting model emerging as the most robust predictor, achieving an exceptional R² of 0.98 with a Root Mean Square Error (RMSE) of just 10.0 for Visakhapatnam, and an R² of 0.97 for Hyderabad. These metrics demonstrate substantial accuracy improvements over conventional statistical approaches and non-optimized machine learning baselines. The resulting framework offers a reliable, scalable, and data-driven decision-support tool for environmental agencies and urban policymakers. It facilitates timely public health advisories, supports the design of targeted pollution mitigation strategies, and empowers citizens with actionable air quality information. Ultimately, this research contributes to the broader vision of smarter, more resilient urban governance, fostering healthier and more sustainable ecosystems across India's evolving smart city landscape.

Keywords
Air Quality Index (AQI)
Machine Learning Framework
Ensemble Learning
Hyperparameter Optimization
Smart Cities
India
Feature Selection
Gradient Boosting
Random Forest
Urban Air Pollution
Poster
IOCTC 2026 AQI Forecasting Poster.pdf
Preliminary Comparative Evaluation of Time Series Models in Forecasting Crude Oil Exports in Saudi Arabia