Improve model card with pipeline tag and paper details

#1
by nielsr HF Staff - opened
Files changed (1) hide show
  1. README.md +14 -3
README.md CHANGED
@@ -1,3 +1,14 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ pipeline_tag: graph-ml
4
+ ---
5
+
6
+ # Multimodal Contrastive Representation Learning in Augmented Biomedical Knowledge Graphs
7
+
8
+ This repository contains the model presented in the paper [Multimodal Contrastive Representation Learning in Augmented Biomedical Knowledge Graphs](https://huggingface.co/papers/2501.01644).
9
+
10
+ ## Abstract
11
+ Biomedical Knowledge Graphs (BKGs) integrate diverse datasets to elucidate complex relationships within the biomedical field. Effective link prediction on these graphs can uncover valuable connections, such as potential novel drug-disease relations. We introduce a novel multimodal approach that unifies embeddings from specialized Language Models (LMs) with Graph Contrastive Learning (GCL) to enhance intra-entity relationships while employing a Knowledge Graph Embedding (KGE) model to capture inter-entity relationships for effective link prediction. To address limitations in existing BKGs, we present PrimeKG++, an enriched knowledge graph incorporating multimodal data, including biological sequences and textual descriptions for each entity type. By combining semantic and relational information in a unified representation, our approach demonstrates strong generalizability, enabling accurate link predictions even for unseen nodes. Experimental results on PrimeKG++ and the DrugBank drug-target interaction dataset demonstrate the effectiveness and robustness of our method across diverse biomedical datasets. Our source code, pre-trained models, and data are publicly available at this https URL
12
+
13
+ ## Code and Data
14
+ The paper states that source code, pre-trained models, and data are publicly available. However, a direct link to a code repository was not provided in the context. Please refer to the paper for more details on accessing these resources.