Roland Theresa, Böck Carl, Tschoellitsch Thomas, Maletzky Alexander, Hochreiter Sepp, Meier Jens, Klambauer Günter
ELLIS Unit Linz, LIT AI Lab, Institute for Machine Learning, Johannes Kepler University Linz, Linz, Austria.
Department of Anesthesiology and Critical Care Medicine, Kepler University Hospital GmbH, Johannes Kepler University Linz, Linz, Austria.
J Med Syst. 2022 Mar 29;46(5):23. doi: 10.1007/s10916-022-01807-1.
Many previous studies claim to have developed machine learning models that diagnose COVID-19 from blood tests. However, we hypothesize that changes in the underlying distribution of the data, so called domain shifts, affect the predictive performance and reliability and are a reason for the failure of such machine learning models in clinical application. Domain shifts can be caused, e.g., by changes in the disease prevalence (spreading or tested population), by refined RT-PCR testing procedures (way of taking samples, laboratory procedures), or by virus mutations. Therefore, machine learning models for diagnosing COVID-19 or other diseases may not be reliable and degrade in performance over time. We investigate whether domain shifts are present in COVID-19 datasets and how they affect machine learning methods. We further set out to estimate the mortality risk based on routinely acquired blood tests in a hospital setting throughout pandemics and under domain shifts. We reveal domain shifts by evaluating the models on a large-scale dataset with different assessment strategies, such as temporal validation. We present the novel finding that domain shifts strongly affect machine learning models for COVID-19 diagnosis and deteriorate their predictive performance and credibility. Therefore, frequent re-training and re-assessment are indispensable for robust models enabling clinical utility.
许多先前的研究声称已经开发出了能通过血液检测诊断新冠肺炎的机器学习模型。然而,我们推测数据潜在分布的变化,即所谓的域转移,会影响预测性能和可靠性,并且是此类机器学习模型在临床应用中失败的一个原因。域转移可能由多种因素引起,例如疾病流行率的变化(传播或检测人群)、改进的逆转录聚合酶链反应(RT-PCR)检测程序(采样方式、实验室程序)或病毒突变。因此,用于诊断新冠肺炎或其他疾病的机器学习模型可能不可靠,并且其性能会随着时间推移而下降。我们研究新冠肺炎数据集中是否存在域转移以及它们如何影响机器学习方法。我们还着手在整个疫情期间以及域转移情况下,基于医院常规采集的血液检测来估计死亡风险。我们通过在大规模数据集上使用不同的评估策略(如时间验证)来评估模型,从而揭示域转移。我们提出了一个新发现,即域转移会强烈影响用于新冠肺炎诊断的机器学习模型,并降低其预测性能和可信度。因此,对于具有临床实用性的稳健模型而言,频繁的重新训练和重新评估是必不可少的。