- Open Access
Improving precision of glomerular filtration rate estimating model by ensemble learning
Journal of Translational Medicine volume 15, Article number: 231 (2017)
Accurate assessment of kidney function is clinically important, but estimates of glomerular filtration rate (GFR) by regression are imprecise.
We hypothesized that ensemble learning could improve precision. A total of 1419 participants were enrolled, with 1002 in the development dataset and 417 in the external validation dataset. GFR was independently estimated from age, sex and serum creatinine using an artificial neural network (ANN), support vector machine (SVM), regression, and ensemble learning. GFR was measured by 99mTc-DTPA renal dynamic imaging calibrated with dual plasma sample 99mTc-DTPA GFR.
Mean measured GFRs were 70.0 ml/min/1.73 m2 in the developmental and 53.4 ml/min/1.73 m2 in the external validation cohorts. In the external validation cohort, precision was better in the ensemble model of the ANN, SVM and regression equation (IQR = 13.5 ml/min/1.73 m2) than in the new regression model (IQR = 14.0 ml/min/1.73 m2, P < 0.001). The precision of ensemble learning was the best of the three models, but the models had similar bias and accuracy. The median difference ranged from 2.3 to 3.7 ml/min/1.73 m2, 30% accuracy ranged from 73.1 to 76.0%, and P was > 0.05 for all comparisons of the new regression equation and the other new models.
An ensemble learning model including three variables, the average ANN, SVM, and regression equation values, was more precise than the new regression model. A more complex ensemble learning strategy may further improve GFR estimates.
Determining glomerular filtration rate (GFR) is essential for screening early impairment of kidney function, evaluating progressive kidney deterioration and complications, adjusting the dosage of drugs, and managing the risks of chronic kidney disease (CKD) . The gold standard GFR assays are inulin clearance and isotope imaging. The high cost of inulin limits its routine clinical use. Isotope imaging is moderately costly, but usage is limited by radiation exposure. Consequently, GFR is often estimated by regression analysis with sex, age and specific serum filtration markers (e.g., serum creatinine and cystatin C) as covariates. However, even the most accurate GFR estimation equations  are biased and imprecise  because they do not include physiological variables that only affect serum filtration markers other than GFR . The performance of equations that estimate GFR may be improved either by adding additional covariates, or by applying more sophisticated or advanced method than traditional regression.
Ensemble learning is a simple, well-known machine learning method in which the results obtained with different mathematic models is combined to generate a better result. It is widely used in engineering applications because of its superior performance in estimation . A previous study found that ensemble learning significantly outperformed single learning models for cataract detection and grading . We hypothesized that performance of regression-based estimated GFR models can be improved by ensemble learning because it is more robust, allowing more complex relationship between GFR and covariates and can account for more complicated interactions among the covariates.
The study enrolled consecutive participants in the Third Affiliated Hospital of Sun Yat-sen University, China. Some of the data evaluated in this study was previously evaluated and published . Additional data obtained from participants with the same inclusion and exclusion criteria became available  and is shown in Additional file 1: Fig. S1. Of the 1815 participants screened, 1419 were included in the study and data collected from 1002 participants from January 2010 to June 2013 were evaluated in the development set. Data obtained from 417 participants from January 2005 to December 2009 were evaluated in the external validation set. The Institutional Review Board at the Third Affiliated Hospital of Sun Yat-sen University approved the study, and participants gave informed consent before inclusion.
GFR was measured by 99mTc-DTPA renal dynamic imaging, and calibrated with a dual plasma sample 99mTc-DTPA GFR as previously described: dual plasma sample DTPA GFR (ml/min/1.73 m2) = (0.167 + 1.057) × DTPA by renal dynamic imaging − GFR . Serum creatinine was by an enzymatic method with a Hitachi 7180 auto analyzer (Hitachi, Tokyo, Japan) and reagents from Roche Diagnostics (Mannheim, Germany). After the year 2010 serum creatinine was recalibrated to comply with the isotope dilution mass spectrometry reference method.
The GFR estimation models were constructed using serum creatinine, age, and sex as covariates and eGFR as output. The three methods used were regression, artificial neural network (ANN) and support vector machine (SVM), and all generated mathematic models to describe the relationship between GFR and the covariates. ANN and SVM are widely used machine learning methods that can accommodate complex and nonlinear relationships that regression cannot. These three models were constructed using the development data set were used to estimate individual GFRs. The ensemble learning model was generated by averaging the output of the three single models. The construction of these models is described in detail in Table 1, Additional file 2: Item S1, Additional file 3: Item S2 and Additional file 4: Item S3. The performance of these four models was evaluated with the external validation data set.
The bias, precision and accuracy of the models were evaluated. Bias was the median difference of the measured GFR (mGFR) and eGFR; precision was the interquartile range (IQR) of the difference. Accuracy was the proportion of participants whose eGFR did not deviate more than 30% from the mGFR. The 95% confidence intervals (CIs) were calculated by the bootstrap method (2000 bootstraps) . The regression model was used as the benchmark for comparison with the ANN, SVM, and ensemble models. Significance testing of the differences in performance was by the Wilcoxon signed rank test for medians, the bootstrap method for IQR, and the McNemar test for accuracy. Statistical significance level was P < 0.05. All statistical calculations were performed using R version 3.0.2 and MATLAB version 2011b.
The baseline characteristics of the two datasets differed significantly (Table 2). The mean (± SD) mGFR was 70 ± 30 (range 10–174) ml/min/1.73 m2 in the development dataset compared with (53 ± 27 ml/min/1.73 m2, P < 0.001) in the external validation dataset. Mean age, body mass index and body weight were also lower in the developmental dataset, mean serum creatinine level and height were higher. There were fewer patients with diabetes and more men in the external validation data set.
Overall performance of GFR estimation models in the external validation data set
As shown in Table 3, the ensemble model was more precise (13.5 ml/min/1.73 m2) than the regression equation (14.0 ml/min/1.73 m2, P < 0.001). All models had similar bias and accuracy. Bias ranged from 2.3 to 3.7 ml/min/1.73 m2 and accuracy ranged from 73.1 to 76.0% (P > 0.05 for all comparisons between the regression equation and the other models).
Subgroup analysis of performance of GFR estimation models in the external validation data set
The performance of the estimation models was also evaluated in subgroups of the external validation data set stratified by GFR (< 30, 30–60, and ≥ 60 ml/min/1.73 m2). The precision of the ensemble model was consistently better than that of the regression model (8.9 vs. 9.2 with GFR < 30, 12.7 vs. 13.5 with GFR 30–60, and 17.9 vs. 19.6 with GFR ≥ 60, all P < 0.001). Bias and accuracy were similar for all models across all GFR subgroups. There were no significant difference between the regression equation and the other models, except for the SVM model, in which bias was worse than the regression model in the GFR < 30 subgroup.
Regression, ANN, SVM, and ensemble learning models were developed to estimate GFR using sex, age, and serum creatinine as covariates. A single demonstration population was used to develop the models, and a single external validation data set was used to compare performance with that of the regression model. Precision was improved in the ensemble learning method, but bias and accuracy in the regression and the ensemble learning models were similar.
Ensemble learning has previously been used to estimate GFR. Inker et al. evaluated cystatin C as an additional covariate , finding that a model averaging creatinine and cystatin C values had similar bias (median difference) but higher precision (IQR of difference) and accuracy (the proportion of participants whose eGFR deviated from mGFR by more than 20 or 30%) than either of the two covariates used alone. In a later study, Inker et al.  used ensemble learning to evaluate β-trace protein (BTP) and β2-microglobulin (B2M) as covariates to estimate GFR. Their existing CKD-EPI creatinine–cystatin C equation, and a new equation including BTP and B2M were evaluated. The precision and accuracy of average of the two equations was similar to those of the individual equations. The ensemble learning method in those two studies used different combinations of covariates. In this study, ensemble learning employed different models all with the same covariates. There was no need to perform additional testing to measure additional covariates from each patient, which did not increase patient costs.
Why did the ensemble learning model perform better than the regression model? ANN and SVM models could have relatively low bias relative to variance [11, 12]. A good single model might be constructed from theoretical data, but the expected performance might not be realized with the datasets and parameter values used during the training process. A single model might, unfortunately have a high variance, but combining several models should decrease the overall variance, as the deviations of the single models would be in different directions . Ensemble learning thus results in a high level of performance together with improved variance.
Ensemble learning performed better than the other models in GFR estimation even though simple averaging was used. Other, methods more complex than averaging such as weighted averaging or stacking have been used . Planning for ensemble learning when designing single models will achieve increase diversity, leading increased performance improvement after model combination. It is to be expected that using more complicated ensemble learning strategies will further improve performance in GFR estimation.
The fact that the ensemble model achieved better precision but worse bias and accuracy than current regression model, demonstrated that it was difficult to develop a new model superior to regression model in all terms of bias, precision and accuracy, especially with only three covariates. Recently deep learning has made big success in medical informatics, but the constructed models are complicated with many covariates. That is, the advantages of complicated models are obvious in dealing with complex problem such as natural language processing, image and speech recognition. However, GFR estimation as a simple regression task limits the capacity of complicated models. It’s supposed that adding more relevant but different covariates may be more promising than improving method of model construction.
The study limitations include the evaluation of data representing a specific group of CKD patients in China. Also, although there were some differences in the characteristics of the patients in the development and the external validation cohorts, both were recruited at same institution. Consequently, external validation of the effectiveness of the ensemble learning method that was used is desirable.
A novel ensemble learning model was developed to estimate GFR, and it achieved better precision than the widely used regression model. The advantage of ensemble learning method in estimation of GFR warrants further investigation.
Levey AS, Coresh J. Chronic kidney disease. Lancet. 2012;379:165–80.
Inker LA, Schmid CH, Tighiouart H, et al. Estimating glomerular filtration rate from serum creatinine and cystatin C. N Engl J Med. 2012;367:20–9.
Levey AS, Inker LA, Coresh J. GFR estimation: from physiology to public health. Am J Kidney Dis. 2014;63:820–34.
Liu X, Foster MC, Tighiouart H, et al. Non-GFR determinants of low-molecular-weight serum protein filtration markers in CKD. Am J Kidney Dis. 2016;68:892–900.
Polikar R. Ensemble based systems in decision making. IEEE Circuits Syst Mag. 2006;6(3):21–45.
Yang JJ, Li J, Shen R, et al. Exploiting ensemble learning for automatic cataract detection and grading. Comput Methods Programs Biomed. 2016;124:45–57.
Liu X, Li NS, Lv LS, et al. A comparison of the performances of an artificial neural network and a regression model for GFR estimation. Am J Kidney Dis. 2013;62:1109–15.
Liu X, Gan X, Chen J, et al. A new modified CKD-EPI equation for Chinese patients with type 2 diabetes. PLoS ONE. 2014;9:e109743.
Efron B, Tibshirani RJ. An introduction to the bootstrap. New York: Chapman and Hall; 1993.
Inker LA, Tighiouart H, Coresh J, et al. GFR estimation using beta-trace protein and beta2-microglobulin in CKD. Am J Kidney Dis. 2016;67:40–8.
Haykin S. Neural networks and learning machines. 3rd ed. London: Pearson Education, Inc; 2008.
Mitchell T. Machine learning. New York City: McGraw Hill; 1997.
XL and NSL wrote manuscript, researched data. LSL, HL and TQL reviewed/edited manuscript. YMF, CLC, CXC, YQY and SML researched data. All authors read and approved the final manuscript.
We would like to thank all the doctors, nurses, technicians, and patients involved in this study for their cooperation. We also thank Dr. David Cushley from International Science Editing, Clare, Ireland for his assistance with the English version of the manuscript.
The authors declare that they have no competing interests.
Availability of data and materials
The datasets used and/or analyzed during the current study are available from the corresponding author on reasonable request.
Consent for publication
Ethics approval and consent to participate
The Institutional Review Board at the Third Affiliated Hospital of Sun Yat-sen University approved the study, and informed consent had been obtained before the study.
The National Natural Science Foundation of China (Grant Nos. 81370866 and 81070612), the China Postdoctoral Science Foundation (Grant No. 201104335), Guangdong Science and Technology Plan (Grant Nos. 2011B031800084 and 2013B021800190), the Fundamental Research Funds for the Central Universities (Grant No. 11ykpy38), the National Project of Scientific and Technical Supporting Programs Funded by Ministry of Science and Technology of China (Grant No. 2011BAI10B00).
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
About this article
Cite this article
Liu, X., Li, N., Lv, L. et al. Improving precision of glomerular filtration rate estimating model by ensemble learning. J Transl Med 15, 231 (2017). https://doi.org/10.1186/s12967-017-1337-y
- Chronic kidney disease
- Glomerular filtration rate
- Ensemble learning