This project builds a comprehensive ML + Generative AI workflow using the TCGA-BRCA (Breast Invasive Carcinoma) dataset. The goal is to:
- Predict Survival: Use Histopathology images (WSIs/Patches) to predict patient outcomes.
- Multimodal Integration: Refine predictions by integrating clinical tabular data.
- Preventive Gen AI: Generate "early-stage" disease images from late-stage samples to visualize disease progression "in reverse".
tcga-brca-genai/
├── data/
│ ├── raw/ # Original data (not committed)
│ ├── processed/ # Preprocessed patches and tensors
├── src/
│ ├── data/ # Data loaders and preprocessing
│ ├── models/ # Deep Learning architectures
│ ├── train/ # Training and validation loops
│ └── utils/ # Helper functions
├── notebooks/ # Experiments and analysis
├── docs/ # Methodology and citations
└── requirements.txt
- Clone the repository.
- Install dependencies:
pip install -r requirements.txt
- (Optional) Install OpenSlide binaries if on Mac:
brew install openslide
Data is sourced from the GDC Portal (TCGA-BRCA).
- Images: Whole Slide Images (WSIs) formatted as
.svs. - Clinical: Tabular data with survival information. Note: This project uses ROI patches extracted from WSIs for computational efficiency.
(To be updated as the project progresses)
- Phase 1: Image-based survival modeling (CNN/ViT).
- Phase 2: Clinical + Image Fusion.
- Phase 3: Generative adversarial/diffusion models for stage translation.
This project is licensed under the MIT License - see LICENSE for details.
- The Cancer Genome Atlas (TCGA) for providing the BRCA dataset
- PyTorch and MONAI communities for excellent tools
- Research community for foundational work in computational pathology
For questions or collaborations, please open an issue or contact [kmoralfig@gmail.com].
Note: This project is for research purposes only and should not be used for clinical decision-making without proper validation and regulatory approval.