You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: _pages/research.md
+20-21Lines changed: 20 additions & 21 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -10,7 +10,26 @@ nav_order: 1
10
10
I am a PhD candidate in Biomedical AI at the University of Edinburgh, working with [Dr Sohan Seth](https://sohanseth.github.io/) at the [Data Science Unit](https://web.inf.ed.ac.uk/data-science-unit). My research sits at the intersection of **Bayesian modelling**, **unsupervised and causal machine learning**, and **healthcare**. I am particularly interested in longitudinal data, the health trajectories that unfold over years, where the goal is not just prediction but understanding. I want to identify meaningful subgroups of patients, the factors that drive their trajectories, and how interventions move them along.
11
11
{: .text-justify}
12
12
13
-
Three threads run through my work to date.
13
+
Four threads run through my work to date.
14
+
15
+
## Scalable, interpretable and robust Bayesian semi-supervised clustering
16
+
17
+
Real-world tabular and image data are often high-dimensional, only partially labelled, and noisy in unpredictable ways. The most common Bayesian clustering methods either do not scale or treat every pairwise constraint as ground truth. My ICML 2026 paper introduces **BASIL**, a scalable Bayesian semi-supervised clustering framework that combines a Hidden Markov Random Field formulation with stochastic variational inference, while jointly learning cluster-specific feature relevance and adaptively down-weighting unreliable supervision. BASIL trains efficiently on settings ranging from MNIST digit benchmarks to ChestMNIST medical imaging and a 501k-patient UK CPRD multimorbidity cohort, where metric-learning baselines fail to converge in over two days. It remains robust under up to 30% noisy constraints.
18
+
{: .text-justify}
19
+
20
+
-**Paper.** Scalable Bayesian Semi-supervised Clustering with Feature Selection and Adaptive Constraint Weighting, ICML 2026. *To appear.*
21
+
-**Talks.** ICML 2026 (forthcoming).
22
+
23
+
## Supervised causal clustering for heterogeneous treatment effects
24
+
25
+
Clinical trials often report no average benefit when in fact a benefit exists for a specific subpopulation, and unsupervised clustering of patient covariates rarely surfaces those subpopulations cleanly. My recent preprint introduces **Bayesian Supervised Causal Clustering (BSCC)**, a framework that uses individual treatment effect as the outcome guiding the clustering process. BSCC recovers homogeneous subgroups whose members are similar both in their covariate profiles and in how they respond to treatment, giving clinicians and trialists subgroups that are operationalisable rather than merely statistical. I evaluated BSCC on simulated benchmarks and on real-world data from the third International Stroke Trial.
26
+
{: .text-justify}
27
+
28
+
This thread builds on earlier work I did as a research assistant on a Turing-funded project, where I evaluated supervised metric-based clustering for recovering subphenotypes of critically ill COVID-19 patients under convalescent plasma treatment. That project introduced a "FavorCP" outcome that improved odds-ratio testing across discovered subgroups, and motivated the move to a fully Bayesian, causal formulation in BSCC.
-**Earlier project.**[Supervised Clustering of Critically Ill Patients](/projects/supervisedClust/)
14
33
15
34
## Interpretable clustering of multi-faceted time series
16
35
@@ -23,33 +42,13 @@ Real-world time series rarely live in a single facet. Patient histories, like mo
23
42
-**Talks.** UAI 2025 poster, [UKAIRS 2025](https://www.ukairs.ac.uk/ukairs-programme/) oral, Joint CDT Conference on AI for Healthcare 2025 oral.
24
43
-**Background reading.**[The real-world data are multi-faceted](/blog/2025/Multi-facet/)
25
44
26
-
## Scalable Bayesian semi-supervised clustering for health data
27
-
28
-
Electronic health records are high-dimensional, partially labelled, and noisy, and the most common Bayesian clustering methods either do not scale to them or treat every pairwise constraint as ground truth. My ICML 2026 paper introduces **BASIL**, a scalable Bayesian semi-supervised clustering framework that combines a Hidden Markov Random Field formulation with stochastic variational inference, while jointly learning cluster-specific feature relevance and adaptively down-weighting unreliable supervision. On a 501k-patient UK CPRD cohort, BASIL trains in around an hour and recovers clinically interpretable multimorbidity subgroups, where metric-learning baselines fail to converge in over two days. It remains robust under up to 30% noisy constraints.
29
-
{: .text-justify}
30
-
31
-
-**Paper.** Scalable Bayesian Semi-supervised Clustering with Feature Selection and Adaptive Constraint Weighting, ICML 2026. *To appear.*
32
-
-**Talks.** ICML 2026 (forthcoming).
33
-
34
45
## Disease trajectories in multimorbidity
35
46
36
47
Multimorbidity, the co-existence of two or more chronic conditions, is increasingly the norm rather than the exception, and care models designed for a single disease break down for these patients. Most existing analyses are cross-sectional. Far fewer take the **temporal order** of disease onset seriously. My MSc dissertation, conducted with the Data Science Unit, used temporal clustering to surface meaningful patterns of how multiple conditions accumulate over time, and to link those patterns to outcomes such as mortality. This thread shapes the longitudinal modelling questions I continue to pursue in the PhD.
37
48
{: .text-justify}
38
49
39
50
-**Project page.**[Clustering Individual Trajectories of Multiple Long-Term Conditions](/projects/clustTrajMLTC/)
40
51
41
-
## Supervised clustering for heterogeneous treatment effects
42
-
43
-
Clinical trials often report no average benefit when in fact a benefit exists for a specific subpopulation, and unsupervised clustering of patient covariates rarely surfaces those subpopulations cleanly. My recent preprint introduces **Bayesian Supervised Causal Clustering (BSCC)**, a framework that uses individual treatment effect as the outcome guiding the clustering process. BSCC recovers homogeneous subgroups whose members are similar both in their covariate profiles and in how they respond to treatment, giving clinicians and trialists subgroups that are operationalisable rather than merely statistical. I evaluated BSCC on simulated benchmarks and on real-world data from the third International Stroke Trial.
44
-
{: .text-justify}
45
-
46
-
This thread builds on earlier work I did as a research assistant on a Turing-funded project, where I evaluated supervised metric-based clustering for recovering subphenotypes of critically ill COVID-19 patients under convalescent plasma treatment. That project introduced a "FavorCP" outcome that improved odds-ratio testing across discovered subgroups, and motivated the move to a fully Bayesian, causal formulation in BSCC.
-**Earlier project.**[Supervised Clustering of Critically Ill Patients](/projects/supervisedClust/)
51
-
-**Forthcoming companion note.**[Comparison on Common Meta-learners for HTE](/blog/)
52
-
53
52
## Where I'm heading
54
53
55
54
A common thread runs through these projects. I want to build **interpretable models under uncertainty**, models that a domain expert can act on while honestly representing what the data does and does not support. I'm increasingly drawn to causal machine learning as the natural next step, moving from asking *which patients look alike* to asking *which interventions change which patients' trajectories, and why*. I have also been pulled toward the inference side of large language models. A co-authored **ICML 2026** paper develops Adaptive Sequential Monte Carlo with cache-coherent resampling for training-free LLM test-time scaling, and I am interested in how Bayesian thinking can sharpen reasoning under compute budgets.
0 commit comments