Deep Learning-based Prediction of Clinical Trial Enrollment with Uncertainty Estimates
This work addresses the challenge of accurate enrollment prediction for clinical trial planners, though it appears incremental as it builds on existing deep learning and probabilistic techniques.
The paper tackled the problem of predicting patient enrollment in clinical trials, a key factor for trial success, by proposing a deep learning method that combines language and tabular features with uncertainty estimation, and demonstrated it outperformed baseline models on real-world data.
Clinical trials are a systematic endeavor to assess the safety and efficacy of new drugs or treatments. Conducting such trials typically demands significant financial investment and meticulous planning, highlighting the need for accurate predictions of trial outcomes. Accurately predicting patient enrollment, a key factor in trial success, is one of the primary challenges during the planning phase. In this work, we propose a novel deep learning-based method to address this critical challenge. Our method, implemented as a neural network model, leverages pre-trained language models (PLMs) to capture the complexities and nuances of clinical documents, transforming them into expressive representations. These representations are then combined with encoded tabular features via an attention mechanism. To account for uncertainties in enrollment prediction, we enhance the model with a probabilistic layer based on the Gamma distribution, which enables range estimation. We apply the proposed model to predict clinical trial duration, assuming site-level enrollment follows a Poisson-Gamma process. We carry out extensive experiments on real-world clinical trial data, and show that the proposed method can effectively predict the number of patients enrolled at a number of sites for a given clinical trial, outperforming established baseline models.