Am J Epidemiol. 2021 Sep 1;190(9):1830-1840. doi: 10.1093/aje/kwab010.
Although variables are often measured with error, the impact of measurement error on machine-learning predictions is seldom quantified. The purpose of this study was to assess the impact of measurement error on the performance of random-forest models and variable importance. First, we assessed the impact of misclassification (i.e., measurement error of categorical variables) of predictors on random-forest model performance (e.g., accuracy, sensitivity) and variable importance (mean decrease in accuracy) using data from the National Comorbidity Survey Replication (2001-2003). Second, we created simulated data sets in which we knew the true model performance and variable importance measures and could verify that quantitative bias analysis was recovering the truth in misclassified versions of the data sets. Our findings showed that measurement error in the data used to construct random forests can distort model performance and variable importance measures and that bias analysis can recover the correct results. This study highlights the utility of applying quantitative bias analysis in machine learning to quantify the impact of measurement error on study results.
尽管变量通常会带有误差进行测量,但测量误差对机器学习预测的影响很少被量化。本研究的目的是评估测量误差对随机森林模型性能和变量重要性的影响。首先,我们使用来自国家共病调查再调查(2001-2003 年)的数据评估了预测变量的分类错误(即分类变量的测量误差)对随机森林模型性能(例如准确性、敏感性)和变量重要性(准确性平均下降)的影响。其次,我们创建了模拟数据集,我们知道真实的模型性能和变量重要性度量,可以验证定量偏差分析在数据集的分类版本中是否能恢复真实情况。我们的发现表明,用于构建随机森林的数据中的测量误差会扭曲模型性能和变量重要性度量,而偏差分析可以恢复正确的结果。本研究强调了在机器学习中应用定量偏差分析来量化测量误差对研究结果的影响的实用性。