Correlation Routing Network for Explainable Lesion Classification in Multi-Parametric Liver MRI
Fakai Wang1, Zhehan Shen2, Huimin Lin3
1Department of Radiology, Ruijin Hospital, Shanghai Jiao Tong University School of Medicine, Shanghai, China; Institute for Medical Imaging Technology, Ruijin Hospital, Shanghai Jiao Tong University School of Medicine, Shanghai, China.
Abstract:
Liver tumor diagnosis is a key task in abdominal imaging examination, and there are numerous researches on automatic classifying focal liver lesions (FLL). More are in CT and ultrasound and fewer utilize MRI despite unique diagnostic advantages. The obstacles lie in dataset curation, technical complexity, and clinical explainability in liver MRI. In this paper, we propose the Correlation Routing Network (CRN) which takes in 10 MRI sequences and predicts lesion types (HCC, Cholangioma, Metastasis, Hemangioma, FNH, Cyst) as well as imaging features, to achieve both high accuracy and explainability. The CRN model consists of encoding branches, correlation routing/relay modules, and the self-attention module. The independent encoding paradigm facilitates information disentangling, the correlation routing scheme helps redirection and decoupling effectively, and the self-attention enforces global feature sharing and prediction consistency. The model predicts detailed lesion imaging features, promoting explainable classification and clinical accountability. We also identify the signal relations and derive quantitative explainability. Our liver lesion classification model achieves malignant-benign accuracy of 97.2%, six-class accuracy of 88%, and averaged imaging feature accuracy of 84.9%, outperforming popular CNN and transformer-based models. We hope to spark insights for multimodal lesion classification and model explainability.
More Related Videos
04:09Predicting Treatment Response to Image-Guided Therapies Using Machine Learning: An Example for Trans-Arterial Treatment of Hepatocellular Carcinoma
Published on: October 10, 2018
10:25Brain Infarct Segmentation and Registration on MRI or CT for Lesion-symptom Mapping
Published on: September 25, 2019
