Machine learning in the prediction of cardiac epicardial and mediastinal fat volumes

TL;DR

This study demonstrates that machine learning regression algorithms can accurately predict cardiac fat volumes from CT images, reducing the need for extensive segmentation and potentially lowering healthcare costs.

Contribution

The paper introduces a novel approach using regression algorithms to predict cardiac fat volumes, simplifying the process and maintaining high accuracy compared to traditional segmentation methods.

Findings

High correlation coefficients achieved (up to 0.9876) for fat volume prediction.

Regression models can accurately predict one fat volume from the other, reducing segmentation effort.

Linear regressors provide interpretable models with acceptable accuracy.

Abstract

We propose a methodology to predict the cardiac epicardial and mediastinal fat volumes in computed tomography images using regression algorithms. The obtained results indicate that it is feasible to predict these fats with a high degree of correlation, thus alleviating the requirement for manual or automatic segmentation of both fat volumes. Instead, segmenting just one of them suffices, while the volume of the other may be predicted fairly precisely. The correlation coefficient obtained by the Rotation Forest algorithm using MLP Regressor for predicting the mediastinal fat based on the epicardial fat was 0.9876, with a relative absolute error of 14.4% and a root relative squared error of 15.7%. The best correlation coefficient obtained in the prediction of the epicardial fat based on the mediastinal was 0.9683 with a relative absolute error of 19.6% and a relative squared error of 24.9%. Moreover, we analysed the feasibility of using linear regressors, which provide an intuitive interpretation of the underlying approximations. In this case, the obtained correlation coefficient was 0.9534 for predicting the mediastinal fat based on the epicardial, with a relative absolute error of 31.6% and a root relative squared error of 30.1%. On the prediction of the epicardial fat based on the mediastinal fat, the correlation coefficient was 0.8531, with a relative absolute error of 50.43% and a root relative squared error of 52.06%. In summary, it is possible to speed up general medical analyses and some segmentation and quantification methods that are currently employed in the state-of-the-art by using this prediction approach, which consequently reduces costs and therefore enables preventive treatments that may lead to a reduction of health problems.