Artificial intelligence application in lymphoma diagnosis with Vision Transformer using weakly supervised training
Fuente:
arXiv
Enregistré dans:
| Auteurs principaux: | , , , , , , , , , , , |
|---|---|
| Format: | Preprint |
| Publié: |
2026
|
| Sujets: | |
| Accès en ligne: | |
| Tags: |
Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
|
| _version_ | 1866915938613329920 |
|---|---|
| author | Nghia Nguyen Wahed, Amer Quesada, Andy Ali, Yasir Achi, Hanadi El Zhang, Y. Helen Ursua, Jocelyn Banerjee, Alex Kalra, Sahib Medeiros, L. Jeffrey Xu, Jie |
| author_facet | Nghia Nguyen Wahed, Amer Quesada, Andy Ali, Yasir Achi, Hanadi El Zhang, Y. Helen Ursua, Jocelyn Banerjee, Alex Kalra, Sahib Medeiros, L. Jeffrey Xu, Jie |
| contents | Vision transformers (ViT) have been shown to allow for more flexible feature detection and can outperform convolutional neural network (CNN) when pre-trained on sufficient data. Due to their promising feature detection capabilities, we deployed ViTs for morphological classification of anaplastic large cell lymphoma (ALCL) versus classic Hodgkin lymphoma (cHL). We had previously designed a ViT model which was trained on a small dataset of 1,200 image patches in fully supervised training. That model achieved a diagnostic accuracy of 100% and an F1 score of 1.0 on the independent test set. Since fully supervised training is not a practical method due to lack of expertise resources in both the training and testing phases, we conducted a recent study on a modified approach to training data (weakly supervised training) and show that labeling training image patch automatically at the slide level of each whole-slide-image is a more practical solution for clinical use of Vision Transformer. Our ViT model, trained on a larger dataset of 100,000 image patches, yields evaluation metrics with significant accuracy, F1 score, and area under the curve (AUC) at 91.85%, 0.92, and 0.98, respectively. These are respectable values that qualify this ViT model, with weakly supervised training, as a suitable tool for a deep learning module in clinical model development using automated image patch extraction. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2604_13795 |
| institution | arXiv |
| publishDate | 2026 |
| record_format | arxiv |
| spellingShingle | Artificial intelligence application in lymphoma diagnosis with Vision Transformer using weakly supervised training Nghia Nguyen Wahed, Amer Quesada, Andy Ali, Yasir Achi, Hanadi El Zhang, Y. Helen Ursua, Jocelyn Banerjee, Alex Kalra, Sahib Medeiros, L. Jeffrey Xu, Jie Computer Vision and Pattern Recognition Machine Learning Vision transformers (ViT) have been shown to allow for more flexible feature detection and can outperform convolutional neural network (CNN) when pre-trained on sufficient data. Due to their promising feature detection capabilities, we deployed ViTs for morphological classification of anaplastic large cell lymphoma (ALCL) versus classic Hodgkin lymphoma (cHL). We had previously designed a ViT model which was trained on a small dataset of 1,200 image patches in fully supervised training. That model achieved a diagnostic accuracy of 100% and an F1 score of 1.0 on the independent test set. Since fully supervised training is not a practical method due to lack of expertise resources in both the training and testing phases, we conducted a recent study on a modified approach to training data (weakly supervised training) and show that labeling training image patch automatically at the slide level of each whole-slide-image is a more practical solution for clinical use of Vision Transformer. Our ViT model, trained on a larger dataset of 100,000 image patches, yields evaluation metrics with significant accuracy, F1 score, and area under the curve (AUC) at 91.85%, 0.92, and 0.98, respectively. These are respectable values that qualify this ViT model, with weakly supervised training, as a suitable tool for a deep learning module in clinical model development using automated image patch extraction. |
| title | Artificial intelligence application in lymphoma diagnosis with Vision Transformer using weakly supervised training |
| topic | Computer Vision and Pattern Recognition Machine Learning |
| url | https://arxiv.org/abs/2604.13795 |