Mortality prediction method and computer program product based on the fusion of medical ontology embedding and attention mechanism

Abstract

The present invention discloses a mortality prediction method and computer program product based on the fusion of medical ontology embedding and an attention mechanism, relating to the field of medical information technology. The method comprises: performing feature mapping on a patient's demographic information, diagnosis information, treatment information, and vital sign information; constructing a medical ontology for each of the diagnosis information nodes and treatment information nodes in the feature mapping; performing a nonlinear transformation on the diagnosis ontology embedding matrix and the treatment ontology embedding matrix, and processing the diagnosis and treatment information using a masked attention mechanism; obtaining corresponding treatment outcomes based on the patient's medical visit records, learning the causal relationship between diagnosis and treatment, calculating the conditional probability between the patient's diagnosis and treatment outcomes, and obtaining an adjacency matrix; and predicting the patient's mortality rate based on the demographic information and vital sign information in the feature mapping, the diagnosis sequence, the treatment sequence, and the adjacency matrix. The present invention can effectively improve the accuracy of mortality prediction.

Description

Technical Field

[0001] The present invention relates to the field of medical information technology, and in particular to a mortality prediction method and computer program product based on the fusion of medical ontology embedding and attention mechanism. Background Art

[0002] With the rapid growth of medical data, particularly in the context of multimodal data such as electronic health records (EHRs), medical imaging, and genomic data, mortality prediction plays a vital role in public health, clinical medicine, and disease management. Accurate mortality prediction models can help doctors identify high-risk patients promptly, enabling them to implement effective interventions and reduce mortality.

[0003] However, traditional mortality prediction methods often rely on a single type of medical data, and the model's performance is limited by data heterogeneity and a lack of comprehensive knowledge representation, resulting in inaccurate predictions. Consequently, a technical solution to improve the accuracy of mortality prediction is urgently needed. Summary of the Invention

[0004] The present invention aims to provide a mortality prediction method and computer program product based on the fusion of medical ontology embedding and attention mechanisms to address the aforementioned technical problems existing in the prior art. The various technical effects achieved by the preferred technical solutions provided by the present invention are detailed below.

[0005] To achieve the above objectives, the present invention provides the following technical solutions:

[0006] The present invention provides a mortality prediction method based on the fusion of medical ontology embedding and attention mechanism, comprising the following steps:

[0007] The patient's demographic information, diagnosis information, treatment information, and vital signs information are respectively used as demographic information nodes, diagnosis information nodes, treatment information nodes, and vital signs information nodes for feature mapping;

[0008] Constructing medical ontologies for the diagnosis information nodes and treatment information nodes of the feature map respectively, and obtaining the diagnosis ontology embedding matrix and treatment ontology embedding matrix of the patient;

[0009] The diagnosis ontology embedding matrix, the treatment ontology embedding matrix, and the diagnosis information and treatment information of the feature map are processed through a masked attention mechanism to obtain a diagnosis sequence and a treatment sequence of the patient;

[0010] Obtaining corresponding treatment results based on the patient's medical visit records, learning the causal relationship between diagnosis and treatment, calculating the conditional probability between the patient's diagnosis and treatment results, and obtaining an adjacency matrix;

[0011] The patient's mortality rate is predicted based on the demographic information, vital sign information, the diagnosis sequence, the treatment sequence, and the adjacency matrix of the feature map.

[0012] In some embodiments, the patient's demographic information, diagnosis information, treatment information, and vital signs information are respectively used as demographic information nodes, diagnosis information nodes, treatment information nodes, and vital signs information nodes for feature mapping, including the following steps:

[0013] After converting the patient's demographic information, diagnostic information, treatment information, and vital sign information into continuous vectors, a binary vector is generated through label encoding; and the label-encoded diagnostic information and treatment information are activated through a nonlinear function.

[0014] In some embodiments, the medical ontology is a directed acyclic graph with a hierarchical structure representing the diagnosis information nodes and treatment information nodes in the form of parent-child relationships, and the output form is the diagnosis ontology embedding matrix and the treatment ontology embedding matrix.

[0015] In some embodiments, the diagnostic ontology embedding matrix and the treatment ontology embedding matrix are activated by a nonlinear function before being processed by a masked attention mechanism.

[0016] In some embodiments, after the diagnosis ontology embedding matrix and the treatment ontology embedding matrix are processed by the masked attention mechanism, the method further includes:

[0017] The output processed by the masked attention mechanism is sequentially subjected to linear function and ReLu function activation, residual connection, normalization, linear transformation, residual connection, and normalization to obtain the diagnosis sequence and treatment sequence of the patient.

[0018] In some embodiments, a priori guided attention mechanism is used to predict the mortality rate of the patient, and the output of the priori guided attention mechanism is subjected to an attention pooling operation to obtain the mortality rate of the patient.

[0019] In some embodiments, before the attention pooling operation, the method further includes:

[0020] The output of the prior-guided attention mechanism is linearly transformed and then processed by the ReLU activation function; the output processed by the ReLU activation function is residual processed and then normalized.

[0021] In some embodiments, the mortality prediction method based on the fusion of medical ontology embedding and attention mechanism also includes weighted processing of the diagnostic ontology embedding matrix, treatment ontology embedding matrix and feature mapping after processing by the masked attention mechanism, and using the pooled features for the classification task of the medical ontology.

[0022] In some embodiments, the mortality prediction method based on the fusion of medical ontology embedding and attention mechanism also includes using the diagnostic ontology embedding matrix and the treatment ontology embedding matrix for the classification task of the medical ontology.

[0023] On the other hand, the present invention also provides a computer program product, including a computer program / instruction, which, when executed by a processor, implements the steps of the mortality prediction method based on the fusion of medical ontology embedding and attention mechanism as described above.

[0024] Implementing one of the above technical solutions of the present invention has the following advantages or beneficial effects:

[0025] The present invention processes the structured and unstructured data in EHR to obtain a medical ontology, then uses an encoder to obtain different vector representations of the relevant information of the medical ontology, and then obtains prediction indicators through a prediction model, thereby improving the accuracy of mortality prediction in the current medical field. BRIEF DESCRIPTION OF THE DRAWINGS

[0026] In order to more clearly illustrate the technical solutions of the embodiments of the present invention, the following briefly introduces the drawings required for describing the embodiments. Obviously, the drawings described below are only some embodiments of the present invention. Those skilled in the art can also derive other drawings based on these drawings without inventive work. In the drawings:

[0027] Figure 1 This is a flowchart of a mortality prediction method based on the fusion of medical ontology embedding and attention mechanism according to an embodiment of the present invention;

[0028] Figure 2 This is a system framework diagram of a mortality prediction method based on the fusion of medical ontology embedding and attention mechanism according to an embodiment of the present invention;

[0029] Figure 3 This is a medical ontology framework diagram constructed by diagnostic information according to an embodiment of the present invention;

[0030] Figure 4 This is a medical ontology framework diagram for constructing treatment information according to an embodiment of the present invention;

[0031] Figure 5This is an example diagram of medical consultation-treatment characteristic information created based on medical consultation records in an embodiment of the present invention. DETAILED DESCRIPTION

[0032] In order to make the objects, technical solutions and advantages of the present invention clearer, the various exemplary embodiments to be described below will refer to the corresponding drawings, which constitute a part of the exemplary embodiments, in which various exemplary embodiments that may be used to implement the present invention are described. Unless otherwise indicated, the same numbers in different drawings represent the same or similar elements. The implementation methods described in the following exemplary embodiments do not represent all implementation methods consistent with the present disclosure. It should be understood that they are only examples of processes, methods and devices that are consistent with some aspects of the present disclosure as detailed in the appended claims, and other embodiments may also be used, or structural and functional modifications may be made to the embodiments listed herein without departing from the scope and essence of the present invention.

[0033] In order to illustrate the technical solution of the present invention, a specific embodiment is provided below, in which only the parts related to the embodiment of the present invention are shown.

[0034] Example 1:

[0035] like Figure 1-Figure 2 As shown, the present invention provides a mortality prediction method based on the fusion of medical ontology embedding and attention mechanism, comprising the following steps:

[0036] S100, performing feature mapping on the patient's demographic information, diagnosis information, treatment information, and vital signs information as a demographic information node, a diagnosis information node, a treatment information node, and a vital signs information node, respectively;

[0037] S200, constructing medical ontologies for the diagnosis information nodes and treatment information nodes of the feature map, respectively, to obtain the patient's diagnosis ontology embedding matrix and treatment ontology embedding matrix;

[0038] S300, processing the diagnosis ontology embedding matrix, the treatment ontology embedding matrix, and the diagnosis information and treatment information of the feature map through a masked attention mechanism to obtain the patient's diagnosis sequence and treatment sequence;

