Lung mass density prediction using machine learning based on ultrasound surface wave elastography and pulmonary function testing
Boran Zhou, Brian J. Bartholmai, Sanjay Kalra, Thomas G. Osborn, Xiaoming Zhang
Abstract
OBJECTIVE: The objective of this study is to predict in vivo lung mass density for patients with interstitial lung disease using different gradient boosting decision tree (GBDT) algorithms based on measurements from lung ultrasound surface wave elastography (LUSWE) and pulmonary function testing (PFT). METHODS: Age and weight of study subjects (57 patients with interstitial lung disease and 20 healthy subjects), surface wave speeds at three vibration frequencies (100, 150, and 200 Hz) from LUSWE, and predicted forced expiratory volume (FEV1% pre) and ratio of forced expiratory volume to forced vital capacity (FEV1%/FVC%) from PFT were used as inputs while lung mass densities based on the Hounsfield Unit from high resolution computed tomography (HRCT) were used as labels to train the regressor in three GBDT algorithms, XGBoost, CatBoost, and LightGBM. 80% (20%) of the dataset was used for training (testing). RESULTS: The results showed that predictions using XGBoost regressor obtained an accuracy of 0.98 in the test dataset. CONCLUSION: The obtained results suggest that XGBoost regressor based on the measurements from LUSWE and PFT may be able to noninvasively assess lung mass density in vivo for patients with pulmonary disease.