Skip to content

MediSwarm is an open-source project for advancing medical deep learning using swarm intelligence and NVFlare. Developed by the Odelia consortium, it ensures data privacy through federated learning, enabling collaborative, decentralized model training across institutions for improved medical research and applications.

License

Notifications You must be signed in to change notification settings

KatherLab/MediSwarm

Repository files navigation

MediSwarm

Introduction

MediSwarm is an open-source project dedicated to advancing medical deep learning through swarm intelligence, leveraging the NVFlare platform. Developed in collaboration with the Odelia consortium, this repository aims to create a decentralized and collaborative framework for medical research and applications.

Key Features

  • Swarm Learning: Utilizes swarm intelligence principles to improve model performance and adaptability.
  • NVFlare Integration: Built on NVFlare, providing robust and scalable federated learning capabilities.
  • Data Privacy: Ensures data security and compliance with privacy regulations by keeping data local to each institution.
  • Collaborative Research: Facilitates collaboration among medical researchers and institutions for enhanced outcomes.
  • Extensible Framework: Designed to support various medical applications and easily integrate with existing workflows.

Prerequisites

Hardware recommendations

  • 64 GB of RAM (32 GB is the absolute minimum)
  • 16 CPU cores (8 is the absolute minimum)
  • an NVIDIA GPU with 48 GB of RAM (24 GB is the minimum)
  • 8 TB of Storage (4 TB is the absolute minimum)

We demonstrate that the system can run on lightweight hardware like this. For less than 10k EUR, you can configure systems from suppliers like Lambda, Dell Precision, and Dell Alienware.

Operating System

  • Ubuntu 20.04 LTS

Usage for Developers

Setup

  1. Clone the repository:

    git clone https://github.com/KatherLab/MediSwarm.git
    cd MediSwarm

Running the Application

  1. Run the minimal CIFAR-10 example: See cifar10/README.md

  2. Run the 3D CNN for classifying breast tumors: See 3dcnn_ptl/README.md

License

This project is licensed under the MIT License - see the LICENSE file for details.

Maintainers

Jeff Ole Schwen Steffen Renisch

Contributing

Feel free to dive in! Open an issue or submit pull requests.

Credits

This project utilizes platforms and resources from the following repositories:

  • NVFLARE: NVFLARE (NVIDIA Federated Learning Application Runtime Environment) is an open-source framework that provides a robust and scalable platform for federated learning applications. We have integrated NVFLARE to efficiently handle the federated learning aspects of our project.

Special thanks to the contributors and maintainers of these repositories for their valuable work and support.


For more details about NVFLARE and its features, please visit the NVFLARE GitHub repository.

About

MediSwarm is an open-source project for advancing medical deep learning using swarm intelligence and NVFlare. Developed by the Odelia consortium, it ensures data privacy through federated learning, enabling collaborative, decentralized model training across institutions for improved medical research and applications.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published