Semi-Supervised Contrastive VAE for Disentanglement of Digital Pathology Images

Mahmudul Hasan,Xiaoling Hu,Shahira Abousamra,Prateek Prasanna,Joel Saltz,Chao Chen
2024-10-03
Abstract:Despite the strong prediction power of deep learning models, their interpretability remains an important concern. Disentanglement models increase interpretability by decomposing the latent space into interpretable subspaces. In this paper, we propose the first disentanglement method for pathology images. We focus on the task of detecting tumor-infiltrating lymphocytes (TIL). We propose different ideas including cascading disentanglement, novel architecture, and reconstruction branches. We achieve superior performance on complex pathology images, thus improving the interpretability and even generalization power of TIL detection deep learning models. Our codes are available at <a class="link-external link-https" href="https://github.com/Shauqi/SS-cVAE" rel="external noopener nofollow">this https URL</a>.
Image and Video Processing,Computer Vision and Pattern Recognition
What problem does this paper attempt to address?