[0039] S400, based on the patient's visit record, obtain the corresponding treatment results, learn the causal relationship between diagnosis and treatment, calculate the conditional probability between the patient's diagnosis and treatment results, and use The function generates a conditional probability distribution and ultimately constructs an adjacency matrix. The adjacency matrix represents the relationship between diagnosis and treatment, providing support for subsequent analysis and prediction.

[0040] S500 , predicting the patient's mortality rate based on demographic information, vital sign information, diagnosis sequence, treatment sequence, and adjacency matrix of the feature map.

[0041] The present invention processes structured and unstructured data in EHRs (electronic health records) to obtain a medical ontology. The relevant information of the medical ontology is then encoded to obtain different vector representations. The prediction model then generates prediction indicators, thereby improving the accuracy of patient mortality prediction.

[0042] Based on the above embodiment, the patient's demographic information, diagnosis information, treatment information, vital signs information, and other data are derived from the open-source, de-identified MIMIC-III (Medical Information for Intensive Care, MIMIC) database. In step S100, the patient's basic hospitalization information and related data are extracted from the MIMIC-III database. This includes a data fusion module with structured and unstructured data. The structured data includes static information and time-series medical codes, while the unstructured data includes time-series clinical record information.

[0043] It should be noted that the MIMIC-III database is a large-scale, publicly available clinical dataset widely used in medical research and artificial intelligence model training. This dataset contains anonymous medical data from intensive care unit patients, covering physiological information, medical records, medications, laboratory results, imaging data, and more. It is widely used in research such as medical data analysis, machine learning, and predictive model development. Of course, the database in this embodiment can also be a subsequent version of MIMIC-III, such as MIMIC-IV.

[0044] Based on the above embodiment, in the above step S100, the demographic information, diagnostic information, treatment information, and vital signs information of all patients are extracted from the data as demographic information node i, diagnostic information node d, treatment information node p, and vital signs information node s, respectively, and the node data is feature mapped using an encoder. Furthermore, after extracting the demographic information, diagnostic information, treatment information, and vital signs information from the data set, these discrete data are then converted into continuous vectors that can capture the intrinsic characteristics and relationships of the data. Ultimately, the demographic information i, vital signs s, diagnostic information d, and treatment information p are converted into encoded information vectors, which are then encoded to generate binary vectors, where 0 or 1 at each position represents the presence or absence of the feature. When these data for all patients are converted into binary vectors, the patient's diagnostic information x d and treatment information x p Two vectors pass through Function performs nonlinear activation and then serves as Input to the masked attention mechanism in the model; patient demographic information x i and vital signs x s As The prior in the model guides the input of the attention mechanism model.

[0045] It should be noted that This is a deep learning model based on self-attention. Its success lies in its innovative use of efficient self-attention, which enables the model to capture long-range dependencies while allowing for modular computation, significantly improving efficiency and performance. Its core technology, the self-attention mechanism, focuses on the interdependencies between positions in the input sequence, enabling the model to dynamically select and aggregate relevant information. Its main components include self-attention, multi-head attention, positional encoding, and encoder-decoder architecture. Initially used for natural language processing, the model has rapidly expanded to multiple fields, including natural language processing, image processing, speech recognition, and time series forecasting.

[0046] Based on the above embodiment, in step S200, the medical ontology is a directed acyclic graph with a hierarchical structure that represents diagnosis information nodes and treatment information nodes in the form of parent-child relationships, and the output form is a diagnosis ontology embedding matrix and a treatment ontology embedding matrix. Specifically,

[0047] like Figure 3-Figure 4 As shown, the nodes in the embedding matrix is a basic embedding via the attention mechanism Calculate the embedding of all leaf nodes and its ancestors, and finally form the embedding matrix of all leaf nodes In the medical ontology, each diagnosis (treatment) node is assigned a basic embedding vector, and then is the basic embedding of each layer node. The nodes in the embedding matrix The formula can be expressed as:

[0048] (1);

[0049] in, It is calculated Time Embedding The attention weight on represents the basic embedding, Represents each layer node and its ancestral embeddings.

[0050] The above attention weights By the following Function Compute:

[0051] (2);

[0052] in, Is a scalar value indicating and Compatibility between basic embeddings of scalar values is calculated by a feed-forward network with a single hidden layer:

[0053] (3);

[0054] in, yes and The concatenated weight matrix, is the bias vector, is the weight vector used to generate the scalar value.

[0055] When the medical ontology embedding matrix of diagnosis information and treatment information is obtained and Afterwards, they will pass through Function performs nonlinear activation and then serves as Input to the model's masked attention module.

