Patent ID: 11963788
Assignee: CITY UNIVERSITY OF HONG KONG
Field: Medical technology (Instruments)
Classification: CPC A  G | IPC A  G

Claim 0:
1. A graph-based prostate diagnosis network for predicting a prostate health status of a patient from a 3D magnetic resonance imaging (MRI) scan containing a plurality of 2D MRI slices, the graph-based prostate diagnosis network comprising:
a feature extractor configured to extract a plurality of preliminary instance embeddings Z corresponding to the plurality 2D MRI slices X respectively;
a pooling operator configured to aggregate the plurality of preliminary instance embeddings Z together to generate a preliminary bag embedding;
a preliminary classifier configured to calculate an instance importance parameter w based on the preliminary bag embedding zbag;
an instance importance calculator configured to calculate a plurality of instance importances a corresponding to the plurality of preliminary instance embeddings respectively based on the instance importance parameter w; and
a plurality of importance-guided graph (IGraph) layers configured to generate a plurality of improved instance embeddings {tilde over (Z)} by performing a plurality of graph convolutions on the plurality of preliminary instance embeddings Z in a sequential manner;
an embedding aggregator configured to generate an improved bag embedding zI by aggregating the plurality of improved instance embeddings {tilde over (Z)} with the plurality of instance importances a; and
a refined classifier configured to process the improved bag embedding zI to generate a refined diagnosis prediction y for predicting prostate health status of the patient.