Abstract
The effective monitoring of PM2.5, a major indicator of air pollution, is crucial to human activities. Compared to traditional physiochemical techniques, image-based methods train PM2.5 estimators by using datasets containing pairs of images and PM2.5 levels, which are efficient, economical, and convenient to deploy. However, existing methods either employ handcrafted features, which can be influenced by the image content, or require additional weather data acquired probably by laborious processes. To estimate the PM2.5 concentration from a single image without requiring extra data, we herein propose a learning-based prior-enhanced (PE) network—comprising a main branch, an auxiliary branch, and a feature fusion attention module—to learn from an input image and its corresponding dark channel (DC) and inverted saturation (IS) maps. In addition, we propose an histogram smoothing (HS) algorithm to solve the problem of imbalanced data distribution, thereby improving the estimation accuracy in cases of heavy air pollution. To the best of our knowledge, this study is the first to address the phenomenon of a data imbalance in image-based PM2.5 estimation. Finally, we construct a new dataset containing multi-angle images and more than 30 types of air data. Extensive experiments on image-based PM2.5 monitoring datasets verify the superior performance of our proposed neural networks and the HS strategy.
X. Fang and Z. Li—Both authors contributed equally to this work.
This is a preview of subscription content, log in via an institution to check access.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
Similar content being viewed by others
References
Bu, X., et al.: Global PM2.5-attributable health burden from 1990 to 2017: Estimates from the global burden of disease study 2017. Environ. Res. 197, 111123 (2021)
Charron, A., Harrison, R.M., Moorcroft, S., Booker, J.: Quantitative interpretation of divergence between PM10 and PM2.5 mass measurement by TEOM and gravimetric (partisol) instruments. Atmos. Environ. 38(3), 415–423 (2004)
Chawla, N.V., Bowyer, K.W., Hall, L.O., Kegelmeyer, W.P.: SMOTE: synthetic minority over-sampling technique. J. Artif. Intell. Res. 16, 321–357 (2002)
Chok, N.S.: Pearson’s versus Spearman’s and Kendall’s correlation coefficients for continuous data, Ph. D. thesis, University of Pittsburgh (2010)
Feng, L., Yang, T., Wang, Z.: Performance evaluation of photographic measurement in the machine-learning prediction of ground PM2.5 concentration. Atmos. Environ. 262, 118623 (2021)
Gu, K., Liu, H., Xia, Z., Qiao, J., Lin, W., Thalmann, D.: PM2.5 monitoring: use information abundance measurement and wide and deep learning. IEEE Trans. Neural Networks Learn. Syst. 32(10), 4278–4290 (2021)
Gu, K., Qiao, J., Li, X.: Highly efficient picture-based prediction of PM2.5 concentration. IEEE Trans. Ind. Electron. 66(4), 3176–3184 (2018)
He, K., Sun, J., Tang, X.: Single image haze removal using dark channel prior. IEEE Trans. Pattern Anal. Mach. Intell. 33(12), 2341–2353 (2010)
He, K., Zhang, X., Ren, S., Sun, J.: Deep residual learning for image recognition. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 770–778 (2016)
He, T., Zhang, Z., Zhang, H., Zhang, Z., Xie, J., Li, M.: Bag of tricks for image classification with convolutional neural networks. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 558–567 (2019)
Hu, J., Shen, L., Sun, G.: Squeeze-and-excitation networks. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 7132–7141 (2018)
Kingma, D.P., Ba, J.: Adam: A method for stochastic optimization. arXiv preprint arXiv:1412.6980 (2014)
Kumari, R., Singh, J., Gosain, A.: SmS: smote-stacked hybrid model for diagnosis of polycystic ovary syndrome using feature selection method. Expert Syst. Appl. 225, 120102 (2023)
Liu, C., Tsow, F., Zou, Y., Tao, N.: Particle pollution estimation based on image analysis. PLoS ONE 11(2), e0145955 (2016)
Liu, Z., et al.: Swin transformer: hierarchical vision transformer using shifted windows. In: Proceedings of the IEEE/CVF International Conference on Computer Vision, pp. 10012–10022 (2021)
Maas, A.L., Hannun, A.Y., Ng, A.Y., et al.: Rectifier nonlinearities improve neural network acoustic models. In: Proceedings of the International Conference on Machine Learning, p. 3. Citeseer (2013)
Marquès, M., Domingo, J.L.: Positive association between outdoor air pollution and the incidence and severity of COVID-19. a review of the recent scientific evidences. Environ. Res. 203, 111930 (2022)
Paszke, A., et al.: Pytorch: an imperative style, high-performance deep learning library. In: Advances in Neural Information Processing Systems, vol. 32 (2019)
Peng, M., et al.: Trainable undersampling for class-imbalance learning. In: Proceedings of the AAAI Conference on Artificial Intelligence, pp. 4707–4714 (2019)
Qiao, J., He, Z., Du, S.: Prediction of PM2.5 concentration based on weighted bagging and image contrast-sensitive features. Stochast. Environ. Res. Risk Assess. 34(3), 561–573 (2020)
Rijal, N., Gutta, R.T., Cao, T., Lin, J., Bo, Q., Zhang, J.: Ensemble of deep neural networks for estimating particulate matter from images. In: 2018 IEEE 3rd International Conference on Image, Vision and Computing, pp. 733–738. IEEE (2018)
Selvaraju, R.R., Cogswell, M., Das, A., Vedantam, R., Parikh, D., Batra, D.: Grad-cam: visual explanations from deep networks via gradient-based localization. In: Proceedings of the IEEE International Conference on Computer Vision, pp. 618–626 (2017)
Shapiro, L.G., Stockman, G.C., et al.: Computer Vision, vol. 3. Prentice Hall New Jersey (2001)
Wang, G., Shi, Q., Wang, H., Sun, K., Lu, Y., Di, K.: Multi-modal image feature fusion-based PM2.5 concentration estimation. Atmos. Pollut. Res. 13(3), 101345 (2022)
Wang, T., et al.: C2AM loss: chasing a better decision boundary for long-tail object detection. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 6980–6989 (2022)
Yang, Y., Zha, K., Chen, Y., Wang, H., Katabi, D.: Delving into deep imbalanced regression. In: International Conference on Machine Learning, pp. 11842–11851. PMLR (2021)
Yue, G., Gu, K., Qiao, J.: Effective and efficient photo-based PM2.5 concentration estimation. IEEE Trans. Instrum. Measur. 68(10), 3962–3971 (2019)
Zhang, Y., Kang, B., Hooi, B., Yan, S., Feng, J.: Deep long-tailed learning: a survey. IEEE Trans. Pattern Anal. Mach. Intell. 45(9), 10795–10816 (2023). IEEE
Acknowledgment
This work is supported by the National Natural Science Foundation of China (No. 62071201), and Guangdong Basic and Applied Basic Research Foundation (No.2022A1515010119).
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2024 The Author(s), under exclusive license to Springer Nature Singapore Pte Ltd.
About this paper
Cite this paper
Fang, X. et al. (2024). Prior-Enhanced Network for Image-Based PM2.5 Estimation from Imbalanced Data Distribution. In: Luo, B., Cheng, L., Wu, ZG., Li, H., Li, C. (eds) Neural Information Processing. ICONIP 2023. Communications in Computer and Information Science, vol 1964. Springer, Singapore. https://doi.org/10.1007/978-981-99-8141-0_20
Download citation
DOI: https://doi.org/10.1007/978-981-99-8141-0_20
Published:
Publisher Name: Springer, Singapore
Print ISBN: 978-981-99-8140-3
Online ISBN: 978-981-99-8141-0
eBook Packages: Computer ScienceComputer Science (R0)