Deep Hierarchical Multiple Instance Learning for Whole Slide Image Classification
Yuanpin Zhou, Yao Lu
Abstract
Whole slide scanning is a powerful tool in clinical diagnosis and pathological research. However, it’s time-consuming to acquire localized annotations in whole slide images (WSIs). Recently, deep multiple instance learning (MIL) approaches were proposed to classify WSIs with only global annotations. Two main challenges, interpretability and utilizing multiple-scale information, remain to be solved in these approaches. In this study, we proposed a deep hierarchical multiple in-stance learning model to tackle these challenges. We introduced max-max ranking loss to better leverage the standard MIL assumption for better interpretability. A hierarchical architecture was designed to reduce computational costs and to utilize multiple-scale information. Our model was evaluated in a large WSI dataset CAMELYON16 with accuracy and AUC as metrics. Experimental results showed that our model achieved the best performance.