# Predicting isocitrate dehydrogenase status in glioma using hierarchical attention-based deep 3D multiple instance learning

## TL;DR

This paper introduces a new AI model that can predict a brain tumor marker (IDH status) from MRI scans, avoiding invasive procedures.

## Contribution

A novel hierarchical attention-based deep 3D multiple instance learning framework for non-invasive IDH status prediction in gliomas.

## Key findings

- The HAB-MIL model achieved an AUC of 0.917 on the TCIA dataset and 0.892 on the hospital dataset.
- The model performs comparably to state-of-the-art methods while requiring no pixel-level annotations.
- Multiple instance learning shows strong potential for IDH classification in gliomas.

## Abstract

According to the 2021 WHO classification of tumors of the central nervous system, isocitrate dehydrogenase (IDH) status serve an independent prognostic biomarker and is closely associated with tumor diagnosis and treatment response. At present, the determination of IDH status still relies on invasive surgical procedures.

A total of 345 patients with pathologically confirmed gliomas diagnosed at the First Affiliated Hospital of Xi’an Jiaotong University between October 2019 and October 2024 were retrospectively included, comprising 148 (42.9%) IDH-wild and 197 (57.1%) IDH-mutant. An additional 495 glioma patients were obtained from the public TCIA dataset. Patients were randomly split into training, validation, and test cohorts 6:2:2. A Hierarchical Attention-Based Multiple Instance Learning (HAB-MIL) framework was developed, integrating auxiliary positional encoding into feature maps to capture spatially specific information and generate refined 3D lesion representations. Model performance was evaluated using five-fold cross-validation, with receiver operating characteristic (ROC) curves, area under the curve (AUC), sensitivity, and specificity as assessment metrics.

HAB-MIL achieved competitive performance, with AUCs of 0.917 and 0.892 on the glioma datasets from TCIA and the First Affiliated Hospital of Xi’an Jiaotong University. Additionally, our work achieves results that are comparable to the state-of-the-art methods in TCIA dataset and demonstrates that multiple instance learning has great potential for IDH prediction.

The proposed HAB-MIL achieved IDH classification based on conventional preoperative MRI images, eliminating the need for pixel-level annotations and significantly reducing the annotation burden for doctors.