An annotation-free whole-slide training approach to pathological classification of lung cancer types using deep learning.
Chi-Long ChenChi-Chung ChenWei-Hsiang YuSzu-Hua ChenMichael HsiaoTai-I HsuMichael HsiaoChao-Yuan YehCheng-Yu ChenPublished in: Nature communications (2021)
Deep learning for digital pathology is hindered by the extremely high spatial resolution of whole-slide images (WSIs). Most studies have employed patch-based methods, which often require detailed annotation of image patches. This typically involves laborious free-hand contouring on WSIs. To alleviate the burden of such contouring and obtain benefits from scaling up training with numerous WSIs, we develop a method for training neural networks on entire WSIs using only slide-level diagnoses. Our method leverages the unified memory mechanism to overcome the memory constraint of compute accelerators. Experiments conducted on a data set of 9662 lung cancer WSIs reveal that the proposed method achieves areas under the receiver operating characteristic curve of 0.9594 and 0.9414 for adenocarcinoma and squamous cell carcinoma classification on the testing set, respectively. Furthermore, the method demonstrates higher classification performance than multiple-instance learning as well as strong localization results for small lesions through class activation mapping.
Keyphrases
- deep learning
- squamous cell carcinoma
- convolutional neural network
- artificial intelligence
- machine learning
- neural network
- virtual reality
- weight loss
- working memory
- big data
- genome wide
- rna seq
- locally advanced
- risk factors
- radiation therapy
- dna methylation
- rectal cancer
- lymph node metastasis
- optical coherence tomography