BKC-Net: Bi-Knowledge Contrastive Learning for Renal Tumor Diagnosis on 3D CT Images

Jindi Kong,Yuting He,Xiaomei Zhu,Pengfei Shao,Yi Xu,Yang Chen,Jean-Louis Coatrieux,Guanyu Yang
DOI: https://doi.org/10.1016/j.knosys.2022.109369
IF: 8.139
2022-01-01
Knowledge-Based Systems
Abstract:Renal tumor diagnosis on abdominal enhanced CT volumes is one of the most significant tasks in kidney disease diagnosis. It helps clinicians decide whether to perform the surgery (subtype classification), perform radical operations or minimally invasive treatment (grade classification). However, inherent challenges greatly limit the performance of the model: 1) Tumor appearance differences caused by non-tumor factors. 2) Small inter-class differences and large intra-class variations. In this paper, we propose a novel diagnosis framework for renal tumors, Bi-knowledge Contrastive Network (BKC-Net), which has two innovations: (1) Focus-perceive learning segments the tumors while perceiving the surrounding healthy tissues, thus adjusting the model’s representation of tumor appearance, helping the BKC-Net represent the inherent features of tumors. (2) Bi-knowledge contrastive learning introduces prior radiomics features, makes the prior radiomics knowledge and latent deep knowledge complementary to each other from the intra-case level, and forces the high cohesion and low coupling embedding feature space from the inter-case levels, helping to discover subtle but essential differences among classes. Experiments demonstrate that our BKC-Net has the best performance in renal tumor diagnosis. Results reveal that our framework has great potential for renal tumor diagnosis in clinical use. Source codes will be released at https://github.com/DD0922/BKC-Net-pytorch .
What problem does this paper attempt to address?