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

Journal: arXiv
Published Date:

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.

Authors

  • Daniel C. MacRae; Luuk van der Hoek; Robert van der Wal; Suzanne P. M. de Vette; Hendrike Neh; Baoqiang Ma; Peter M. A. van Ooijen; Lisanne V. van Dijk