Breast Tumor Image Classification in Bright Challenge VIA Multiple Instance Learning and Deep Transformers

Yangen Zhan,Hao Bian,Yang Chen,Xiu Li,Yongbing Zhang
DOI: https://doi.org/10.1109/isbic56247.2022.9854733
2022-01-01
Abstract:Artificial intelligence models have become increasingly promising in automated computed-aided cancer diagnostics. In this paper, a new deep learning method is proposed for solving breast tumor image classification in the BRIGHT Challenge. Given two types of data, regions of interest (ROIs) and whole slide images (WSIs), the proposed method first utilize ROI data to train a model that is able to select important image patches in WSI data and simultaneously capture a low-dimensional representation for image patches. With features extracted from important image patches in each WSI, another deep learning model following the Transformer framework is trained to perform the final WSI-level tumor classification, in which transfer learning is also employed to fully exploit ROI data. Evaluated on the test dataset, the proposed method achieves the 4th best results in the challenge. Ablation experiments are also carried out to analyze the proposed method in detail.
What problem does this paper attempt to address?