Advanced Hybrid Deep Learning Model for Enhanced Classification of Osteosarcoma Histopathology Images

TL;DR

This paper introduces a hybrid deep learning model combining CNN and ViT to classify osteosarcoma histopathology images with high accuracy, advancing diagnostic methods for this bone cancer.

Contribution

The study presents a novel hybrid CNN-ViT model that achieves four-class classification of osteosarcoma images, setting a new benchmark in accuracy and diagnostic potential.

Findings

Achieved 99.08% accuracy on TCIA dataset

First four-class classification for osteosarcoma images

Model outperforms existing methods in accuracy and precision

Abstract

Recent advances in machine learning are transforming medical image analysis, particularly in cancer detection and classification. Techniques such as deep learning, especially convolutional neural networks (CNNs) and vision transformers (ViTs), are now enabling the precise analysis of complex histopathological images, automating detection, and enhancing classification accuracy across various cancer types. This study focuses on osteosarcoma (OS), the most common bone cancer in children and adolescents, which affects the long bones of the arms and legs. Early and accurate detection of OS is essential for improving patient outcomes and reducing mortality. However, the increasing prevalence of cancer and the demand for personalized treatments create challenges in achieving precise diagnoses and customized therapies. We propose a novel hybrid model that combines convolutional neural networks (CNN) and vision transformers (ViT) to improve diagnostic accuracy for OS using hematoxylin and eosin (H&E) stained histopathological images. The CNN model extracts local features, while the ViT captures global patterns from histopathological images. These features are combined and classified using a Multi-Layer Perceptron (MLP) into four categories: non-tumor (NT), non-viable tumor (NVT), viable tumor (VT), and none-viable ratio (NVR). Using the Cancer Imaging Archive (TCIA) dataset, the model achieved an accuracy of 99.08%, precision of 99.10%, recall of 99.28%, and an F1-score of 99.23%. This is the first successful four-class classification using this dataset, setting a new benchmark in OS research and offering promising potential for future diagnostic advancements.