Water Quality Sensor Model Based on an Optimization Method of RBF Neural Network


In order to solve the problem that the traditional radial basis function (RBF) neural network is easy to fall into local optimal and slow training speed in the data fusion of multi water quality sensors, an optimization method of RBF neural network based on improved cuckoo search (ICS) was proposed. The method uses RBF neural network to construct a fusion model for multiple water quality sensor data. RBF network can seek the best compromise between complexity and learning ability, and relatively few parameters need to be set. By using ICS algorithm to find the best network parameters of RBF network, the obtained network model can realize the non-linear mapping between input and output of data sample. The data fusion processing experiment was carried out based on the data released by Zhejiang province surface water quality automatic monitoring data system from March to April 2018. Compared with the traditional BP neural network, the experimental results show that the RBF neural network based on gradient descent (GD) and genetic algorithm (GA), the new method proposed in this paper can effectively fuse the water quality data and obtain higher classification accuracy of water quality.

Share and Cite:

Huang, W. and Yang, Y. (2020) Water Quality Sensor Model Based on an Optimization Method of RBF Neural Network. Computational Water, Energy, and Environmental Engineering, 9, 1-11. doi: 10.4236/cweee.2020.91001.

1. Introduction

The problem of water resources pollution and shortage has become a major problem in the development of social and economic development, so it is necessary to monitor the quality of water resources, measure the types of pollutants in water bodies, and concentrate of various pollutants and their changing trends, which provides reference data and basic means for water quality evaluation and prediction. However, the water quality data monitored by water quality sensors in the current automatic monitoring system is all single indicators, and there is no comprehensive evaluation of the monitored section water quality. According to the characteristics of the water quality data detected by the water quality monitoring system, data fusion technology was introduced [1] . Since the data obtained by various water quality sensors is multi-source data, multi-source data fusion processing is required [2] . The neural network can provide an effective network model for data fusion because of its nonlinear mapping ability and strong self-learning ability. It has also achieved remarkable results in some practical applications. Mahmood et al. proposed data fusion for different sensor monitoring data to improve the prediction ability of soil properties [3] . Qiu et al. proposed a multi-layer perceptron (MLP) data fusion algorithm based on ZigBee protocol architecture. The algorithm uses three layer perceptron models and improves the efficiency and total efficiency of the fusion of layers through the idea of cross layer fusion model [4] . Luciano et al. combines RBF neural network with genetic algorithm (GA) as cross-correlation method, and adopts multi-objective differential evolution and adaptive mode to optimize the network, which is used to predict swimmer’s velocity distribution [5] .

In this paper, based on the structural features and nonlinear capability of RBF neural network, the data fusion algorithm of water quality sensor based on ICS optimization RBF is proposed. Cuckoo Search (CS) [6] is used to find the optimal parameters of radial basis center, width and network weight, which solves the problem that the traditional training methods, such as gradient descent method and PSO that are easy to fall into local optimal and slow training speed. The cuckoo algorithm is improved by using adaptive discovery probability and update step length, so that the optimal optimization of RBF network parameters can be achieved more accurately.

2. Water Quality Convergence Factors and Classification Standards

2.1. Water Quality Convergence Factor

The classification of water quality parameters can be divided into biological, chemical and physical parameters based on biological, chemical and physical characteristics of water quality. Even if different water quality parameters are selected for fusion in the same water area, the final classification results may be different.

The criteria for specifying the status of water quality are divided into individual indicators and comprehensive indicators. This paper obtains comprehensive indicators by fusing the data measured by multiple types of water quality sensors to achieve water quality assessment and classification. The data studied in this paper comes from Zhejiang province surface water quality automatic monitoring data system. The total phosphorus, dissolved oxygen, ammonia nitrogen and potassium permanganate monitored by water quality sensors in Ningbo were selected as four fusion factors.

2.2. Classification Standard for Water Quality

The classification criteria adopted in this paper are based on the Surface Water Environmental Quality Standard (GB3838-2002). According to the ground water uses purpose and protection target, surface water can be divided into five categories, which have corresponding functions and purposes. The standard value of each basic item of surface water is divided into five categories, and the different functional categories implement the standard values of the corresponding categories. The standard limits for the fusion factors total phosphorus, dissolved oxygen, ammonia nitrogen, and potassium permanganate used in this paper are shown in Table 1.

