Skip to content

JhuangLab/BMSGC

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 

Repository files navigation

BM-Diagnosis: Two-Stage Interpretable Deep Learning for Bone Marrow Disease Classification

License: MIT R-4.3.0 Python 3.9+ Nextflow Snakemake PyTorch Version

Repository: https://github.com/JhuangLab/BMSGC

Principal Investigator: Jhuanglab

Contact: hiekeen $$at$$ gmail.com

The source code will be made available following the manuscript's publication.

Project Overview

Accurate diagnosis of bone marrow diseases from hematoxylin-eosin (HE)-stained whole-slide images (WSIs) remains challenging due to diffuse growth patterns, complex spatial heterogeneity, and overlapping morphological features. This repository implements an interpretable two-stage deep learning framework that closely mirrors the routine pathological diagnostic workflow, enabling automated, robust, and clinically aligned classification of bone marrow disorders.

Key Features

  • 🔀 Two-Stage Diagnostic Pipeline: Mimics real-world clinical workflow Stage 1: Normal vs. Abnormal screening → Stage 2: Disease subtyping
  • 🌐 Region-Aware Graph Modeling: Captures spatial heterogeneity and long-range tissue topology for superior feature aggregation
  • 📉 Distribution-Robust: Minimal performance degradation across temporal and cross-institutional shifts
  • 🔍 Built-in Interpretability: Region-level attention maps highlight diagnostically relevant morphological patterns
  • 🏥 Clinical-Ready Design: Optimized for integration into digital pathology pipelines

Clone the Repository

git clone https://github.com/JhuangLab/BMSGC.git

cd BMSGC

Install Dependencies

Install via pip:

pip install -r requirements.txt

requirements.txt includes:

torch==1.13.1 torchvision==0.14.1 numpy==1.24.3 pandas==2.0.2 matplotlib==3.7.1

scikit-learn==1.2.2 opencv-python==4.7.0.72 pillow==9.5.0 tqdm==4.65.0

lime==0.2.0.1 grad-cam==1.4.6 onnx==1.14.0 onnxruntime==1.15.1 seaborn==0.12.2

License

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

Contact & Citation

Contact

For questions, issues, or collaboration requests, please contact:

Acknowledgements

  • We thank our collaborator for providing clinical data.

  • We acknowledge the open-source community for the foundation models and tools used in this project.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages