Convolutional neural network for soil microplastic contamination screening using infrared spectroscopy

Abstract Microplastics are emerging pollutants that exist in our environment. Microplastics are synthetic polymers that have particles size smaller than 5 mm. Rapid screening of microplastics contamination in the soil could assist in identifying anomalous concentrations of microplastics in the terrestrial environment. Because there is no rule on the maximum concentration limit on how much microplastics can exist within the soil, the concentration of microplastics collected from industrial areas around metropolitan Sydney was used as a baseline. Spectra obtained from the visible-near-infrared (vis-NIR) spectra has been shown to be feasible in predicting microplastics in the soil. Instead of creating a regression model predicting the concentration of microplastic, a classification model for screening was proposed. A Convolutional Neural Network (CNN) model was trained to classify the soil sample into various degrees of contamination based on concentration. We also delved into the CNN model to understand how the CNN model classifies the spectral data input. The model performance was first tested on two levels of classification (contaminated vs. non-contaminated). The model was able to classify the uncontaminated samples into the appropriate class more accurately than the contaminated samples. When the number of classes were gradually increased, the classification accuracy for the higher level of contaminated samples improved. Transfer learning CNN model further improved the classification prediction only on the extremes, but not the intermediate classes.