Lisanne V. van Dijk | Searcharxiv

arXiv2026

PR3DICTR: A modular AI framework for medical 3D image-based detection and outcome prediction

Three-dimensional medical image data and computer-aided decision making, particularly using deep learning, are becoming increasingly important in the medical field. To aid in these developments we introduce PR3DICTR: Platform for Research in 3D Image Classification and sTandardised tRaining. Built using community-standard distributions (PyTorch and MONAI), PR3DICTR provides an open-access, flexible and convenient framework for prediction model development, with an explicit focus on classification using three-dimensional medical image data. By combining modular design principles and standardization, it aims to alleviate developmental burden whilst retaining adjustability. It provides users with a wealth of pre-established functionality, for instance in model architecture design options, hyper-parameter solutions and training methodologies, but still gives users the opportunity and freedom to ``plug in'' their own solutions or modules. PR3DICTR can be applied to any binary or event-based three-dimensional classification task and can work with as little as two lines of code.

cs.CV↗

arXiv2024

MRI-based Head and Neck Tumor Segmentation Using nnU-Net with 15-fold Cross-Validation Ensemble

The superior soft tissue differentiation provided by MRI may enable more accurate tumor segmentation compared to CT and PET, potentially enhancing adaptive radiotherapy treatment planning. The Head and Neck Tumor Segmentation for MR-Guided Applications challenge (HNTSMRG-24) comprises two tasks: segmentation of primary gross tumor volume (GTVp) and metastatic lymph nodes (GTVn) on T2-weighted MRI volumes obtained at (1) pre-radiotherapy (pre-RT) and (2) mid-radiotherapy (mid-RT). The training dataset consists of data from 150 patients, including MRI volumes of pre-RT, mid-RT, and pre-RT registered to the corresponding mid-RT volumes. Each MRI volume is accompanied by a label mask, generated by merging independent annotations from a minimum of three experts. For both tasks, we propose adopting the nnU-Net V2 framework by the use of a 15-fold cross-validation ensemble instead of the standard number of 5 folds for increased robustness and variability. For pre-RT segmentation, we augmented the initial training data (150 pre-RT volumes and masks) with the corresponding mid-RT data. For mid-RT segmentation, we opted for a three-channel input, which, in addition to the mid-RT MRI volume, comprises the registered pre-RT MRI volume and the corresponding mask. The mean of the aggregated Dice Similarity Coefficient for GTVp and GTVn is computed on a blind test set and determines the quality of the proposed methods. These metrics determine the final ranking of methods for both tasks separately. The final blind testing (50 patients) of the methods proposed by our team, RUG_UMCG, resulted in an aggregated Dice Similarity Coefficient of 0.81 (0.77 for GTVp and 0.85 for GTVn) for Task 1 and 0.70 (0.54 for GTVp and 0.86 for GTVn) for Task 2.

physics.med-ph↗