Paper deep dive
GLaDiGAtor: Language-Model-Augmented Multi-Relation Graph Learning for Predicting Disease-Gene Associations
Osman Onur Kuzucu, Tunca DoÄan
Intelligence
Status: succeeded | Model: Gemma-4-26B-A4B | Prompt: intel-v1 | Confidence: 92%
Last extracted: 7/20/2026, 9:23:02 PM
Summary
The paper introduces GLaDiGAtor, a novel Graph Neural Network framework for predicting disease-gene associations. It constructs a heterogeneous biological graph integrating gene-gene, disease-disease, and gene-disease interactions from databases like DisGeNET and BioGRID. The model enriches nodes with contextual features using ProtT5 for protein sequences and BioBERT for disease text. GLaDiGAtor employs an encoder-decoder architecture with a GCN encoder and bilinear decoder, achieving superior predictive accuracy and generalization compared to 14 existing methods.
Entities (10)
Relation Signals (9)
GLaDiGAtor ā solves ā Disease-Gene Association
confidence 95% Ā· GLaDiGAtor... for disease-gene association prediction.
GLaDiGAtor ā uses ā Graph Convolutional Network
confidence 95% Ā· GLaDiGAtor uses a GCN encoder to obtain dd-dimensional node embeddings
GLaDiGAtor ā usesdatafrom ā DisGeNET
confidence 95% Ā· The primary source of diseaseāgene associations is DisGeNET (v7.0)
GLaDiGAtor ā uses ā ProtT5
confidence 94% Ā· protein sequences are embedded using ProtT5 [7]
GLaDiGAtor ā uses ā BioBERT
confidence 94% Ā· disease descriptions are encoded via BioBERT [21]
GLaDiGAtor ā usesdatafrom ā BioGRID
confidence 92% Ā· We integrated proteināprotein interactions (PPIs) from the BioGRID database
GLaDiGAtor ā outperforms ā ResMGCN
confidence 90% Ā· Other strong performers included ResMGCN (0.925 ROC-AUC)
GLaDiGAtor ā outperforms ā HOGCN
confidence 90% Ā· GLaDiGAtor achieved a ROC-AUC of 0.950... The next best model (HOGCN... reached ROC-AUC 0.936
GLaDiGAtor ā ā
Cypher Suggestions (0)
No Cypher suggestions yet.
Abstract
Abstract:Understanding disease-gene associations is essential for unravelling disease mechanisms and advancing diagnostics and therapeutics. Traditional approaches based on manual curation and literature review are labour-intensive and not scalable, prompting the use of machine learning on large biomedical data. In particular, graph neural networks (GNNs) have shown promise for modelling complex biological relationships. To address limitations in existing models, we propose GLaDiGAtor (Graph Learning-bAsed DIsease-Gene AssociaTiOn pRediction), a novel GNN framework with an encoder-decoder architecture for disease-gene association prediction. GLaDiGAtor constructs a heterogeneous biological graph integrating gene-gene, disease-disease, and gene-disease interactions from curated databases, and enriches each node with contextual features from well-known language models (ProtT5 for protein sequences and BioBERT for disease text). In evaluations, our model achieves superior predictive accuracy and generalisation, outperforming 14 existing methods. Literature-supported case studies confirm the biological relevance of high-confidence novel predictions, highlighting GLaDiGAtor's potential to discover candidate disease genes. These results underscore the power of graph convolutional networks in biomedical informatics and may ultimately facilitate drug discovery by revealing new gene-disease links. The source code and processed datasets are publicly available at this https URL.
Tags
Links
- Source: https://arxiv.org/abs/2602.18769v1
- Canonical: https://arxiv.org/abs/2602.18769v1
Trouble viewing inline? Open PDF directly ā
Full Text
46,475 characters extracted from source content.
Expand or collapse full text
GLaDiGAtor: Language-Model-Augmented Multi-Relation Graph Learning for Predicting DiseaseāGene Associations Osman Onur Kuzucu and Tunca DoÄan O. O. Kuzucu and T. DoÄan are with the Biological Data Science Lab, Dept. of Computer Engineering, Hacettepe University, 06800, Ankara, Turkey.T. DoÄan is also with the Dept. of Bioinformatics, Graduate School of Health Sciences, Hacettepe University, 06800, Ankara, Turkey.T. DoÄan is also with the Dept. of Health Informatics, Institute of Informatics, Hacettepe University, 06800, Ankara, Turkey.Correspondence to: Tunca DoÄan (tuncadogan@gmail.com). Abstract Understanding disease-gene associations is essential for unravelling disease mechanisms and advancing diagnostics and therapeutics. Traditional approaches based on manual curation and literature review are labour-intensive and not scalable, prompting the use of machine learning on large biomedical data. In particular, graph neural networks (GNNs) have shown promise for modelling complex biological relationships. To address limitations in existing models, we propose GLaDiGAtor (Graph Learning-bAsed DIseaseāGene AssociaTiOn pRediction), a novel GNN framework with an encoderādecoder architecture for diseaseāgene association prediction. GLaDiGAtor constructs a heterogeneous biological graph integrating geneāgene, diseaseādisease, and geneādisease interactions from curated databases, and enriches each node with contextual features from well-known language models (ProtT5 for protein sequences and BioBERT for disease text). In evaluations, our model achieves superior predictive accuracy and generalisation, outperforming 14 existing methods. Literature-supported case studies confirm the biological relevance of high-confidence novel predictions, highlighting GLaDiGAtorās potential to discover candidate disease genes. These results underscore the power of graph convolutional networks in biomedical informatics and may ultimately facilitate drug discovery by revealing new geneādisease links. The source code and processed datasets are publicly available at https://github.com/HUBioDataLab/GLaDiGAtor. I Introduction Uncovering the molecular basis of human disease is a central challenge in biomedical research and directly informs target identification, patient stratification, and therapeutic development [40, 35]. In particular, establishing reliable diseaseāgene associations helps elucidate disease mechanisms, supports variant interpretation in clinical genetics, and enables the prioritization of candidate genes for downstream experimental and therapeutic discovery studies [4]. Conventional strategies for uncovering diseaseāgene associations often rely on manual curation and literature review, which, while valuable, are limited by scalability, update frequency, and susceptibility to human bias. To address these limitations, the research community has increasingly adopted machine learning techniques capable of learning from large-scale biological data [12]. In particular, graph-based learning models have proven effective in capturing relational dependencies among biological entities such as genes, proteins, diseases and drugs. From a data-structural and modeling perspective, diseaseāgene association discovery can be viewed as link prediction on a heterogeneous graph, enabling representation learning for hypothesis generation; similar graph-based formulations also underpin related biomedical tasks such as drug/compoundātarget interaction (DTI) prediction [6] and graph-generative molecular design [42]. Numerous GNN-based methods have emerged in recent years to tackle prediction tasks within biological graphs. For example, SkipGNN [14], HOGCN [16], and ResMGCN [48] leverage neighborhood-level message passing for accurate molecular and disease interaction predictions. Foundational models like Graph Convolutional Networks (GCN) [18], variational graph autoencoders (VGAEs) [19], and graph isomorphism networks (GINs) [46] have helped define the theoretical landscape for graph-based learning, while works like JK-Net [47] and MixHop [1] address architectural depth and expressive power. Beyond structural modeling, other approaches have focused on similarity heuristics such as L3 [20] and spectral clustering [39], or utilized unsupervised network embeddings through methods like DeepWalk [31], node2vec [9], and struc2vec [34]. Despite these advancements, a persistent challenge lies in achieving high generalization, integration of rich biological semantics, and prediction accuracy for novel diseaseāgene associations. In this study, we present GLaDiGAtor (Graph Learning bAsed DIseaseāGene AssociaTiOn pRediction), a deep learning framework designed to address these gaps. GLaDiGAtor employs an encoderādecoder architecture to predict associations between diseases and genes using a heterogeneous graph structure. It integrates three types of biological relationshipsāgeneāgene, diseaseādisease, and geneādiseaseāfrom curated biomedical databases. Furthermore, we incorporate contextual node features derived from language models: protein sequences are embedded using ProtT5 [7], while disease descriptions are encoded via BioBERT [21]. This multi-relation representation enables GLaDiGAtor to learn enriched feature spaces and infer high-confidence associations. For reproducibility, GLaDiGAtorās source code and datasets are pubicly shared at https://github.com/HUBioDataLab/GLaDiGAtor. The primary contributions of this work are as follows: ⢠A new GCN-based architecture: We develop GLaDiGAtor, a new graph convolutional network model using an encoderādecoder paradigm on a heterogeneous geneādisease graph. This design enables the integration of multiple relation types and direct prediction of diseaseāgene links. ⢠Contextual biological embeddings: We enhance node feature representations with well-known high-performance pretrained embeddings. Each gene/protein node is represented by a ProtT5 sequence embedding, and each disease node by a BioBERT text embedding, providing rich biochemical and semantic context to the model. ⢠State-of-the-art performance: We benchmark GLaDiGAtor against 14 existing methods for diseaseāgene association prediction, demonstrating consistent improvements in predictive accuracy and generalisation across evaluation datasets. ⢠Literature validation of predictions: We validate GLaDiGAtorās novel predictions through literature evidence, confirming the biological plausibility of newly predicted geneādisease associations. These case studies illustrate the modelās utility in prioritising candidate genes for diseases, with potential implications for biomedical discovery and drug development. Figure 1: Workflow of the proposed GLaDiGAtor model, illustrating the end-to-end pipeline from input data preparation to encoderādecoder-based graph learning and diseaseāgene association prediction. I Materials and Methods I-A Heterogeneous Graph Construction We constructed a comprehensive heterogeneous graph that includes genes and diseases as nodes, with three types of edges representing known relationships among them. The primary source of diseaseāgene associations is DisGeNET (v7.0) [32], a large database of geneādisease links compiled from multiple resources. DisGeNET provides 1,134,942 geneādisease associations (GDAs) [5] with confidence scores and evidence annotations. We obtained two versions of the GDA data from DisGeNET: (a) the complete set via the DisGeNET API, which includes associations aggregated from literature and various databases (each GDA has a score reflecting evidence strength), and (b) a curated subset of high-confidence associations as identified in DisGeNETās expert-curated section. In this study, we treat proteins and their coding genes as one entity; therefore, we created a single node for each gene/protein. Naturally, diseaseāgene associations also become diseaseāprotein relationships. We integrated proteināprotein interactions (PPIs) from the BioGRID database [29] (also called geneāgene associationsāGGAsāhere, for the sake of consistency in naming) and diseaseādisease associations (DDAs) from DisGeNET (computed as disease semantic or phenotypic similarity, e.g., Jaccard indices). We also considered an external knowledge graph: the Open Graph Benchmark (OGB) ogbl-biokg dataset [13], from which we extracted proteinādisease edges as an alternative source of associations for comparison with methods from the literature (this forms one graph variant, described below). The rest of the relationship types are gathered as described above. Using these sources, we assembled six graph variants to evaluate the model under different data conditions (summarized from Graph 1 to Graph 6). Each graph follows the same construction pipeline but differs in the origin or filtering of the edges: ⢠Graph 1: DisGeNET (comprehensive data), GDAs filtered to high confidence (score ā„0.9ā„ 0.9). ⢠Graph 2: DisGeNET, GDAs with score ā„0.5ā„ 0.5. ⢠Graph 3: DisGeNET, GDAs with score ā„0.1ā„ 0.1. ⢠Graph 4: DisGeNET, GDAs with score ā„0.05ā„ 0.05. ⢠Graph 5: DisGeNET, curated GDAs (no score threshold, all curated associations). ⢠Graph 6: OGB (ogbl-biokg), proteinādisease associations, in place of DisGeNET GDAs. All graph variants include the same set of geneāgene interactions from BioGRID and diseaseādisease similarity links from DisGeNET (as available) to provide additional context. Graph statistics are provided in Table I. TABLE I: Summary statistics of constructed heterogeneous input graphs. Graph Genes Diseases GGA DDA GDA Graph 1 741 836 3,274 462 899 Graph 2 4,188 5,989 74,180 11,114 10,152 Graph 3 13,881 20,481 494,378 120,409 335,260 Graph 4 14,185 21,220 513,255 127,112 375,609 Graph 5 8,770 10,231 269,384 30,063 78,097 Graph 6 8,581 9,683 9,058 27,782 71,064 The number of nodes and edges varies widely across these graphs, from Graph 1 (the smallest, with under 1k genes and diseases) to Graph 4 (the largest, with ā¼ 14k genes, ā¼ 21k diseases, and over 1 million total edges across GGA, DDA, GDA). We used consistent node identifiers (NCBI gene IDs and UMLS disease concepts) across all datasets to allow meaningful comparisons. I-B Node Feature Representations A key feature of GLaDiGAtor is the integration of pretrained embeddings for both gene and disease nodes, providing biological context beyond the graph topology. We generated protein sequence embeddings for each gene node using ProtT5 [7], a transformer model trained on large protein sequence databases. Specifically, for each geneās protein product, we obtained the ProtT5 embedding (a 1024-dimensional vector) that captures biochemical and evolutionary features of the amino acid sequence. These embeddings encapsulate properties relevant to protein function and interactions. Similarly, for each disease node, we derived a textual embedding using BioBERT [21], a BERT-based language model pre-trained on biomedical literature. We input the diseaseās name and description into BioBERT to obtain a 768-dimensional vector representation reflecting the semantic context of the disease. This provides the model with an understanding of disease-related information in textual form. Because the gene and disease embeddings have different dimensionalities (1024 vs. 768), we performed a simple feature alignment to combine them. We zero-padded each vector to a fixed length of 1792 (the sum of 1024 and 768). In practice, we pad the gene vectors with zeros at the end (suffix) and the disease vectors with zeros at the beginning, so that both become 1792-dimensional and can be used in the same feature space. This differential padding scheme was chosen based on preliminary testing of alternatives (as discussed in the ablation study): padding gene and disease embeddings on opposite ends yielded slightly better performance than other alignment strategies. After alignment, each node (whether gene or disease) is represented by a 1792-dimensional initial feature vector. These node features are used as input to the graph neural network. I-C Dataset Finalisation and Partitioning Once the graph was constructed, we performed a UniRef protein sequence clusters-based [38] data split to prepare training and evaluation sets. To prevent information leakage due to closely related genes/proteins appearing in both training and test sets, we grouped them by their UniRef50 protein clusters. Proteins within the same cluster (i.e., sharing ā„50%ā„ 50\% sequence identity) were assigned to the same data partition. After enforcing this, we randomly split the geneādisease edges (the target associations) into training, validation, and test sets in an 8:1:1 ratio. The same protocol is used across all graph variants in Table I. All geneāgene and diseaseādisease edges were kept available during training (they are considered part of the known network context, not prediction targets) unless explicitly removed in an ablation experiment. For model supervision, we treated the presence of a geneādisease link as a positive instance. We then generated an equal number of negative instances by sampling geneādisease pairs that are not connected in the graph, ensuring that each negative instance consists of a gene and a disease that do not have a known association. Negative sampling was type-constrained (only gene nodes paired with disease nodes) and one-to-one balanced to match the count of positive edges in each split. This strategy yields a balanced binary classification setup. We also applied degree-aware sampling for negatives to mitigate bias: when drawing negative edges, diseases were weighted by node degree so that high-degree disease nodes were sampled with higher probability. This preserves a realistic difficulty where widely studied diseases have many candidate gene links to reject potentially. We assembled graphs in NetworkX [10] for flexible manipulation and integrity checks, then converted them to PyTorch Geometric [8] format to enable efficient mini-batch training and GPU-accelerated message passing in GNNs. I-D Model Architecture We model a heterogeneous, undirected graph G=(V,E)G=(V,E) with three relation types R=G,D,GDR=\G,D,GD\. Each node has a 1Ć17921Ć 1792 feature vector (aligned gene/protein and disease embeddings). GLaDiGAtor uses a GCN encoder to obtain d-dimensional node embeddings (d=28d=28) and a bilinear decoder to score geneādisease links. The end-to-end pipeline follows the draft: relation-wise normalization with self-loops, convex mixing of relations, two linear GCN transforms with ReLU, and a bilinear logit scored with a sigmoid for probabilities. I-D1 Encoder The encoder of GLaDiGAtor is a two-layer Graph Convolutional Network (GCN) that transforms high-dimensional node features into compact latent embeddings. Let the input feature matrix be given by Xāā|V|Ć1792X ^|V|Ć 1792 (1) where |V||V| is the number of nodes and each node is described by 1792 features. We denote H(0)=XH^(0)=X. The first GCN layer aggregates information from neighboring nodes and applies a linear transformation followed by a non-linear activation: H(1)=Ļā(A~āH(0)āW(0)),H^(1)=Ļ\! ( AH^(0)W^(0) ), (2) where A~ A is the adjacency matrix of the constructed graph, W(0)āā1792Ć112W^(0) ^1792Ć 112 is a trainable weight matrix, and Ļā(ā )Ļ(Ā·) denotes the ReLU activation function. The second GCN layer further refines the node representations: Z=Ļā(A~āH(1)āW(1)),Z=Ļ\! ( AH^(1)W^(1) ), (3) with W(1)āā112Ć28W^(1) ^112Ć 28. The output Zāā|V|Ć28Z ^|V|Ć 28 provides a 28-dimensional latent embedding for each node. To mitigate overfitting, dropout with fixed rate p is applied after each GCN layer. The choice of two layers and a relatively low-dimensional final embedding (d=28d=28) was guided by validation experiments, balancing expressiveness and generalization. I-D2 Decoder The decoder of GLaDiGAtor takes the latent node embeddings produced by the encoder and generates predictions about potential relationships between node pairs. Specifically, it operates on the 28-dimensional embeddings Zāā|V|Ć28Z ^|V|Ć 28 obtained from the second GCN layer of the encoder. Given two nodes i and j, let hi,hjāā28h_i,h_j ^28 denote their embeddings from Z. The decoder proceeds in three steps. Element-wise multiplication: The embeddings are combined through element-wise multiplication: e=hiāhje=h_i h_j (4) where ā denotes the Hadamard product. This operation captures feature-level interactions between the two nodes. Summation: The resulting interaction vector e is reduced to a scalar score by summing over all dimensions: s=āk=128eks= _k=1^28e_k (5) This step aggregates the pairwise feature interactions into a single measure of similarity or compatibility. Sigmoid activation: Finally, the scalar score is passed through a sigmoid function: ziāj=Ļā(s)=11+expā”(ās),z_ij=Ļ(s)= 11+ (-s), (6) producing an output in the range (0,1)(0,1). This value can be interpreted as the probability of an edge existing between nodes i and j. The encoder maps high-dimensional input features into compact latent embeddings, while the decoder transforms these embeddings into probabilistic predictions of nodeānode relationships. Together, they form an end-to-end framework for learning graph-structured associations. I-D3 Training Objective and Optimization We formulate diseaseāgene association prediction as a binary edge classification task. For each candidate pair (u,v)(u,v), the decoder outputs a logit score sā(u,v)s(u,v), which is transformed into a probability y y via the sigmoid function. The model is trained to minimize the binary cross-entropy (BCE) loss between y y and the ground-truth label yā0,1yā\0,1\, where y=1y=1 denotes a known association and y=0y=0 a sampled negative. For numerical stability, we adopt the logits-based formulation of BCE (i.e., BCEWithLogitsLoss), which directly takes sā(u,v)s(u,v) as input: āā(s,y)=w1āyā logā”(1+eās)+w0ā(1āy)ā logā”(1+es), (s,y)=w_1\,yĀ· \! (1+e^-s )+w_0(1-y)Ā· \! (1+e^s ), (7) where w0w_0 and w1w_1 are optional class weights for negatives and positives. In our balanced dataset, we set w0=w1=1w_0=w_1=1, i.e., no weighting. Preliminary experiments with slightly higher positive weights (to reflect the open-world imbalance) did not yield significant improvements, so we retained the unweighted loss. The total loss for a mini-batch ā¬B is the average of āā(sā(u,v),y) (s(u,v),y) across all pairs. Optimization: We optimize parameters using the AdamW optimizer [17] with learning rate 0.001. Training proceeds for a fixed number of epochs T without aggressive scheduling or early stopping. The final model is selected as the checkpoint achieving the highest validation AUROC during training. All reported test results are computed at this checkpoint. Mini-batch training: Since the full graph may contain millions of edges, we adopt mini-batch training with subgraph sampling. At each iteration, we sample a batch ā¬B of positive and negative edges and construct the induced subgraph Gā¬G_B containing those edges and their immediate neighbors. The GCN encoder is then applied to Gā¬G_B, enabling message passing within the relevant local context. This strategy reduces memory usage and accelerates training, while ensuring that over multiple epochs the model is exposed to the entire graph. To further improve efficiency, we precompute auxiliary structures such as neighbor lists and UniRef cluster IDs. The per-batch complexity of a GCN layer with output dimension dād_ is ā(|Eā¬|ā dā)O(|E_B|Ā· d_ ), and batch sizes are chosen to fit within a single GPU. I-E Evaluation Metrics We evaluated model performance using standard binary classification metrics: Accuracy (Acc), F1-score (F1), Precision (Prec), Recall (Rec), Area Under the ROC Curve (ROC-AUC), Area Under the PrecisionāRecall Curve (PR-AUC), and Specificity / True Negative Rate (Spec). I Results I-A Performance Evaluation Across Different Input Graphs We first examine GLaDiGAtorās predictive performance on the six constructed graph variants (Graph 1ā6). Table I presents the modelās accuracy, F1, precision, recall, ROC-AUC, PR-AUC, and specificity on the validation set for each graph, and Table I shows the same for the test set. TABLE I: Validation performance of GLaDiGAtor across different input graphs. Graph ID Acc F1 Prec Rec ROC-AUC PR-AUC Spec 1 0.719 0.741 0.688 0.800 0.735 0.688 0.637 2 0.830 0.836 0.805 0.871 0.896 0.893 0.789 3 0.890 0.893 0.875 0.911 0.953 0.957 0.869 4 0.891 0.895 0.867 0.955 0.955 0.960 0.858 5 0.899 0.902 0.869 0.938 0.960 0.966 0.859 6 0.857 0.866 0.815 0.923 0.943 0.947 0.791 TABLE I: Test performance of GLaDiGAtor across different input graphs. Graph ID Acc F1 Prec Rec ROC-AUC PR-AUC Spec 1 0.761 0.786 0.711 0.877 0.807 0.746 0.644 2 0.827 0.840 0.784 0.900 0.904 0.900 0.750 3 0.890 0.892 0.874 0.911 0.953 0.957 0.869 4 0.888 0.892 0.867 0.918 0.954 0.959 0.860 5 0.889 0.893 0.859 0.930 0.955 0.960 0.848 6 0.876 0.886 0.821 0.962 0.965 0.967 0.791 Overall, GLaDiGAtor achieves high scores on all graphs, but the performance trends reflect the varying data characteristics: ⢠On the smallest and most stringent dataset (Graph 1, using only very high-confidence DisGeNET associations), the model attains a test ROC-AUC of 0.807 and PR-AUC of 0.746 with accuracy 0.761. The precisions and specificity on Graph 1 is relatively low (0.711 and 0.644), indicating some false positives, likely due to the limited number of associations making negatives harder to distinguish. ⢠As the graph becomes more densely connected and includes lower-confidence but more comprehensive associations (Graphs 2, 3, 4), performance improves markedly. By Graph 3 and Graph 4, GLaDiGAtor reaches ROC-AUC ā0.95ā 0.95 and PR-AUC ā0.96ā 0.96 on test data. These larger graphs provide a richer context, enabling the model to generalise better. ⢠Graph 5 (DisGeNET curated subset) yields performance comparable to Graph 4, with test ROC-AUC ā¼0.955 0.955 and PR-AUC ā¼0.960 0.960. It is also important to note that Graph 5 is a subset of Graph 4 focusing on high-quality associations. ⢠Graph 6, derived from the OGB biokg dataset, also achieves strong results (ROC-AUC ā¼0.965 0.965, PR-AUC ā¼0.967 0.967), with slightly lower precision and specificity values compared to Graphs 3, 4 and 5. This indicates that GLaDiGAtor can transfer to a different source of associations (here, proteinādisease edges from a knowledge graph) with minimal loss in performance. ⢠Overall, GLaDiGAtor demonstrates strong predictive performance under a stringent traināvalidationātest partitioning strategy stratified by 50% protein sequence similarity. I-B Comparison with Baseline Methods We compared GLaDiGAtor to 13 baseline models on the DisGeNET curated GDAs (Graph 5) dataset, to benchmark its performance against existing approaches. The baseline model performance scores are directly obtained from the SkipGNN [14], HOGCN [16] and ResMGCN [48] studies, since re-training and testing the models might lead to a decrease in their performance measurements due to using non-optimal hyperparameters. Table IV summarizes the ROC-AUC and PR-AUC for each method. GLaDiGAtor achieved a ROC-AUC of 0.950 and PR-AUC of 0.956, which is the highest among all methods evaluated. The next best model (HOGCN, a GCN variant for heterogeneous graphs) reached ROC-AUC 0.936 and PR-AUC 0.941, slightly below our model. Other strong performers included ResMGCN (0.925 ROC-AUC) and MixHop (0.916). Traditional network embedding approaches like node2vec or DeepWalk showed lower performance (ROC-AUC ā¼ 0.83) in this task, likely because they do not incorporate rich biological features. Simpler heuristics (L3, spectral clustering) and basic GCN/VGAE models also underperformed relative to our approach. These results demonstrate that GLaDiGAtor outperforms the state-of-the-art in diseaseāgene association prediction. The performance gain in GLaDiGAtor can be attributed to our modelās ability to integrate heterogeneous data (multiple edge types and node features) in a unified framework. In contrast, many baselines use either homogeneous networks or limited feature information. For example, SkipGNN and HOGCN incorporate graph structure but lack the multimodal feature integration that GLaDiGAtor has with ProtT5 and BioBERT embeddings. Our results suggest that combining graph topology with rich biological context (via pretrained embeddings) yields a more powerful predictor. It is important to note that there are discrepancies between GLaDiGAtorās performance results in Table I and Table IV due to different splitting ratios. In Table I, the splitting ratio is 8:1:1, whereas in Table IV, the splitting ratio is 7:1:2, as described in the SkipGNN study [14]. TABLE IV: Performance comparison with baseline models on the DisGeneNET curated GDAs benchmark (Graph 5). The ranking of methods is based on ROC-AUC. Method ROC-AUC PR-AUC Ranking GLaDiGAtor 0.950 0.956 1 HOGCN 0.936 0.941 2 ResMGCN 0.925 0.935 3 MixHop 0.916 0.912 4 SkipGNN 0.912 0.915 5 struc2vec 0.909 0.910 6 GCN 0.906 0.909 7 GIN 0.900 0.916 8 JK-Net 0.898 0.891 9 VGAE 0.873 0.902 10 SC 0.863 0.905 11 node2vec 0.834 0.828 12 L3 0.832 0.899 13 DeepWalk 0.832 0.827 14 I-C Ablation Study We conducted ablation experiments to assess the contribution of key design components in GLaDiGAtor. In each ablation, we modified one aspect of the model or data processing and observed the impact on performance (using Graph 2 as a representative setting). Validation and test scores are provided in Table V. ⢠Embedding aggregation strategy: We tried an alternative padding alignment for combining ProtT5 and BioBERT embeddings (e.g., appending 256 zeros to the terminus of the disease embeddings, making 1024-D in total, and no addition to protein embeddings, which is already 1024-D). We found that the alternative padding scheme led to a slight drop in performance (Table V) and less stable training. Our default approach preserves the full information from both modalities in a consistent orientation. ⢠Negative sampling strategy: We experimented with sampling all types of edges as negatives (including geneāgene or diseaseādisease false edges) versus our constrained approach (only geneādisease negatives, in the default model). Allowing arbitrary false edges hurts performance (Table V), whereas the constrained negative sampling improves stability and maintains focus on the biologically relevant task. In summary, the ablation results indicate that removing GLaDiGAtorās default modules leads to consistent performance degradation, most notably in accuracy and F1 score. The only metric that remains comparatively stable under ablation is the test ROC-AUC, suggesting that the full configuration is primarily responsible for improvements in classification quality at the operating point rather than changes in overall ranking performance. TABLE V: Ablation study performance comparison on the validation and test datasets. Embedding aggregation Negative edge gen. Split Acc F1 Prec Rec ROC-AUC Nā Yā Val. 0.823 0.826 0.810 0.843 0.895 Test 0.823 0.835 0.783 0.893 0.916 Y N Val. 0.739 0.782 0.671 0.938 0.870 Test 0.750 0.800 0.675 0.974 0.925 Y Y Val. 0.830 0.836 0.805 0.871 0.896 Test 0.827 0.840 0.784 0.900 0.904 āN: no, Y: yes. Our default model is given at the bottom. I-D Case Study Analysis To assess the biological relevance of GLaDiGAtorās predictions, we conducted literature validation of selected geneādisease associations derived from Graph 1 (GDA score ā„0.9ā„ 0.9). In this analysis, both positive and negative predictions are evaluated. For each selected case, GLaDiGAtorās predictions and database cross-references are provided in Table VI. TABLE VI: Case study geneādisease association examples (Yes: the corresponding gene-disease asssociation exists according to DisGeNET -constituting the training dataset-, GLaDiGAtor or the literature; No: otherwise). Gene Symbol Disease Name DisGeNET GLaDiGAtor Literature ABCA1 Tangier Disease Yes Yes Yes ABCA4 Stargardt Disease Yes Yes Yes ACADM MCADD Yes Yes Yes FANCA Polycystic Kidney Disease I No No No PINK1 Herlitz Disease No No No NPR2 PHARC Syndrome No No No PADI4 Rheumatoid Arthritis No Yes Yes GATA2 Emberger Syndrome No Yes Yes SETBP1 SchinzelāGiedion Syndrome No Yes Yes ⢠Known associations rediscovered: GLaDiGAtor correctly predicted established geneādisease relationships (Table VI). For instance, (i) the model linked ABCA1 with Tangier disease (TD), a rare autosomal recessive disorder marked by defective HDL metabolism [3, 26]. Mutations in ABCA1 impair cholesterol efflux, leading to HDL deficiency, peripheral cholesterol accumulation, and increased cardiovascular risk [33]. (i) GLaDiGAtor identified the link between ABCA4 and Stargardt disease (STGD), an inherited retinal degeneration. The gene encodes a photoreceptor transporter vital for vitamin A cycle homeostasis. Mutations cause toxic lipofuscin buildup in the retinal pigment epithelium, leading to central vision loss [22, 36, 25]. (i) GLaDiGAtor correctly predicted the association between ACADM and Medium-chain Acyl-CoA Dehydrogenase Deficiency (MCADD). Mutations impede mitochondrial fatty acid oxidation, causing metabolite buildup and symptoms like hypoglycemia and seizures [11, 44]. ⢠Correct rejection of false links: GLaDiGAtor also correctly rejected several geneādisease pairings that might seem plausible but are not supported biologically (Table VI). For example, (i) FANCA and Polycystic Kidney Disease I (PKD1): GLaDiGAtor correctly rejected the association between FANCA and PKD1. While FANCA mutations cause Fanconi anemia (DNA repair defect), PKD1 stems from PKD1 gene mutations affecting renal cell architecture [30, 24]. (i) GLaDiGAtor predicted no association between PINK1 and Herlitz disease. PINK1 mutations are linked to early-onset Parkinsonās disease and mitochondrial dysfunction [43], while Herlitz disease is an autoimmune blistering disorder targeting type VII collagen [23]. (i) GLaDiGAtor correctly identified no link between NPR2 and PHARC syndrome. PHARC arises from ABHD12 mutations and affects neurological and sensory functions [27]. NPR2 encodes a membrane receptor with guanylate cyclase activity (GC-B/NPR-B) that binds C-type natriuretic peptide (CNP), thereby catalyzing cGMP production [41]. This function is unrelated to PHARC pathophysiology. ⢠Novel predictions supported by literature: Importantly, GLaDiGAtor proposed several geneādisease associations that were not present in DisGeNET (at the time of our analysis) but for which we found independent literature evidence, suggesting they are valid novel findings (Table VI). Three compelling examples are: (i) GLaDiGAtor accurately predicted the association between PADI4 and Rheumatoid Arthritis (RA). PADI4 encodes a protein citrullination enzyme that produces autoantigens implicated in RA. Multiple studies reported a positive association between PADI4 variants and RA susceptibility [15]. (i) GLaDiGAtor correctly linked GATA2 to Emberger syndrome, a condition combining immunodeficiency, lymphedema, and MDS/AML. Haploinsufficiency of GATA2 disrupts hematopoietic and lymphatic regulation [37, 28]. (i) GLaDiGAtor predicted the association between SETBP1 and SchinzelāGiedion syndrome (SGS), which is caused by recurrent de novo gain-of-function SETBP1 variants [2] and is supported by transcriptomic dysregulation in a Setbp1S858R mouse model [45]. None of these geneādisease pairs were indexed in DisGeNET at the time of our analysis, yet our model predicted them as positive associations (which are counted as false positives in our performance evaluation). However, subsequent literature review revealed studies supporting each link. IV Discussion In this work, we introduced GLaDiGAtor, a machine learning framework that integrates graph representation learning with contextual biomedical embeddings to predict diseaseāgene associations. The motivation for GLaDiGAtor arises from the growing need for scalable, accurate methods to map the genetic landscape of diseases, as traditional curation cannot keep up with data growth. GLaDiGAtor performs multi-relation graph convolution over geneāgene, diseaseādisease, and geneādisease edges, and augments nodes with ProtT5 (protein) and BioBERT (disease-text) embeddings. Model comparisons with state-of-the-art models (e.g., HOGCN [16], ResMGCN [48], SkipGNN [14]) demonstrate GLaDiGAtorās superior or competitive performance in diseaseāgene association prediction. Ablation studies confirm the importance of effective embedding aggregation and constrained negative edge generation strategies. The case study analysis highlights GLaDiGAtorās robust ability to prioritise biologically meaningful geneādisease associations. Among the nine evaluated cases derived from Graph 1 (GDA score ā„0.9ā„ 0.9), the model consistently aligned with known ground truth, correctly identifying both true and false associations. IV-A Limitations Despite the promising outcomes, our approach has several limitations that must be acknowledged: ⢠Data dependence: The modelās accuracy is influenced by the quality, completeness, and biases of the input databases. If important geneādisease links are missing or if spurious associations are present in the data, the modelās learning and outputs will be affected. GLaDiGAtor inherits any biases in DisGeNET or other sources (e.g., well-studied genes might appear overly important). ⢠Computational complexity: Training a GCN on large heterogeneous graphs is resource-intensive. GLaDiGAtorās scalability may be constrained by memory and time requirements, especially as we move to graphs with millions of nodes or edges. Techniques such as mini-batching and subgraph sampling help, but very large graphs or those requiring higher-dimensional embeddings could pose challenges. ⢠Limited interpretability of predictions: Although the model can rank candidate geneādisease links, it does not explicitly provide mechanistic rationales (e.g., which specific neighbors, relation types, or embedding dimensions drive an individual prediction). Additional explainability analyses would be needed to support clinical or experimental decision-making. IV-B Future Work Future work should explore improved embedding algorithms, alternative GCN architectures, dynamic graph learning, and expanded datasets to refine accuracy and increase real-world applicability. Extending GLaDiGAtor toward multimodal association prediction represents a promising direction for advancing the field of biological/biomedical association prediction. V Conclusion GLaDiGAtor provides a scalable, graph-driven approach for prioritising gene candidates for genetic diseases by combining heterogeneous network structure with contextual biomedical embeddings. The proposed framework achieves strong predictive performance, making it a practical tool for guiding follow-up biomedical investigations. References [1] S. Abu-El-Haija, B. Perozzi, A. Kapoor, N. Alipourfard, K. Lerman, H. Harutyunyan, G. Ver Steeg, and A. Galstyan (2019) MixHop: higher-order graph convolutional architectures via sparsified neighborhood mixing. In International Conference on Machine Learning, p. 21ā29. Cited by: §I. [2] R. Acuna-Hidalgo et al. (2017) Overlapping setbp1 gain-of-function mutations in schinzelāgiedion syndrome and hematologic malignancies. PLOS Genetics 13 (3), p. e1006683. Cited by: 3rd item. [3] S. Barbosa-Gouveia, S. Fernandez-Crespo, H. Lazare-Iglesias, A. Gonzalez-Quintela, N. Vazquez-Agra, and A. Hermida-Ameijeiras (2023) Association of a novel homozygous variant in abca1 gene with tangier disease. Journal of Clinical Medicine 12. Cited by: 1st item. [4] D. Botstein and N. Risch (2003) Discovering genotypes underlying human phenotypes: past successes and future approaches. Nature Genetics 33 (Suppl), p. 228ā237. Cited by: §I. [5] DisGeNET DisGeNET database information. Note: Accessed: 2024-02-27 Cited by: §I-A. [6] T. DoÄan, E. A. Güzelcan, M. Baumann, A. Koyas, H. AtaÅ, I. R. Baxendale, M. Martin, and R. Cetin-Atalay (2021) Protein domain-based prediction of drug/compoundātarget interactions and experimental validation on lim kinases. PLOS Computational Biology 17 (11), p. e1009171. Cited by: §I. [7] A. Elnaggar, M. Heinzinger, C. Dallago, G. Rehawi, Y. Wang, L. Jones, others, and B. Rost (2021) ProtTrans: toward understanding the language of life through self-supervised learning. IEEE Transactions on Pattern Analysis and Machine Intelligence 44 (10), p. 7112ā7127. Cited by: §I, §I-B. [8] M. Fey and J. Lenssen (2019) Fast graph representation learning with pytorch geometric. Note: ICLR Workshop on Representation Learning on Graphs and Manifolds Cited by: §I-C. [9] A. Grover and J. Leskovec (2016) Node2vec: scalable feature learning for networks. In Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, p. 855ā864. Cited by: §I. [10] A. Hagberg, P. Swart, and D. Chult (2008) Exploring network structure, dynamics, and function using networkx. Technical report Los Alamos National Laboratory (LANL). Cited by: §I-C. [11] K. Hara, G. Tajima, S. Okada, M. Tsumura, R. Kagawa, K. Shirao, Y. Ohno, S. Yasunaga, M. Ohtsubo, I. Hata, N. Sakura, Y. Shigematsu, Y. Takihara, and M. Kobayashi (2016) Significance of acadm mutations identified through newborn screening of mcad deficiency in japan. Molecular Genetics and Metabolism 118, p. 9ā14. Cited by: 1st item. [12] M. He, C. Huang, B. Liu, Y. Wang, and J. Li (2021) Factor graph-aggregated heterogeneous network embedding for disease-gene association prediction. BMC Bioinformatics 22, p. 165. Cited by: §I. [13] W. Hu, M. Fey, M. Zitnik, Y. Dong, H. Ren, B. Liu, M. Catasta, and J. Leskovec (2021) Open graph benchmark: datasets for machine learning on graphs. Cited by: §I-A. [14] K. Huang, C. Xiao, L. Glass, M. Zitnik, and J. Sun (2020) SkipGNN: predicting molecular interactions with skip-graph networks. Scientific Reports 10. Cited by: §I, §I-B, §I-B, §IV. [15] T. Iwamoto, K. Ikari, T. Nakamura, M. Kuwahara, Y. Toyama, T. Tomatsu, others, and N. Kamatani (2006) Association between PADI4 and rheumatoid arthritis: a meta-analysis. Rheumatology 45 (7), p. 804ā807. Cited by: 3rd item. [16] K. KC, R. Li, F. Cui, and A. Haake (2022) Predicting biomedical interactions with higher-order graph convolutional networks. IEEE/ACM Transactions on Computational Biology and Bioinformatics 19, p. 676ā687. Cited by: §I, §I-B, §IV. [17] D. P. Kingma and J. Ba (2014) Adam: a method for stochastic optimization. Note: arXiv preprint arXiv:1412.6980 Cited by: §I-D3. [18] T. N. Kipf and M. Welling (2016) Semi-supervised classification with graph convolutional networks. Note: arXiv preprint arXiv:1609.02907 Cited by: §I. [19] T. N. Kipf and M. Welling (2016) Variational graph auto-encoders. Note: arXiv preprint arXiv:1611.07308 Cited by: §I. [20] I. Kovacs, K. Luck, K. Spirohn, Y. Wang, C. Pollis, W. Bian, D. Kim, N. Kishore, T. Hao, et al. (2019) Network-based prediction of protein interactions. Nature Communications 10, p. 1240. Cited by: §I. [21] J. Lee, W. Yoon, S. Kim, D. Kim, S. Kim, C. So, and J. Kang (2020) BioBERT: a pre-trained biomedical language representation model for biomedical text mining. Bioinformatics 36, p. 1234ā1240. Cited by: §I, §I-B. [22] L. Molday, D. Wahl, M. Sarunic, and R. Molday (2018) Localization and functional characterization of the p.asn965ser (n965s) abca4 variant in mice reveal pathogenic mechanisms underlying stargardt macular degeneration. Human Molecular Genetics 27, p. 295ā306. Cited by: 1st item. [23] NCBI MedGen (2024) Junctional epidermolysis bullosa, herlitz type. Note: National Center for Biotechnology Information (NCBI)Accessed: 2024-02-27 External Links: Link Cited by: 2nd item. [24] NCBI MedGen (2024) Polycystic kidney disease, adult type. Note: National Center for Biotechnology Information (NCBI)Accessed: 2024-02-27 External Links: Link Cited by: 2nd item. [25] NCBI MedGen (2024) Stargardt disease. Note: National Center for Biotechnology Information (NCBI)Accessed: 2024-02-27 External Links: Link Cited by: 1st item. [26] NCBI MedGen (2024) Tangier disease. Note: National Center for Biotechnology Information (NCBI)Accessed: 2024-02-27 External Links: Link Cited by: 1st item. [27] OMIM (2024) Polyneuropathy, hearing loss, ataxia, retinitis pigmentosa, and cataract. Note: Online Mendelian Inheritance in Man (OMIM)Accessed: 2024-02-27 External Links: Link Cited by: 2nd item. [28] P. Ostergaard, M. A. Simpson, F. C. Connell, C. G. Steward, G. Brice, W. J. Woollard, others, and S. Mansour (2011) Mutations in GATA2 cause primary lymphedema associated with a predisposition to acute myeloid leukemia (emberger syndrome). Nature Genetics 43 (10), p. 929ā931. Cited by: 3rd item. [29] R. Oughtred, C. Stark, B. Breitkreutz, J. Rust, L. Boucher, C. Chang, N. Kolas, L. OāDonnell, G. Leung, R. McAdam, et al. (2019) The biogrid interaction database: 2019 update. Nucleic Acids Research 47, p. D529āD541. Cited by: §I-A. [30] J. D. Peake and E. Noguchi (2022) Fanconi anemia: current insights regarding epidemiology, cancer, and dna repair. Human Genetics 141 (12), p. 1811ā1836. Cited by: 2nd item. [31] B. Perozzi, R. Al-Rfou, and S. Skiena (2014) DeepWalk: online learning of social representations. In Proceedings of the 20th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, p. 701ā710. Cited by: §I. [32] J. PiƱero, J. M. RamĆrez-Anguita, J. Sauch-Pitarch, F. Ronzano, E. Centeno, F. Sanz, and L. I. Furlong (2020) The DisGeNET knowledge platform for disease genomics: 2019 update. Nucleic Acids Research 48 (D1), p. D845āD855. Cited by: §I-A. [33] F. Quazi and R. Molday (2013) Differential phospholipid substrates and directional transport by atp-binding cassette proteins abca1, abca7, and abca4 and disease-causing mutants. Journal of Biological Chemistry 288, p. 34414ā34426. Cited by: 1st item. [34] L. Ribeiro, P. Saverese, and D. Figueiredo (2017) Struc2vec: learning node representations from structural identity. In Proceedings of the 23rd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, p. 385ā394. Cited by: §I. [35] S. Richards, N. Aziz, S. Bale, D. Bick, S. Das, J. Gastier-Foster, W. W. Grody, M. Hegde, E. Lyon, E. Spector, K. Voelkerding, and H. L. Rehm (2015) Standards and guidelines for the interpretation of sequence variants: a joint consensus recommendation of the American College of Medical Genetics and Genomics and the Association for Molecular Pathology. Genetics in Medicine 17 (5), p. 405ā424. Cited by: §I. [36] N. Shroyer, R. Lewis, A. Yatsenko, and J. Lupski (2001) Null missense abcr (abca4) mutations in a family with stargardt disease and retinitis pigmentosa. Investigative Ophthalmology & Visual Science 42, p. 2757ā2761. Cited by: 1st item. [37] M. A. Spinner, L. A. Sanchez, A. P. Hsu, P. A. Shaw, C. S. Zerbe, K. R. Calvo, D. C. Arthur, W. Gu, C. M. Gould, C. C. Brewer, E. W. Cowen, A. F. Freeman, K. N. Olivier, G. Uzel, A. M. Zelazny, J. R. Daub, C. D. Spalding, R. J. Claypool, N. K. Giri, B. P. Alter, E. M. Mace, J. S. Orange, J. Cuellar-Rodriguez, D. D. Hickstein, and S. M. Holland (2014) GATA2 deficiency: a protean disorder of hematopoiesis, lymphatics, and immunity. Blood 123 (6), p. 809ā821. External Links: Document Cited by: 3rd item. [38] B. E. Suzek, Y. Wang, H. Huang, P. McGarvey, C. Wu, and U. Consortium (2015) UniRef clusters: a comprehensive and scalable alternative for improving sequence similarity searches. Bioinformatics 31, p. 926ā932. Cited by: §I-C. [39] L. Tang and H. Liu (2011) Leveraging social media networks for classification. Data Mining and Knowledge Discovery 23, p. 447ā478. Cited by: §I. [40] E. Topol (2019) High-performance medicine: the convergence of human and artificial intelligence. Nature Medicine 25, p. 44ā56. Cited by: §I. [41] UniProt Consortium (2024) Natriuretic peptide receptor 2 (Homo sapiens) (npr2) (uniprotkb: p20594). Note: UniProtAccessed: 2024-02-27 External Links: Link Cited by: 2nd item. [42] A. Ćnlü, E. Ćevrim, M. G. YiÄit, A. Sarıgün, H. Ćelikbilek, O. Bayram, others, and T. DoÄan (2025) Target-specific de novo design of drug candidate molecules with graph-transformer-based generative adversarial networks. Nature Machine Intelligence 7 (9), p. 1524ā1540. Cited by: §I. [43] E. M. Valente, P. M. Abou-Sleiman, V. Caputo, M. M. K. Muqit, K. Harvey, S. Gispert, Z. Ali, D. Del Turco, A. R. Bentivoglio, D. G. Healy, A. Albanese, R. Nussbaum, R. Gonzalez-Maldonado, T. Deller, S. Salvi, P. Cortelli, W. P. Gilks, D. S. Latchman, R. J. Harvey, B. Dallapiccola, G. Auburger, and N. W. Wood (2004) Hereditary early-onset parkinsonās disease caused by mutations in PINK1. Science 304 (5674), p. 1158ā1160. Cited by: 2nd item. [44] F. Ventura, P. Leandro, A. Luz, I. Rivera, M. Silva, R. Ramos, H. Rocha, A. Lopes, H. Fonseca, A. Gaspar, L. Diogo, E. Martins, E. Leao-Teles, L. Vilarinho, and I. Almeida (2014) Retrospective study of the medium-chain acyl-coa dehydrogenase deficiency in portugal. Clinical Genetics 85, p. 555ā561. Cited by: 1st item. [45] J. H. Whitlock et al. (2023) Cell-type-specific gene expression and regulation in the cerebral cortex and kidney of atypical setbp1 s858r schinzelāgiedion syndrome mice. Journal of Cellular and Molecular Medicine 27 (22), p. 3565ā3577. Cited by: 3rd item. [46] K. Xu, W. Hu, J. Leskovec, and S. Jegelka (2018) How powerful are graph neural networks?. Note: arXiv preprint arXiv:1810.00826 Cited by: §I. [47] K. Xu, C. Li, Y. Tian, T. Sonobe, K. Kawarabayashi, and S. Jegelka (2018) Representation learning on graphs with jumping knowledge networks. In International Conference on Machine Learning, p. 5453ā5462. Cited by: §I. [48] Z. Yin (2023) ResMGCN: residual message graph convolution network for fast biomedical interactions discovering. Note: arXiv preprint arXiv:2311.07632 Cited by: §I, §I-B, §IV.