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

TL;DR

PR3DICTR is an open-source, modular AI framework built on PyTorch and MONAI, designed to simplify and standardize 3D medical image classification and prediction tasks.

Contribution

It introduces a flexible, standardized platform that reduces development effort while allowing customization for 3D medical image analysis.

Findings

Provides extensive pre-built modules for model design and training.

Supports rapid deployment with minimal code, as few as two lines.

Facilitates research and development in 3D medical image classification.

Abstract

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.