PR3DICTR: A modular AI framework for medical 3D image-based detection and outcome prediction
This framework addresses the need for efficient and flexible tools in medical imaging research, though it is incremental as it builds on existing community standards.
The authors tackled the challenge of developing deep learning models for 3D medical image classification by introducing PR3DICTR, a modular and standardized framework built on PyTorch and MONAI, which reduces developmental burden and allows customization with minimal code.
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.