Pan-Organ Tumor Detection on Whole Slide Images

Computer VisionCancer ResearchCNN

About the Research

This was a prestigious collaborative research project between IIT Bombay Medical Lab and the Tata Memorial Cancer Center. The goal was to develop a robust, generalizable computer vision model capable of detecting tumors across multiple distinct human organs using high-resolution, gigapixel Whole Slide Images (WSIs).

My Role & Contributions

As a Machine Learning Researcher, I led the model architecture design and the complex data preprocessing pipelines required for handling massive WSI files. I worked closely with medical professionals to ensure the output was clinically interpretable.

Technical Challenges Solved

  • Fine-tuned KimiaNet (a histopathology-pretrained CNN) utilizing a DenseNet-121 backbone, achieving an 89% AUC on the multi-organ test set.
  • Engineered an adaptive patch extraction pipeline capable of parsing gigapixel WSIs efficiently without exhausting VRAM.
  • Implemented Reinhard color normalization to account for varying stain intensities across different hospital labs, ensuring cross-organ generalization (kidney, breast, colon).

Technologies Used

PyTorchKimiaNetDenseNet-121Computer VisionReinhard NormalizationWhole Slide Images

Lessons Learned

In medical AI, data preprocessing (like color normalization and artifact removal) often plays a larger role in model generalization than the raw architecture itself. Working with the Tata Memorial Center highlighted the critical need for model explainability when dealing with clinical diagnoses.