Machine learning-based estimation of patient body weight from radiation dose metrics in computed tomography

Currently, precise patient body weight (BW) at the time of diagnostic imaging cannot always be used for radiation dose management. Various methods have been explored to address this issue, including the application of deep learning to medical imaging and BW estimation using scan parameters. This study develops and evaluates machine learning-based BW prediction models using 11 features related to radiation dose obtained from computed tomography (CT) scans.

A dataset was obtained from 3996 patients who underwent positron emission tomography CT scans, and training and test sets were established. Dose metrics and descriptive data were automatically calculated from the CT images or obtained from Digital Imaging and Communications in Medicine metadata. Seven machine-learning models and three simple regression models were employed to predict BW using features such as effective diameter (ED), water equivalent diameter (WED), and mean milliampere-seconds. The mean absolute error (MAE) and correlation coefficient between the estimated BW and the actual BW obtained from each BW prediction model were calculated.

Our results found that the highest accuracy was obtained using a light gradient-boosting machine model, which had an MAE of 1.99 kg and a strong positive correlation between estimated and actual BW (ρ = 0.972). The model demonstrated significant predictive power, with 73% of patients falling within a ±5% error range. WED emerged as the most relevant dose metric for BW estimation, followed by ED and sex.

The proposed machine-learning approach is superior to existing methods, with high accuracy and applicability to radiation dose management. The model's reliance on universal dose metrics that are accessible through radiation dose management software enhances its practicality. In conclusion, this study presents a robust approach for BW estimation based on CT imaging that can potentially improve radiation dose management practices in clinical settings.