MedMNIST 和 MedOOD
收藏资源简介:
MedMNIST是一个多样化的标准医学成像数据集,包含多种成像模态和任务,用于机器学习研究。MedOOD是作者团队新构建的基准数据集,包含各种临床相关的分布转移,用于评估模型在多样化和具有挑战性的条件下的泛化能力。MedOOD数据集涵盖了转换、人口、模态、诊断和器官变化等多个方面的分布转移。
MedMNIST is a diverse standard medical imaging dataset encompassing multiple imaging modalities and tasks, dedicated to machine learning research. MedOOD is a newly constructed benchmark dataset developed by the authors' team, which includes various clinically relevant distribution shifts for evaluating the generalization capability of models under diverse and challenging conditions. The MedOOD dataset covers distribution shifts across multiple dimensions, including transformation, demographic, modality, diagnostic, and organ variations.
MedOODFlow 数据集概述
数据集简介
MedOODFlow是一个基于正常化流(normalizing flows)的医学图像异常检测方法,旨在提高医学成像系统的可靠性。该方法可以在不改变预训练模型权重的情况下进行后处理集成。研究首先在MedMNIST基准数据集上评估了该方法的有效性,并引入了一个新的OOD(Out-of-Distribution)基准数据集,用于进一步验证方法。
数据集构成
- MedMNIST数据集:28x28像素的医学图像数据集,用于初步评估方法的有效性。
- MedOOD数据集:一个专门为医学图像异常检测基准设计的集合,包括来自多种成像模态和临床相关分布偏移的数据集。
数据集获取
- MedMNIST数据集:通过
medmnist包安装后,使用脚本下载。 - MedOOD数据集:准备说明见
medood/README.md文件。
相关资源
- OpenOOD:MedOODFlow项目基于OpenOOD构建。
- 预训练模型:ResNet18在OrganAMNIST上的预训练模型可以从这里下载。
使用说明
- 安装:创建Python 3.10+环境,安装所需包。
- 训练和评估脚本:位于
scripts/ood/nflow文件夹中。 - 训练3D-ResNet18:使用
scripts/basics/brats20_t1/train_brats20_t1.sh脚本。
引用
@article{lotfi2025medoodflow, title={Enhancing Out-of-Distribution Detection in Medical Imaging with Normalizing Flows}, author={Dariush Lotfi and Mohammad-Ali Nikouei Mahani and Mohamad Koohi-Moghadam and Kyongtae Ty Bae}, journal={arXiv preprint arXiv:2502.11638}, year={2025} }

- 1Enhancing Out-of-Distribution Detection in Medical Imaging with Normalizing Flows香港大学李嘉诚医学院诊断放射学系 · 2025年



