Multi-center Ovarian Tumor Classification Using Hierarchical Transformer-Based Multiple-Instance Learning
摘要
Malignant ovarian tumors (OTs) are a leading cause of gynecological cancer deaths, and often remain asymptomatic until advanced stages, making early and accurate diagnosis crucial for effective treatment and good patient outcome. Current diagnostic methods often fall short due to the heterogeneous nature of OTs and the complexities in distinguishing benign from malignant forms. To overcome these limitations, this study proposes a novel framework leveraging transformer-based multiple-instance learning (MIL) and hierarchical self-supervised pre-training. To validate the model, a comprehensive multi-center dataset has been compiled, encompassing diverse patient demographics and imaging protocols. Benchmarking against conventional radiomics methods and other deep learning approaches, the hierarchical MIL model demonstrates superior performance with a median AUROC of 0.84 and high recall of 0.91. These results highlight significant improvements in sensitivity, essential for minimizing false negatives in clinical settings. The performed study emphasizes the importance of multi-center validation and external dataset testing to ensure generalization of the proposed model and obtain a higher robustness. The encountered complexity of multi-center data is found significant, since various clinical factors play an influential role. This makes baseline comparisons virtually impossible and the need for more multi-center research increasingly compelling and encouraging.