Multi-Scale Hybrid Vision Transformer for Learning Gastric Histology: AI-Based Decision Support System for Gastric Cancer Treatment

Yujin Oh,Go Eun Bae,Kyung-Hee Kim,Min-Kyung Yeo,Jong Chul Ye
DOI: https://doi.org/10.1109/JBHI.2023.3276778
2023-08-15
Abstract:Gastric endoscopic screening is an effective way to decide appropriate gastric cancer (GC) treatment at an early stage, reducing GC-associated mortality rate. Although artificial intelligence (AI) has brought a great promise to assist pathologist to screen digitalized whole slide images, existing AI systems are limited in fine-grained cancer subclassifications and have little usability in planning cancer treatment. We propose a practical AI system that enables five subclassifications of GC pathology, which can be directly matched to general GC treatment guidance. The AI system is designed to efficiently differentiate multi-classes of GC through multi-scale self-attention mechanism using 2-stage hybrid Vision Transformer (ViT) networks, by mimicking the way how human pathologists understand histology. The AI system demonstrates reliable diagnostic performance by achieving class-average sensitivity of above 0.85 on a total of 1,212 slides from multicentric cohort. Furthermore, AI-assisted pathologists show significantly improved diagnostic sensitivity by 12% in addition to 18% reduced screening time compared to human pathologists. Our results demonstrate that AI-assisted gastric endoscopic screening has a great potential for providing presumptive pathologic opinion and appropriate cancer treatment of gastric cancer in practical clinical settings.
Image and Video Processing,Computer Vision and Pattern Recognition,Machine Learning
What problem does this paper attempt to address?
The paper aims to address several key issues in the pathological diagnosis of gastric cancer (GC) and proposes an artificial intelligence (AI)-based decision support system to assist clinicians in making appropriate treatment choices for gastric cancer patients at an early stage. Specifically, the objectives of the study include: 1. **Multi-class Gastric Cancer Pathology:** Proposing an AI system capable of subdividing gastric cancer into 5 categories: No Dysplasia (NFD), Tubular Adenoma (TA), Differentiated Carcinoma (Diff-CA), Undifferentiated Carcinoma (Undiff-CA), and Mucosa-Associated Lymphoid Tissue Lymphoma (MALT). This subdivision can better match existing gastric cancer treatment guidelines. 2. **Efficient Pathological Image Analysis:** Designing a two-stage multi-scale hybrid vision transformer network that can effectively handle large-scale whole slide images (WSI) and capture important tissue structural features at different scales. This method mimics the process of pathologists observing tissue samples from low to high magnification to understand global structural relationships and local morphological features. 3. **Improving Diagnostic Accuracy and Efficiency:** The paper demonstrates that the proposed AI system has a high average sensitivity (above 0.85) on both internal and external datasets. In practical clinical applications, with AI assistance, pathologists can significantly improve diagnostic sensitivity (reaching 0.93±0.06) while saving time. In summary, the core contribution of this paper is the proposal of a practical AI-assisted decision support system that not only accurately classifies gastric cancer but also directly guides clinicians in selecting appropriate treatment methods for patients. Additionally, the system has shown good generalization ability and has achieved excellent results in the detection of gastrointestinal organ cancers.