MIP Candy让医学影像处理更灵活,只需改一个函数就能跑完整流程。
MIP Candy: A Modular PyTorch Framework for Medical Image Processing
- 通过延迟配置机制实现模块自由替换,无需继承类即可定制网络组件。
- 内置交叉验证、区域检测、训练恢复等功能,支持端到端全流程管理。
- 适合需要快速搭建可定制医学影像模型的研究者,尤其适合多中心协作。
医学图像处理需要专门软件来应对高维体数据、异构文件格式和领域特异性训练流程。现有框架或提供底层组件需大量集成,或采用僵化单体流程难以修改。我们提出MIP Candy(MIPCandy),一个开源的PyTorch框架,专为医学图像处理设计。该框架提供从数据加载、训练、推理到评估的完整模块化流程,研究人员仅需实现一个方法(build_network)即可获得可运行的工作流,同时保留对每个组件的精细控制。核心设计为LayerT,一种延迟配置机制,可在不继承类的前提下动态替换卷积、归一化和激活模块。框架还内置k折交叉验证、自动感兴趣区域检测、深度监督、指数移动平均、多前端实验追踪(Weights & Biases、Notion、MLflow)、训练状态恢复及基于商回归的验证分数预测。可扩展的模块生态系统提供符合统一训练-预测模式的预建模型,可无缝集成至核心框架。MIPCandy基于Apache-2.0许可证开源,要求Python 3.12及以上版本。源码与文档见https://github.com/ProjectNeura/MIPCandy。
原文摘要 · Abstract (English)
Medical image processing demands specialized software that handles high-dimensional volumetric data, heterogeneous file formats, and domain-specific training procedures. Existing frameworks either provide low-level components that require substantial integration effort or impose rigid, monolithic pipelines that resist modification. We present MIP Candy (MIPCandy), a freely available, PyTorch-based framework designed specifically for medical image processing. MIPCandy provides a complete, modular pipeline spanning data loading, training, inference, and evaluation, allowing researchers to obtain a fully functional process workflow by implementing a single method, $\texttt{build_network}$, while retaining fine-grained control over every component. Central to the design is $\texttt{LayerT}$, a deferred configuration mechanism that enables runtime substitution of convolution, normalization, and activation modules without subclassing. The framework further offers built-in $k$-fold cross-validation, dataset inspection with automatic region-of-interest detection, deep supervision, exponential moving average, multi-frontend experiment tracking (Weights & Biases, Notion, MLflow), training state recovery, and validation score prediction via quotient regression. An extensible bundle ecosystem provides pre-built model implementations that follow a consistent trainer--predictor pattern and integrate with the core framework without modification. MIPCandy is open-source under the Apache-2.0 license and requires Python~3.12 or later. Source code and documentation are available at https://github.com/ProjectNeura/MIPCandy.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。