Skip to main content
Log in

PM2.5 concentration prediction based on WD-SA-LSTM-BP model: a case study of Nanjing city

  • Research Article
  • Published:
Environmental Science and Pollution Research Aims and scope Submit manuscript

Abstract

PM2.5 concentration is an important indicator to measure the concentration of air pollutants, and it is of important social significance and application value to realize accurate prediction of PM2.5 concentration. To further improve the accuracy of PM2.5 concentration prediction, this paper proposes a hybrid machine learning model (WD-SA-LSTM-BP model) based on simulated annealing (SA) optimization and wavelet decomposition. Firstly, the wavelet decomposition algorithm was used to realize the multiscale decomposition and single-branch reconstruction of PM2.5 concentrations to weaken the prediction error caused by time series data. Secondly, the SA optimization method was used to optimize the super-parameters of each machine learning model under each reconstructed component, so as to solve the problem that it is difficult to determine the parameters of machine learning model. Thirdly, the optimized machine learning model was used to predict the PM2.5 concentration, and the error value was calculated from the actual measured value. Then, the optimized machine learning model was used to predict the error value. Finally, the predicted error value was added to the predicted PM2.5 concentration to obtain the final predicted PM2.5 concentration. The study is experimentally validated based on daily PM2.5 data collected from Nanjing air quality monitoring stations. The experimental results showed that the RMSE and MAE values of the constructed WD-SA-LSTM-BP model were 5.26 and 3.72, respectively, which were superior to those of the WD-LSTM and LSTM models, indicating that the hybrid machine learning model based on SA optimization and wavelet decomposition could better predict the PM2.5 concentration.

This is a preview of subscription content, log in via an institution to check access.

Access this article

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Fig. 1
Fig. 2
Fig. 3
Fig. 4
Fig. 5
Fig. 6
Fig. 7
Fig. 8
Fig. 9

Similar content being viewed by others

Data availability

The datasets used and/or analyzed during the current study are available from the corresponding author on reasonable request.

References

Download references

Funding

This research was supported in part by the Innovative Research Group project of the National Natural Science Foundation of China (project: Cooperative Control Theory and Application of Autonomous Unmanned System (No. 61921004)).

Author information

Authors and Affiliations

Authors

Contributions

S.H. and P.L. conducted the research. P.L. and Q.W. revised the paper and guided the research. Y.Z. and Y.Q. were responsible for collecting data and creating the figures. Y.Y. revised and improved the paper. All authors have read and agreed to the published version of the manuscript.

Corresponding author

Correspondence to Pengfei Liu.

Ethics declarations

Ethics approval and consent to participate

Not applicable.

Consent for publication

Not applicable.

Competing interests

The authors declare no competing interests.

Additional information

Responsible Editor: Marcus Schulz

Publisher's note

Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Appendix

Appendix

Fig. 10
figure 10

Structure diagram of the three-layer neural network model

10

Fig. 11
figure 11

LSTM unit structure

11,

Fig. 12
figure 12

SVR parameter space optimization result

12,

Fig. 13
figure 13

BP parameter space optimization result

13,

Fig. 14
figure 14

LSTM parameter space optimization result

14

Rights and permissions

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Hu, S., Liu, P., Qiao, Y. et al. PM2.5 concentration prediction based on WD-SA-LSTM-BP model: a case study of Nanjing city. Environ Sci Pollut Res 29, 70323–70339 (2022). https://doi.org/10.1007/s11356-022-20744-7

Download citation

  • Received:

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s11356-022-20744-7

Keywords

Navigation