Back to articles
Article
Volume: 34 | Article ID: COLOR-157
Image
Deep learning approach for classifying contamination levels with limited samples
  DOI :  10.2352/EI.2022.34.15.COLOR-157  Published OnlineJanuary 2022
Abstract
Abstract

The detection of the contaminants in daily food and drinking water is crucial for global public health. For heavy metals detection of Mercury (Hg) and Arsenic (As), our group has proposed a novel paper-based and microfluidic device integrated with a mobile phone and an image analysis pipeline to capture and analyze the sensor images on-site. Still, the detection of lower contamination levels remains challenging due to the small number of available data samples and large intra-class variance of our application. To overcome this challenge, we explore traditional data augmentation and GAN-based augmentation techniques for synthesizing realistic colorimetric images; and we propose a CNN classifier for five-contamination-levels classification. Our proposed system is trained and evaluated on a limited dataset of 126 phone captured images of five contamination levels. Our system yields 88.1% classification accuracy and 91.92% precision, demonstrating the feasibility of this approach. We believe that this approach of training deep learning models on limited detection images datasets presents a clear path toward phone-based contamination-levels detection.

Subject Areas :
Views 106
Downloads 34
 articleview.views 106
 articleview.downloads 34
  Cite this article 

Min Zhao, Susana Diaz-Amaya, Amanda J. Deering, Lia Stanciu, George T.C. Chiu, Jan P. Allebach, "Deep learning approach for classifying contamination levels with limited samplesin Electronic Imaging,  2022,  pp 157-1 - 157-5,  https://doi.org/10.2352/EI.2022.34.15.COLOR-157

 Copy citation
  Copyright statement 
Copyright © 2022, Society for Imaging Science and Technology 2022
ei
Electronic Imaging
2470-1173
2470-1173
Society for Imaging Science and Technology
IS&T 7003 Kilworth Lane, Springfield, VA 22151 USA