This standard is used as the basic standard for data fusion of multiple water quality sensors, and each category corresponds to one output unit of the RBF neural network.

3. Water Quality Sensor Data Fusion Algorithm

3.1. RBF Neural Network

RBF neural network (RBF) is a three-layer feedforward neural network composed of single hidden layer, which has a strong ability to classify and approximate any nonlinear continuous function [7] . The number of hidden layer nodes, that is, the network structure, needs to be determined according to the specific problems of the study, so that the applicability of the network is better.

1) Determination of network structure

Since the number of input fusion factors is 4, the number of input layer nodes in the RBF neural network is 4. The output layer can be divided into five categories according to the water quality category, so the number of output layer nodes is 5. There is no unified method for determining the number of nodes in the hidden layer. If the number of nodes is too large, the training time will be longer, the learning phenomenon will occur, and the network generalization ability and the prediction ability will decrease. Too few nodes result in less nonlinear approximation ability and less fault tolerance. The commonly used empirical formula to determine the number of hidden layer nodes is:

h = n + m + b (1)

where h is the number of hidden layer nodes, n is the number of input layer nodes,

Table 1. Surface water environmental quality standard.

and m is the number of output layer nodes, which is generally a constant of 1 to 10. We train the networks with 4 to 13 nodes respectively. The experimental results show that the training error is the minimum when the number of hidden layer nodes is 12; therefore the number of hidden nodes is 12, which is the best number of nodes and has faster training speed and better generalization ability. The structure of the RBF neural network is shown in Figure 1.

The center of the hidden layer of the RBF neural network selects the Gaussian kernel function, then the output of the ith node of the output layer is:

y i ( X ) = j = 1 h ω i j exp ( 1 2 σ j 2 X c j ) (2)

where X = { x 1 , x 2 , x 3 , x 4 } T represents input from the input layer, ω i j represents the weight relationship between the ith node of the hidden layer and the jth node of the output layer. c j represents the node center of the hidden layer, σ j represents the width of the hidden layer node.

2) Network training methods

RBF neural network parameters including the hidden layer nodes center c j , the width of the hidden layer nodes σ j and the weights between hidden layer and output layer ω i j , network parameters training method can have an important impact on the performance of the network. Currently, the commonly used training methods are: clustering method [8] , gradient descent method [9] , orthogonal least squares algorithm (OLS) [10] [11] , pseudo-inverse method and intelligent search algorithm. In order to solve the problem that the gradient descent

Figure 1. RBF neural network structure.

method is easy to fall into local optimum and slow convergence, the ICS algorithm is used to optimize the network parameters.

3.2. Improved Cuckoo Search Algorithm

The CS algorithm is based on the behavior of cuckoo nest spawning and the characteristics of Levi’s flight. Three idealization criteria need to be assumed:

a) Each cuckoo produces only one egg at a time (that is, has an optimal solution), and randomly selects a host bird’s nest;

b) In a randomly selected nest, only high-quality eggs will be hatched, and the corresponding optimal nest will remain in the next generation;

c) The number of host nests available in each iteration is fixed, and the probability that the host bird nest finds alien cuckoo eggs is found. In this case, the host nest can destroy the egg or abandon the old nest and build a new one.

According to the above three criteria, there are two ways to update the nest position in cuckoo algorithm.

1) Update the position of the nest by Levy flight:

Then, the updated formula of nest position is:

x i t + 1 = x i t + α L ( λ ) , ( i = 1 , 2 , , n ) (3)

x i t is the position of nest i in iteration t; n represents the number of nests; represents the product of elements; α is the step size control factor, which controls the step size of random search, and the expression is:

α = α 0 ( x i t x b e s t t ) (4)

x b e s t t represents the best bird nest position in the tth iteration, and α 0 is a constant of 0.01 by default;

L ( λ ) represents Levy’s random search step size, which is subject to Levy distribution. When executing Levy flight according to Mantegna algorithm, the expression is [12] :

L ( λ ) = u | v | 1 / β (5)

λ = β + 1 , 0 < β < 2 , β affects the random search step size L ( λ ) , the default value is 1.5; u and v are normally distributed.

