Search papers, labs, and topics across Lattice.
This paper introduces the KOAL framework, which enhances the prediction of Gleason Grade Group (GGG) in prostate cancer by integrating clinical variables and expert knowledge into the learning process. By employing a Clinical-Context Modulation module to adjust image representations based on non-image data and a Knowledge-Guided Prototype Alignment module to use expert insights from radiology reports, KOAL effectively addresses the limitations of existing methods that oversimplify GGG classification. The results show that KOAL significantly outperforms state-of-the-art approaches on both public and in-house datasets, demonstrating its potential for improving non-invasive cancer grading accuracy.
KOAL's innovative use of clinical context and expert knowledge allows for a more nuanced and accurate prediction of prostate cancer grading, outperforming traditional methods.
Non-invasive prediction of Gleason Grade Group (GGG) in prostate cancer using multiparametric MRI (mpMRI) is clinically vital for reducing unnecessary biopsies. Existing GGG prediction methods face two major limitations. First, they often overlook non-image information critical for GGG prediction, including age, prostate-specific antigen (PSA), and expert priors embedded in radiology reports. Second, they tend to oversimplify GGG as flat categorical labels, failing to account for its intrinsic hierarchy of primary and secondary Gleason patterns. To this end, we propose a novel Knowledge-Driven Ordinal-Aware Learning (KOAL) framework with three synergistic modules. Specifically, the Clinical-Context Modulation (CCM) module uses clinical variables (e.g., age and PSA) to dynamically modulate discriminative image representations. The Knowledge-Guided Prototype Alignment (KGPA) module leverages an LLM to extract group-specific expert knowledge from training radiology reports and clinical guidelines, producing offline semantic anchors describing grade-specific radiological findings without requiring patient-specific reports at inference. Through prototype contrastive alignment, patient-specific mpMRI representations are matched with these anchors to promote pathology-aligned representation learning. The Hierarchical Ordinal-aware Constraints (HOC) module decouples primary and secondary Gleason pattern prediction and maps their probabilistic outputs to GGG via a Differentiable Bio-logic Mapping Layer (DBML), ensuring pathological grading consistency. Experiments on public PI-CAI and in-house datasets demonstrate that KOAL outperforms state-of-the-art methods. Code is available at: https://github.com/Gother-GZ/KOAL.
