A vision transformer-based framework for knowledge transfer from multi-modal to mono-modal lymphoma subtyping models

Bilel Guetarni,Feryal Windal,Halim Benhabiles,Marianne Petit,Romain Dubois,Emmanuelle Leteurtre,Dominique Collard
2024-05-29
Abstract:Determining lymphoma subtypes is a crucial step for better patient treatment targeting to potentially increase their survival chances. In this context, the existing gold standard diagnosis method, which relies on gene expression technology, is highly expensive and time-consuming, making it less accessibility. Although alternative diagnosis methods based on IHC (immunohistochemistry) technologies exist (recommended by the WHO), they still suffer from similar limitations and are less accurate. Whole Slide Image (WSI) analysis using deep learning models has shown promising potential for cancer diagnosis, that could offer cost-effective and faster alternatives to existing methods. In this work, we propose a vision transformer-based framework for distinguishing DLBCL (Diffuse Large B-Cell Lymphoma) cancer subtypes from high-resolution WSIs. To this end, we introduce a multi-modal architecture to train a classifier model from various WSI modalities. We then leverage this model through a knowledge distillation process to efficiently guide the learning of a mono-modal classifier. Our experimental study conducted on a lymphoma dataset of 157 patients shows the promising performance of our mono-modal classification model, outperforming six recent state-of-the-art methods. In addition, the power-law curve, estimated on our experimental data, suggests that with more training data from a reasonable number of additional patients, our model could achieve competitive diagnosis accuracy with IHC technologies. Furthermore, the efficiency of our framework is confirmed through an additional experimental study on an external breast cancer dataset (BCI dataset).
Image and Video Processing,Computer Vision and Pattern Recognition,Quantitative Methods
What problem does this paper attempt to address?
The aim of this paper is to address the problem of automatically distinguishing between the two subtypes of Diffuse Large B-Cell Lymphoma (DLBCL) — Activated B-Cell-like subtype (ABC) and Germinal Center B-Cell-like subtype (GCB) — using high-resolution Whole Slide Images (WSI). The current standard diagnostic method (RT-MLPA) is accurate but costly and time-consuming, while alternative methods such as Immunohistochemistry (IHC) are faster and less expensive but lack accuracy and require manual involvement by pathologists. Therefore, the researchers propose a method based on the Vision Transformer framework, utilizing multimodal data to train a classifier and transferring the learned information to a unimodal classifier through a knowledge distillation process. This approach aims to achieve efficient and accurate DLBCL subtype classification using only Hematoxylin and Eosin Staining (H&E Staining, HES) images. Experimental results show that this unimodal classifier outperforms six state-of-the-art methods and, with the addition of more training data, its diagnostic accuracy is expected to rival that of IHC technology. Furthermore, the framework has been validated through additional experiments on a breast cancer dataset, further demonstrating its effectiveness.