Geometric deep learning adapted to prediction of liver resection zone

Abstract

Due to patient-specific anatomical variations in the presence of liver cancer, resection planning can be complex requiring thorough preoperative planning. In addition to the calculation of the Future Liver Remnant, the assessment of any vascular and biliary structures that may be at risk is essential to minimize postoperative morbidity and mortality. Despite the progress of modern technologies, this resection planning is still mostly performed mentally, but can be supported by volumetric calculations or planning on a three-dimensional (3D) model.

The aim of this work is to investigate the effectiveness of geometric deep learning (DL) in predicting liver resection zones. We adopted a geometric DL framework, specifically RandLA-Net, a lightweight and efficient neural network designed for semantic segmentation of large-scale 3D point clouds, to support surgical planning for liver tumor resections using 3D geometric data, presented in either mesh or point cloud format. RandLA-Net can process up to one million points in a single pass and operates up to 200 times faster than comparable frameworks, making it particularly well suited for high-resolution anatomical data in clinical settings.

We conducted our experiment in two stages. In the first stage, the pilot study, we evaluated two geometric deep learning models in combination with four different loss functions: Cross-Entropy (CE), Dice coefficient (DICE), Intersection over Union (IoU), and a hybrid loss (a combination of CE and IoU) to efficiently predict the resection volume. Among all the configurations tested, RandLA-Net combined with hybrid loss achieved the best performance. In the second stage, the extended study, we increased the dataset size and repeated the experiment using the best-performing configuration identified in the pilot study, with minor modifications. The extended study demonstrated improved performance, with a mean IoU of 0.76, F1-score of 0.84, precision of 0.86, and recall of 0.82.