2) Update the nest position by finding the probability p a :

The nest position updating formula is as follows:

x i t + 1 = x i t + γ H ( p a ε ) ( x j t x k t ) (6)

γ and ε obey the random number uniformly distributed between [0, 1], H ( p a ε ) is the Heviside function, which compares p a with the random number ε to control whether the nest position is updated. The function value is 0 when p a > ε , 0.5 when p a = ε , and 1 when p a < ε ; x j t and x k t are different random vectors in the tth iteration.

The search direction and distance of the standard CS algorithm are randomly selected according to Levi’s flight and can easily jump from one area to another. It has a strong global search capability, but lacks local search capabilities. The fast convergence and precision of the standard CS algorithm cannot be guaranteed. When p a is small and α is large, the number of iterations of the algorithm searching for optimal parameters will increase significantly. When p a is large and α is small, the algorithm converges rapidly, but the global optimal solution cannot be obtained [13] . To solve the problem of the standard CS algorithm, the parameter p a and step length parameter α were improved.

i) Adaptive discovery probability p a

Discovery probability p a is associated with the number of iterative steps. It should gradually decrease as the number of iterations increases. At the initial stage, x should be a large value, and global optimization is carried out to increase the evolutionary strength and diversity of the population. At the end of the iteration, each nest is near the optimal solution, and p a should be set to a smaller value, which is conducive to convergence to the optimal solution, increase the probability of generating a new nest, and avoid falling into a local optimal solution. The cosine function is introduced into the discovery probability, so that p a adaptively changes with the iteration:

p a t = p min + ( p max p min ) × cos ( π 2 × t 1 T 1 ) (7)

p min and p max are respectively the minimum and maximum discovery probability, t is the number of current iterations, and T is the maximum number of iterations.

ii) Step parameter α adaptive update

The step size parameter α is related to the search precision of the algorithm, and the step size should gradually decrease with the increase of the number of iterations. On this basis, for each nest in different iterations, different steps should be used to find new nests, and the fitness value of the nest is introduced into the step length update. The nest with poor fitness value can increase the step length to expand the search area and maintain the diversity of the nest. The nest with good fitness can be reduced so that it can search around the optimal value, which can accelerate the convergence speed. Therefore, the formula of step length update is:

α i t = ( α max ( α max α min ) × t T ) × f i t f a v e r t (8)

α min and α max are respectively the minimum and maximum iteration step sizes; f i t is the fitness value of the ith nest at the tth iteration; f a v e r t is the average value of all nest fitness values in the tth iteration.

f i t = j = 1 a ( y j x j t ( i ) ) 2 (9)

y j represents the target value, and y represents the actual value obtained by the jth nest at the tth iteration.

4. Algorithm Implementation Steps

The RBF neural network optimization method based on ICS can be divided into ICS algorithm part and RBF algorithm part. The part of the ICS algorithm mainly realizes the training and optimization of RBF network parameters, while the part of the RBF algorithm mainly obtains the parameters after the training of the ICS algorithm, and integrates and classifies the test sample data.

The algorithm implementation steps are as follows:

1) The structure of RBF neural network is first determined. The water quality data collected by the water quality sensor are preprocessed to determine the input sample data.

2) Randomly generate n nests. Each nest is composed of the hidden layer center of the RBF network. The width and the network weight of the hidden layer to the output layer, and is coded in the order of [ c , σ , ω ] , and each parameter of the ICS algorithm is set. Each nest is decoded, and the fitness value of each nest is calculated according to Formula (9) to obtain the minimum fitness value and the current optimal bird nest position.

3) Levy flight was used to update the position of the bird’s nest, that is, calculate the new bird’s nest position according to Equation (3) and calculate the fitness value of the new bird’s nest.

4) Comparing the fitness value of the new bird nest with that of the old bird nest. If the fitness value of the new bird nest is smaller than that of the old bird nest, replace the old bird nest with a new one, and a group of better bird nests will be obtained.

5) A set of random Numbers is generated and compared with the discovery probability p a , and the solution larger than p a is eliminated, and the corresponding quantity of new solutions is replaced according to Formula (6).

6) The fitness values of all updated nests were calculated, and the minimum fitness values and corresponding nest positions were found.

