Predicting Locations of Pollution Sources using Convolutional Neural Networks

Download (PDF)

Licensed according to this deed.

Published on


Pollution is a severe problem today, and the main challenge in water pollution controls and eliminations is detecting and locating pollution sources. This research project aims to predict the locations of pollution sources given diffusion information of pollution in the form of array or image data. These predictions are done using machine learning. The relations between time, location, and pollution concentration are first formulated as pollution diffusion equations, which are partial differential equations (PDEs), and then deep convolutional neural networks are built and trained to solve these PDEs. The convolutional neural networks consist of convolutional layers, reLU layers and pooling layers with chosen parameters. This model is able to solve diffusion equations with an error rate of 5.09 percent. This model of convolutional neural network can be applied to locate pollution sources and is thus helpful for pollution analysis and control.

Cite this work

Researchers should cite this work as follows:

  • Yiheng Chi; Nickolas D Winovich; Guang Lin (2017), "Predicting Locations of Pollution Sources using Convolutional Neural Networks,"

    BibTex | EndNote


  1. undergraduate research
  3. NCN URE
  4. Machine Learning