[0056] It should be noted that the above medical ontology It is a formal description of concepts in the medical field and the relationships between them. It provides important support for data structuring by defining the hierarchical structure and relationships between medical terms and concepts. Medical ontology can effectively integrate knowledge from different data sources (such as clinical texts, medical images, genetic data, etc.) and map these multimodal information into a unified framework. Embedding based on medical ontology The method provides a unified semantic representation by converting medical knowledge into a low-dimensional vector space, which makes it possible to integrate and efficiently process multimodal data.

[0057] Medical ontology information embedding It is an ontology in the medical field The technology of converting information into a numerical representation (i.e., embedding vector) that can be processed by machines. Its purpose is to use the medical concepts and terms defined in the ontology and convert this information into a form that can be understood by computers through embedding learning methods. The core goal of medical ontology information embedding is to efficiently map the concepts and relationships in the medical ontology into a low-dimensional processing space through machine learning technology, so that these semantic information can be processed by computers. Medical ontology information embedding is usually achieved with the help of deep learning, graph neural networks, etc. , natural language processing and other technologies to achieve this.

[0058] Based on the above embodiment, the diagnosis ontology embedding matrix and the treatment ontology embedding matrix are processed by the mask attention mechanism through a nonlinear function (such as Activation function) is used to activate the diagnostic ontology embedding matrix and the treatment ontology embedding matrix to obtain enhanced feature representation. Furthermore, after the diagnostic ontology embedding matrix and the treatment ontology embedding matrix are processed by the masked attention mechanism, they also include:

[0059] The output processed by the masked attention mechanism is sequentially subjected to linear function and ReLu function activation, residual connection, normalization, linear transformation, residual connection, and normalization to obtain the patient's diagnosis sequence and treatment sequence.

[0060] It is understandable that in step S300, The model uses a masked attention mechanism to process diagnosis sequences and treatment sequences, thereby considering the dependencies between elements in the sequence. The diagnosis and treatment ontology embedding matrices after nonlinear activation and the two binary vectors of diagnosis information and treatment information are used as The model masks the input to the attention mechanism, which allows the model to take into account the dependencies between elements in the sequence when processing sequential data.

[0061] The above residual connection adds the input directly to the (Mask Attention Mechanism) on the processed output, which helps alleviate the gradient vanishing problem in deep networks.

[0062] The above normalization is implemented using layer normalization, which normalizes the data to improve the stability of training.

[0063] The above linear function is activated with the ReLu function ( ) represents a feedforward neural network The layer consists of a linear transformation followed by a ReLU activation function. This step performs a nonlinear transformation on the data, increasing the expressive power of the model.

[0064] Then, residual connections and layer normalization (normalization) are performed to process the output of the feedforward neural network layer. This also helps to improve the stability and efficiency of model training.

[0065] The above linear transformation is achieved through the linear layer, which transforms the output of the feedforward network into the final output. The linear layer is usually used to project the data back to the original dimension or transform it into a specific dimension required by the model.

[0066] Based on the above embodiment, in step S400, the corresponding treatment results are obtained based on the patient's visit record visit, and the causal relationship between diagnosis and treatment is learned, and the conditional probability of the diagnosis result and the treatment result is calculated, and the patient's visit record visit is used to obtain the corresponding treatment result. Function generates conditional probability distribution and creates adjacency matrix , serves as a guide for the attention mechanism to represent the relationship between diagnosis and treatment.

[0067] It is understandable that Learn the causal relationship between diagnosis and treatment, and calculate the conditional probability of diagnosis and treatment results.

[0068] It is a combined graph topology network and A novel deep learning model based on the proposed mechanism is used to process complex data containing graph structures and sequence information. Mainly for graph structure data, it can find the node neighbors and the relationship between them, It excels in natural language processing, can handle insanely long sequences and capture global dependencies, Fusion Powerful representation capabilities in graph structure data processing and Advantages in capturing long-range dependencies. The core technology of the graph includes node features and adjacency information in graph data. Its main principles include layer , Multi-head self-attention mechanism , position encoding , graph attention layer .in It has been widely used in the fields of social network analysis, molecular and drug discovery, knowledge graphs and question-answering systems, and recommendation systems.

[0069] like Figure 5 As shown in the figure, starting from the top visit node, the visit is composed of multiple types of feature information, and the situations are divided into: demographic information, visit time series, cough, headache, and diabetes. Among them, dextromethorphan, ibuprofen, aspirin, and insulin reflect the doctor's decision-making process.