7) Determine the iterative termination condition. If the fitness value reaches the target accuracy or the iteration reaches the maximum number, stop the iteration and save the bird nest position corresponding to the current minimum fitness value. If the iteration condition is not satisfied, skip to step 3) to continue the iteration.

8) The final optimal bird nest is decoded as the radial basis, width and network weights of the hidden layer to the output layer of the RBF neural network.

9) Simulation experiments were conducted on the test samples.

5. Experimental Analysis

The experimental simulation data comes from the water quality data monitored by the water quality sensor in the automatic monitoring data system for surface water quality in Zhejiang (from March to April 2018), and 250 sets of data were selected. Each set of data includes four elements: total phosphorus, dissolved oxygen, ammonia nitrogen, and potassium permanganate. 200 sets of data were used as training samples and 50 sets were used as test samples.

5.1. Parameter Settings

The number of nests n = 10 was set in the RBF neural network optimization method based on ICS. The parameter β = 1.5 that affects Levy’s search step size. Since the fitness value changes gradually as the probability of discovery becomes smaller, the minimum and maximum discovery probabilities are set to p min = 0.01 , p max = 0.6 . According to the experiment in literature [14] , the maximum and minimum step length parameters α max = 0.5 and α min = 0.01 were set.

5.2. Experiment Analysis

In order to verify the convergence speed and the classification accuracy of test samples of the RBF neural network optimization method based on Improved Cuckoo Search (ICS), this Algorithm was compared with the classification Algorithm based on BP neural network, the RBF Algorithm based on Gradient Descent (GD) and the RBF Algorithm based on Genetic Algorithm (GA). Because the algorithm proposed in this paper is to optimize the RBF neural network by improving the CS (ICS) algorithm, the proposed algorithm is recorded as ICS-RBF, the classification algorithm using BP neural network is recorded as BP, the RBF algorithm based on GD is recorded as GD-RBF, and RBF algorithm based on GA is recorded as GA-RBF. The performance of these algorithms is compared with the two indicators of fitness value change trained by RBF neural network and test sample prediction accuracy rate.

1) Fitness value

First, the fitness values of the RBF neural network are compared when the samples are trained, and the fitness value is the mean square error of the training sample target output and the actual output. The curve of fitness value change is shown in Figure 2.

It shows that the ICS-RBF algorithm has stronger global search and local search capabilities. When the number of iterations reaches 2000, the fitness value of GA-RBF algorithm is 0.2780, the fitness value of ICS-RBF algorithm was 0.2223, the fitness value of the GD-RBF algorithm is 0.3072, and the fitness value of the BP algorithm is 0.3402. Since the fitness value represents the error between the target output and the actual output of training samples, the fitness value is smaller, which indicates that the prediction performance of RBF network for training samples is better. Therefore, the ICS-RBF algorithm has a better effect on the training of data samples. The trained RBF network model can more accurately predict the training samples and achieve a better non-linear fitting of the data samples to the output.

2) Prediction accuracy of test samples

The RBF network based on four methods is trained by training samples. At this time, 50 sets of test data samples are used to test and verify the three RBF neural network models. The prediction classification effect of single test samples and the accuracy of the prediction classification of the whole test set are compared. Five groups of test sample data and actual sample prediction classification results were selected to compare and analysis. The selected test sample data is shown in Table 2.

After the output results of the four RBF model prediction test samples are normalized, the maximum output probability obtained is added to the corresponding water quality category. The actual output results obtained are shown in Figure 3.

It shows that the predicted output result of RBF neural network based on the ICS algorithm is closer to the target output of the test sample, while the BP neural network model predicts the worst classification effect. When the BP and GD-RBF algorithms predict the data classification of the first group, the output prediction result is class II, which is inconsistent with the target result as class I. Therefore, compared with the other three models, the RBF network based on ICS algorithms has better effect on water quality data fusion processing, and the prediction results are closer to the actual results.

The prediction accuracy of the four network models for test samples is shown in Table 3.

Figure 2. Fitness curve.

Table 2. 5 groups of test sample data.

Figure 3. The actual output of the test sample.

Table 3. Predictive classification accuracy of test samples.

