A CNN-Based Method for Heavy-Metal Ion Detection

Abstract

Data processing is an essential component of heavy-metal ion detection. Most of the research now uses a conventional data-processing approach, which is inefficient and time-consuming. The development of an efficient and accurate automatic measurement method for heavy-metal ions has practical implications. This paper proposes a CNN-based heavy-metal ion detection system, which can automatically, accurately, and efficiently detect the type and concentration of heavy-metal ions. First, we used square-wave voltammetry to collect data from heavy-metal ion solutions. For this purpose, a portable electrochemical constant potential instrument was designed for data acquisition. Next, a dataset of 1200 samples was created after data preprocessing and data expansion. Finally, we designed a CNN-based detection network, called HMID-NET. HMID-NET consists of a backbone and two branch networks that simultaneously detect the type and concentration of the ions in the solution. The results of the assay on 12 sets of solutions with different ionic species and concentrations showed that the proposed HMID-NET algorithm ultimately obtained a classification accuracy of 99.99% and a mean relative error of 8.85% in terms of the concentration.