[0070] The above has created an adjacency matrix for the patient's visit record visit , that is, the The modification aims to introduce the adjacency matrix and prior knowledge (in the form of attention mask to learn about electronic health records ( ). Among them, the adjacency matrix Middle The conditional probability of a node By the following Function Compute:

[0071] (4);

[0072] in, and They are Node query ( ) and key ( )vector, yes The column dimensions of the trainable parameters in the module (i.e., the vector dimensions of the query and key), is a mask matrix used to introduce prior knowledge (such as some irrelevant nodes).

[0073] Furthermore, Calculated by the following formula:

[0074] (5);

[0075] in, Indicates the convolutional node embedding, Indicates the indivual The trainable parameter matrices of the module are used to generate query and key vectors respectively.

[0076] Furthermore, Calculated by the following formula:

[0077] (6);

[0078] in, Indicates the convolutional node embedding, Indicates the indivual The trainable parameter matrices of the module are used to generate query and key vectors respectively.

[0079] Furthermore, Each convolution node in Calculated by the following formula:

[0080] (7);

[0081] (8);

[0082] in, Indicates the convolutional node embedding, is the first A convolutional multilayer perceptron is used to further transform the node embeddings, Indicates the convolution conditional probability matrix, Indicates the convolutional node embedding, Indicates the indivual The module's trainable parameter matrix, used to generate values ( )vector.

[0083] Based on the above embodiment, in step S500, the patient's mortality rate is predicted using the priori guided attention mechanism, and the output of the priori guided attention mechanism is subjected to an attention pooling operation to obtain the patient's mortality rate.

[0084] Furthermore, before the attention pooling operation, it also includes:

[0085] The output of the prior-guided attention mechanism is linearly transformed and then processed by the ReLU activation function; the output processed by the ReLU activation function is residual processed and then normalized.

[0086] Based on the above embodiment, it also includes weighted processing of the diagnostic ontology embedding matrix, the treatment ontology embedding matrix and the diagnostic information and treatment information of the feature map processed by the masked attention mechanism, and the features after pooling processing are used for the classification task of the medical ontology.

[0087] Based on the above embodiment, it also includes the classification task of medical ontology using the diagnosis ontology embedding matrix and the treatment ontology embedding matrix.

[0088] In summary, this embodiment processes structured and unstructured data from EHRs to generate a medical ontology. This information is then encoded into different vector representations, and then used in a prediction model to generate prediction indicators, thereby improving the accuracy of current mortality predictions in the medical field. This method has high prediction accuracy, strong model expression capabilities, and high training stability, effectively avoiding the vanishing gradient problem in deep networks.

[0089] Example 2:

[0090] This embodiment provides a computer program product, including a computer program / instruction, which, when executed by a processor, implements the steps of the mortality prediction method based on the fusion of medical ontology embedding and attention mechanism as described in Example 1.

[0091] Those skilled in the art will appreciate that all or part of the features / steps of the aforementioned method embodiments can be implemented via methods, data processing systems, or computer programs. These features can be implemented entirely in software, without hardware, or through a combination of hardware and software. The aforementioned computer program can be stored in one or more computer-readable storage media. The storage medium stores the computer program, and when executed (e.g., by a processor), the computer program performs the steps of the aforementioned embodiment of the mortality prediction method based on the fusion of medical ontology embedding and attention mechanism.

[0092] The aforementioned storage media that can store computer programs / instruction codes include: static hard disks, solid-state hard disks, random access memories (SRAM), electrically erasable programmable read-only memories (EEPROM), erasable programmable read-only memories (EPROM), programmable read-only memories (PROM), read-only memories (ROM), optical storage devices, magnetic storage devices, flash memories, magnetic disks or optical disks and / or combinations of the above devices, that is, they can be implemented by any type of volatile or non-volatile storage devices or their combinations.

[0093] It should be understood that, although the steps in the flowcharts of the above embodiments are shown in sequence as indicated by the arrows, these steps are not necessarily performed in the order indicated by the arrows. Unless otherwise specified herein, there is no strict order restriction on the execution of these steps, and these steps can be performed in other orders. Moreover, at least a portion of the steps in the flowcharts of the above embodiments may include multiple steps or multiple stages, and these steps or stages are not necessarily performed at the same time, but can be performed at different times. The execution order of these steps or stages is not necessarily to be performed in sequence, but can be performed in turn or alternately with other steps or at least a portion of steps or stages in other steps.

