arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.03379stat.MEcs.LGstat.ML

missForestPredict——面向预测场景的缺失数据填补

missForestPredict -- Missing data imputation for prediction settings

KU Leuven(KU Leuven(荷语天主教鲁汶大学));Leuven Unit for Health Technology Assessment Research (LUHTAR)(鲁汶卫生技术评估研究单元(LUHTAR));School for Public Health and Primary Care, Maastricht University(马斯特里赫特大学公共卫生与基层保健学院);Leiden University Medical Center(莱顿大学医学中心)

Elena Albu, Shan Gao, Laure Wynants, Ben Van Calster

AI总结:

missForestPredict R包针对预测场景改进missForest算法,通过随机森林迭代填补并保存填补模型,在模拟和公开数据集上相比多种填补方法取得有竞争力的结果且计算时间短。

摘要:

预测模型用于基于输入变量预测结果。输入变量中的缺失数据常出现在模型开发阶段和预测阶段。missForestPredict R包提出了一种对missForest填补算法的改进方案,该方案快速、用户友好且专为预测场景量身定制。该算法使用随机森林迭代填补变量,直到满足收敛准则(该准则统一适用于连续变量和分类变量,并基于袋外误差)。每个变量和每次迭代的填补模型都会被保存,之后可在预测阶段应用于新观测数据。missForestPredict包提供了扩展的误差监控、对填补所用变量的控制以及自定义初始化。这使得用户能够根据自身具体需求定制填补过程。missForestPredict算法在八个具有模拟缺失机制的模拟数据集(48种场景)和八个大型公开数据集上,使用不同的预测模型,与均值/众数填补、线性回归填补、mice、k近邻、bagging、miceRanger和IterativeImputer进行了比较。missForestPredict在预测场景中提供了具有竞争力的结果,且计算时间短。

英文摘要:

Prediction models are used to predict an outcome based on input variables. Missing data in input variables often occurs at model development and at prediction time. The missForestPredict R package proposes an adaptation of the missForest imputation algorithm that is fast, user-friendly and tailored for prediction settings. The algorithm iteratively imputes variables using random forests until a convergence criterion (unified for continuous and categorical variables and based on the out-of-bag error) is met. The imputation models are saved for each variable and iteration and can be applied later to new observations at prediction time. The missForestPredict package offers extended error monitoring, control over variables used in the imputation and custom initialization. This allows users to tailor the imputation to their specific needs. The missForestPredict algorithm is compared to mean/mode imputation, linear regression imputation, mice, k-nearest neighbours, bagging, miceRanger and IterativeImputer on eight simulated datasets with simulated missingness (48 scenarios) and eight large public datasets using different prediction models. missForestPredict provides competitive results in prediction settings within short computation times.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