← Search

Soheila Molaei

7 accepted papers

2026

Neuro-Symbolic Federated Learning over Heterogeneous Data-Views: A Structured Approach to Distributive EHR Modelling

AAAI 2026technical

Federated learning (FL) enables privacy-preserving model training across distributed Electronic Health Records (EHRs), but its deployment remains limited by data-view heterogeneity, where institutions maintain incompatible local schemas. Most existing methods address this by enforcing flat, aligned

Cited by 0SourcePDFScholar
2025

Information Transfer Across Clinical Tasks via Adaptive Parameter Optimisation

AISTATS 2025oral

This paper presents Adaptive Parameter Optimisation (APO), a novel framework for optimising shared models across multiple clinical tasks, addressing the challenges of balancing strict parameter sharing—often leading to task conflicts—and soft parameter sharing, which may limit effective cross-task i…

Cited by 0SourceScholar
2025

Optimising Clinical Federated Learning through Mode Connectivity-based Model Aggregation

AISTATS 2025poster

Federated Learning (FL) involves a server aggregating local models from clients to compute a global model. However, this process can struggle to position the global model in low-loss regions of the parameter space for all clients, resulting in subpar convergence and inequitable performance across cl…

Cited by 0SourceScholar
2024

Dynamic Inter-treatment Information Sharing for Individualized Treatment Effects Estimation

AISTATS 2024poster

Estimation of individualized treatment effects (ITE) from observational studies is a fundamental problem in causal inference and holds significant importance across domains, including healthcare. However, limited observational datasets pose challenges in reliable ITE estimation as data have to be sp…

2024

Federated Learning For Heterogeneous Electronic Health Records Utilising Augmented Temporal Graph Attention Networks

AISTATS 2024poster

The proliferation of decentralised electronic healthcare records (EHRs) across medical institutions requires innovative federated learning strategies for collaborative data analysis and global model training, prioritising data privacy. A prevalent issue during decentralised model training is the dat…

Cited by 9SourcePDFScholar
2023

Adversarial De-confounding in Individualised Treatment Effects Estimation

AISTATS 2023poster

Observational studies have recently received significant attention from the machine learning community due to the increasingly available non-experimental observational data and the limitations of the experimental studies, such as considerable cost, impracticality, small and less representative sampl…

2023

GNNEvaluator: Evaluating GNN Performance On Unseen Graphs Without Labels

NeurIPS 2023poster

Evaluating the performance of graph neural networks (GNNs) is an essential task for practical GNN model deployment and serving, as deployed GNNs face significant performance uncertainty when inferring on unseen and unlabeled test graphs, due to mismatched training-test graph distributions. In this p…

Cited by 15SourcePDFScholar