HGCN(O): A Self-Tuning GCN HyperModel Toolkit for Outcome Prediction in Event-Sequence Data
Organizations: College of Computing and Mathematical Sciences, Khalifa University, P.O. Box 127788, Abu Dhabi, United Arab Emirates · Department of Computer Science, University of Milan, Via Festa del Perdono 7, 20122, Milano, MI, Italy
Abstract
Outcome prediction from event sequence data requires effective use of information at two distinct semantic levels: attributes of individual events and attributes of the complete sequence. Although graph models can explicitly represent temporal relations between events, existing approaches commonly rely on predefined representations and manually configured architectures, with limited consideration of how event and sequence information should be incorporated at their corresponding GNN levels or how GNN models should be fitted to different datasets without manual tuning. We propose HGCN(O), an end to end self tuning graph toolkit for hierarchical information representation and automated model configuration. Its core T-GCN architecture processes event information at the node level and sequence information at the graph level, while O-GCN provides a one level reference for early integration of sequence attributes. TP-GCN and TE-GCN extend T-GCN with a pseudo informative duration embedding and a learned activity embedding, respectively. HGCN(O) integrates these architecture specific designs with hypermodel optimization to jointly determine model structure and training configuration. The core T-GCN achieves F1 scores of 1.0000 on two balanced benchmarks and a weighted F1 of 0.8639 on a highly imbalanced dataset, compared with 0.8318 for GAT and 0.8453 for TransformerConv. Prefix and ablation analyses further demonstrate its effectiveness for outcome prediction from incomplete sequences and quantify the contributions of temporal edge weights and graph level sequence attributes.
Figures & tables
| Model | Role | Event Level Input | Sequence Level Input | Sequence Attribute Integration | Auxiliary Representation | Key Formulation |
| O-GCN | Reference | Event attributes | Sequence attributes replicated across nodes | Node level, before graph convolution | None | |
| T-GCN | Core model | Event attributes | Sequence attributes | Graph level, after pooling | None | |
| TP-GCN | Extension | Event attributes | Sequence attributes | Graph level, after pooling | Pseudo informative duration embedding | |
| TE-GCN | Extension | Event attributes | Sequence attributes | Graph level, after pooling | Learned activity embedding; optional pseudo embedding | or |
| Component | Hyperparameter | Search Space |
| Architecture | Graph convolution layers | 1-5 per applicable branch |
| Graph convolution hidden units | 32-256 per layer | |
| Fully connected layers | 1-3 per applicable branch | |
| Fully connected hidden units | 32-256 per layer | |
| TE embedding dimension | 10-50 | |
| Graph pooling | mean, max, add |
| Dataset | #Seq | #Act | #Cls | IR | Med( ) | Range( ) | Med( ) | Graph attr | Node attr |
| BPI12-A | 6,672 | 9 | 3 | 1:1 | 6 | 3-7 | 14,579 | 1 numerical | universal: 2 categorical |
| BPI12-O | 2,406 | 7 | 3 | 1:1 | 5 | 4-30 | 20,542 | 1 numerical | universal: 2 categorical |
| Patients | 2,140 | 10 | 6 | 36:1 | 7 | 4-9 | 8,505 | 3 numerical | specific: 3 numerical; 1 categorical |
| 1 categorical | universal: 1 categorical |
| Dataset | Model | Accuracy | Macro Average | Weighted Average | ||||
| Precision | Recall | F1 | Precision | Recall | F1 | |||
| BPI12-A | T-GCNConv | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 |
| T-GraphConv | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | |
| GAT | 0.9610 | 0.9649 | 0.9612 | 0.9609 | 0.9651 | 0.9610 | 0.9609 | |
| TransformerConv | 0.9235 | 0.9338 | 0.9239 | 0.9228 | 0.9341 | 0.9235 | 0.9228 | |
| BPI12-O | T-GCNConv | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 |
| O-GCNConv | O-GraphConv | T-GCNConv | T-GraphConv | ||||||||||
| Class | Precision | Recall | F1 | Precision | Recall | F1 | Precision | Recall | F1 | Precision | Recall | F1 | Support |
| 0 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 92 |
| 1 | 0.7838 | 1.0000 | 0.8788 | 0.7953 | 0.9828 | 0.8792 | 0.7838 | 1.0000 | 0.8788 | 0.7838 | 1.0000 | 0.8788 | 174 |
| 2 | 1.0000 | 1.0000 | 1.0000 | 0.625 | 1.0000 | 0.7692 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 5 |
| 3 | 0.9545 | 1.0000 | 0.9767 | 1.0000 | 0.9048 | 0.95 | 1.0000 | 0.9048 | 0.95 | 0.9524 | 0.9524 | 0.9524 | 21 |
| 4 | 0.8649 | 1.0000 | 0.9275 | 0.8611 | 0.9688 | 0.9118 | 0.9143 | 1.0000 | 0.9552 | 0.8889 | 1.0000 | 0.9412 | 32 |
| Dataset | Model | Prefix (%) | Accuracy | Macro Average | Weighted Average | ||||
| Precision | Recall | F1 | Precision | Recall | F1 | ||||
| BPI12-A | T-GCNConv | 30 | 0.7114 | 0.7121 | 0.7110 | 0.7094 | 0.7122 | 0.7114 | 0.7096 |
| 50 | 0.7286 | 0.7700 | 0.7279 | 0.7211 | 0.7704 | 0.7286 | 0.7217 | ||
| 70 | 0.8396 | 0.8513 | 0.8390 | 0.8367 | 0.8511 | 0.8396 | 0.8369 | ||
| 100 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | ||
| Cochran’s , ( ) | |||||||||
| Dataset | Model | T-GCNConv | T-GraphConv | ||||||||||||
| Acc | Macro Average | Weighted Average | Acc | Macro Average | Weighted Average | ||||||||||
| P | R | F1 | P | R | F1 | P | R | F1 | P | R | F1 | ||||
| BPI12-A | Full | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 |
| 0.9625 | 0.9661 | 0.9627 | 0.9624 | 0.9663 | 0.9625 | 0.9624 | 0.9633 | 0.9654 | 0.9634 | 0.9632 | 0.9656 | 0.9633 | 0.9632 | ||
| 0.9610 | 0.9649 | 0.9612 | 0.9609 | 0.9651 | 0.9610 | 0.9609 | 0.9610 | 0.9649 | 0.9612 | 0.9609 | 0.9651 | 0.9610 | 0.9609 | ||
| BPI12-O | Full | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 | 1.0000 |
Temporal GNNs
Temporal graph neural networks model how node relationships, graph structure, and features change over time to support prediction on evolving networks.
See all 83 papers