论文标题
对使用机器学习的不良事件预测中涉及时轴的数据拆分策略的调查
Investigation of a Data Split Strategy Involving the Time Axis in Adverse Event Prediction Using Machine Learning
论文作者
论文摘要
不良事件是药物开发中的一个严重问题,并且已经开发了许多使用机器学习的预测方法。随机拆分交叉验证是机器学习中模型构建和评估的事实上的标准,但是在不利事件预测中应注意,因为这种方法与现实世界的情况不符。使用时轴的时间拆分被认为适用于现实世界预测。但是,由于缺乏可比的研究,使用时间和随机分裂获得的模型性能差异尚不清楚。为了了解差异,我们使用九种类型的复合信息作为输入,八个不良事件作为目标和六种机器学习算法进行了比较时间和随机分裂之间的模型性能。在曲线值下,随机分裂显示的面积比八个目标中的六个时间分配比分裂更高。训练和测试数据集的化学空间相似,这表明适用性域的概念不足以解释从分裂中得出的差异。对于蛋白质相互作用,曲线差异下的面积比其他数据集更小。随后的详细分析表明,在分裂时间内使用基于知识的信息的危险。这些发现表明,在不利事件预测中了解时间和随机分裂之间的差异的重要性,并强烈表明适当使用分裂策略和结果的解释对于不利事件的现实预测是必要的。我们提供本研究中使用的分析代码和数据集(https://github.com/mizuno-group/ae_prediction)。
Adverse events are a serious issue in drug development and many prediction methods using machine learning have been developed. The random split cross-validation is the de facto standard for model building and evaluation in machine learning, but care should be taken in adverse event prediction because this approach does not match to the real-world situation. The time split, which uses the time axis, is considered suitable for real-world prediction. However, the differences in model performance obtained using the time and random splits are not clear due to the lack of the comparable studies. To understand the differences, we compared the model performance between the time and random splits using nine types of compound information as input, eight adverse events as targets, and six machine learning algorithms. The random split showed higher area under the curve values than did the time split for six of eight targets. The chemical spaces of the training and test datasets of the time split were similar, suggesting that the concept of applicability domain is insufficient to explain the differences derived from the splitting. The area under the curve differences were smaller for the protein interaction than for the other datasets. Subsequent detailed analyses suggested the danger of confounding in the use of knowledge-based information in the time split. These findings indicate the importance of understanding the differences between the time and random splits in adverse event prediction and strongly suggest that appropriate use of the splitting strategies and interpretation of results are necessary for the real-world prediction of adverse events. We provide analysis code and datasets used in the present study (https://github.com/mizuno-group/AE_prediction).