It can be seen from the table that the accuracy of prediction and classification of water quality sensor data fusion based on ICS optimized RBF is 90%, which is 18% higher than that of BP neural network. The accuracy rate is 6% higher than that of RBF network based on GA, and the accuracy rate is 4% higher than that of the RBF network based on GA. Therefore, the optimization method achieved a better classification effect.

6. Conclusion

As a resource for human survival, water resources are directly related to human life safety. Therefore, it is necessary to conduct water quality tests on rivers and lakes. Water quality monitoring can timely, accurately and comprehensively reflect the current status and development trend of water quality, and provide scientific basis for water environment management, pollution source control, and environmental planning, which plays a crucial role. In water quality monitoring, the objective is to integrate data from various water quality sensors. This paper proposes a RBF neural network optimization method based on cuckoo search. The RBF neural network has strong learning ability and generalization ability in small sample data sets, and can well fit the internal nonlinear relationship of the water quality sample data and achieve higher classification accuracy of water quality evaluation. It provides a new approach for the comprehensive evaluation of water quality and has important theoretical and practical significance.

Conflicts of Interest

The authors declare no conflicts of interest.


[1] Yager, R.R. (2004) A Framework for Multi-Source Data Fusion. Elsevier Science Inc., Amsterdam.
[2] Wang, X., Xu, L.Z., Yu, H.Z., et al. (2017) Multi-Source Surveillance Information Fusion Technology and Application. Science.
[3] Mahmood, H.S., Hoogmoed, W.B. and Henten, E.J.V. (2012) Sensor Data Fusion t-o Predict Multiple Soil Properties. Precision Agriculture, 13, 628-645.
[4] Qiu, J., Zhang, L., Fan, T., et al. (2016) Data Fusion Algorithm of Multilayer Neural Network by ZigBee Protocol Architecture. International Journal of Wireless & Mobile Computing, 10, 214-223.
[5] Cruz, L.F.D., Freire, R.Z., Reynoso-Meza, G., et al. (2017) RBF Neural Network Combined with Self-Adaptive MODE and Genetic Algorithm to Identify Velocity Profile of Swimmers. Computational Intelligence, Athens, 6-9 December 2016, 1-7.
[6] Yang, X.S. and Deb, S. (2010) Cuckoo Search via Levy Flights. World Congress on Nature Biologically Inspired Computing, Coimbatore, 9-11 December 2009, 210-214.
[7] Li, M.M. and Verma, B. (2016) Nonlinear Curve Fitting to Stopping Power Data Using RBF Neural Networks. Expert Systems with Applications, 45, 161-171.
[8] Karayiannis, N.B. and Mi, G.W. (1997) Growing Radial Basis Neural Networks: Merging Supervised and Unsupervised Learning with Network Growth Techniques. IEEE Transactions on Neural Networks, 8, 1492.
[9] Naik, B., Nayak, J. and Behera, H.S. (2015) A Novel FLANN with a Hybrid PSO and GA Based Gradient Descent Learning for Classification. In: Proceedings of the 3rd International Conference on Frontiers of Intelligent Computing: Theory and Applications, Springer, Cham, 745-754.
[10] Wang, N., Er, M.J. and Han, M. (2014) Parsimonious Extreme Learning Machine Using Recursive Orthogonal Least Squares. IEEE Transactions on Neural Networks and Learning Systems, 25, 1828-1841.
[11] Yang, Y.W. (2018) Research on Data fusion Algorithm of Water Quality Sensor. Hohai University, Nanjing.
[12] Mantegna, R.N. (1994) Fast, Accurate Algorithm for Numerical Simulation of Lévy Stable Stochastic Processes. Physical Review E: Statistical Physics Plasmas Fluids & Related Interdisciplinary Topics, 49, 4677.
[13] Valian, E., Tavakoli, S., Mohanna, S., et al. (2013) Improved Cuckoo Search for Reliability Optimization Problems. Computers & Industrial Engineering, 64, 459-468.
[14] Valian, E., Mohanna, S. and Tavakoli, S. (2011) Improved Cuckoo Search Algorithm for Global Optimization.

Copyright © 2024 by authors and Scientific Research Publishing Inc.

Creative Commons License

This work and the related PDF file are licensed under a Creative Commons Attribution 4.0 International License.