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

• Published in: arXiv.org 2024-05
• Main Authors: Guetarni, Bilel, Windal, Feryal, Halim Benhabiles, Petit, Marianne, Dubois, Romain, Leteurtre, Emmanuelle, Collard, Dominique
• Format: Article
• Language: English
• Subjects: Cancer; Classification; Classifiers; Deep learning; Diagnosis; Distillation; Gene expression; Knowledge management; Lymphoma; Machine learning
• ISSN: 2331-8422
• DOI: 10.48550/arxiv.2308.01328

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).