[0094] The foregoing is merely a preferred embodiment of the present invention. Those skilled in the art will appreciate that various changes or equivalent substitutions may be made to these features and embodiments without departing from the spirit and scope of the present invention. Furthermore, under the guidance of the present invention, these features and embodiments may be modified to suit specific circumstances and materials without departing from the spirit and scope of the present invention. Therefore, the present invention is not limited to the specific embodiments disclosed herein, and all embodiments falling within the scope of the claims of this application are intended to be within the scope of the present invention.

Claims

1. A mortality prediction method based on the fusion of medical ontology embedding and attention mechanism, characterized by: The steps include: The patient's demographic information, diagnosis information, treatment information, and vital signs information are respectively used as demographic information nodes, diagnosis information nodes, treatment information nodes, and vital signs information nodes for feature mapping; Constructing medical ontologies for the diagnosis information nodes and treatment information nodes of the feature map respectively, and obtaining the diagnosis ontology embedding matrix and treatment ontology embedding matrix of the patient; The diagnostic ontology embedding matrix, the treatment ontology embedding matrix, and the diagnostic information and treatment information of the feature map are processed through a masked attention mechanism to obtain a diagnosis sequence and a treatment sequence after enhanced feature embedding; before the diagnostic ontology embedding matrix and the treatment ontology embedding matrix are processed through the masked attention mechanism, they are activated through a nonlinear function; Obtaining corresponding treatment results based on the patient's medical visit records, learning the causal relationship between diagnosis and treatment, calculating the conditional probability between the patient's diagnosis results and treatment results, and obtaining an adjacency matrix; The patient's mortality rate is predicted based on the demographic information, vital sign information, the diagnosis sequence, the treatment sequence, and the adjacency matrix of the feature map.

2. The mortality prediction method based on the fusion of medical ontology embedding and attention mechanism according to claim 1 is characterized in that: The patient's demographic information, diagnosis information, treatment information, and vital signs information are respectively used as demographic information nodes, diagnosis information nodes, treatment information nodes, and vital signs information nodes for feature mapping, including the following steps: After converting the patient's demographic information, diagnosis information, treatment information, and vital sign information into continuous vectors, a binary vector is generated by label encoding; The diagnostic information and treatment information encoded by the label are activated through a nonlinear function.

3. The mortality prediction method based on the fusion of medical ontology embedding and attention mechanism according to claim 1 is characterized in that: The medical ontology is a directed acyclic graph with a hierarchical structure representing the diagnosis information nodes and the treatment information nodes in the form of a parent-child relationship, and the output form is the diagnosis ontology embedding matrix and the treatment ontology embedding matrix.

4. The mortality prediction method based on the fusion of medical ontology embedding and attention mechanism according to claim 1 is characterized in that: After the diagnosis ontology embedding matrix and the treatment ontology embedding matrix are processed by the masked attention mechanism, the following further comprises: The output processed by the masked attention mechanism is sequentially subjected to linear function and ReLu function activation, residual connection, normalization, linear transformation, residual connection, and normalization to obtain the diagnosis sequence and treatment sequence of the patient.

5. The mortality prediction method based on the fusion of medical ontology embedding and attention mechanism according to claim 1 is characterized in that: The mortality rate of the patient is predicted by using a priori guided attention mechanism, and the output of the prior guided attention mechanism is subjected to an attention pooling operation to obtain the mortality rate of the patient.

6. The mortality prediction method based on the fusion of medical ontology embedding and attention mechanism according to claim 5 is characterized in that: Before the attention pooling operation, it also includes: The output of the prior-guided attention mechanism is linearly transformed and then processed by the ReLU activation function; the output processed by the ReLU activation function is residual processed and then normalized.

7. The mortality prediction method based on the fusion of medical ontology embedding and attention mechanism according to any one of claims 1 to 6, characterized in that: It also includes weighted processing of the diagnostic ontology embedding matrix, the treatment ontology embedding matrix and the diagnostic information and treatment information of the feature map after processing by the masked attention mechanism, and using the features after pooling processing for the classification task of the medical ontology.

8. The mortality prediction method based on the fusion of medical ontology embedding and attention mechanism according to any one of claims 1 to 6, characterized in that: It also includes a classification task for the medical ontology using the diagnosis ontology embedding matrix and the treatment ontology embedding matrix.

9. A computer program product comprising a computer program / instructions, characterized in that When the computer program / instructions are executed by a processor, the steps of the mortality prediction method based on the fusion of medical ontology embedding and attention mechanism are implemented as described in any one of claims 1 to 8.

Images