Effective Dose Estimation in Computed Tomography by Machine Learning

Background: Computed tomography scans are widely used in everyday medical practice due to speed, image reliability, and detectability of a wide range of pathologies. Each scan exposes the patient to a radiation dose, and performing a fast estimation of the effective dose (E) is an important step for radiological safety. The aim of this work is to estimate E from patient and CT acquisition parameters in the absence of a dose-tracking software exploiting machine learning. Methods: In total, 69,037 CT acquisitions were collected with the dose-tracking software (DTS) available at our institution. E calculated by DTS was chosen as the target value for prediction. Different machine learning algorithms were selected, optimizing parameters to achieve the best performance for each algorithm. Effective dose was also estimated using DLP and k-factors, and with multiple linear regression. Mean absolute error (MAE, mean absolute percentage error (MAPE), and R2 were used to evaluate predictions in the test set and in an external dataset of 3800 acquisitions. Results: The random forest regressor (MAE: 0.416 mSv; MAPE: 7%; and R2: 0.98) showed best performances over the neural network and the support vector machine. However, all three machine learning algorithms outperformed effective dose estimation using k-factors (MAE: 2.06; MAPE: 26%) or multiple linear regression (MAE: 0.98; MAPE: 44.4%). The random forest regressor on the external dataset showed an MAE of 0.215 mSv and an MAPE of 7.1%. Conclusions: Our work demonstrated that machine learning models trained with data calculated by a dose-tracking software can provide good estimates of the effective dose just from patient and scanner parameters, without the need for a Monte Carlo approach.