← Search

Jake Fawkes

7 accepted papers

2026

$f$-Trajectory Balance: A Loss Family for Tuning GFlowNets, Generative Models, and LLMs with Off- and On-Policy Data

ICML 2026poster

In GFlowNets and variational inference, it has been shown that the mean square error between target and model log probabilities is an effective, low variance, surrogate loss for training generative models. This loss has the property that when evaluated \emph{on-policy} its gradients correspond to th…

Cited by 0SourceScholar
2025

Is Merging Worth It? Securely Evaluating the Information Gain for Causal Dataset Acquisition

AISTATS 2025poster

Merging datasets across institutions is a lengthy and costly procedure, especially when it involves private information. Data hosts may therefore want to prospectively gauge which datasets are most beneficial to merge with, without revealing sensitive information. For causal estimation this is part…

Cited by 0SourcecodeScholar
2025

The Hardness of Validating Observational Studies with Experimental Data

AISTATS 2025poster

Observational data is often readily available in large quantities, but can lead to biased causal effect estimates due to the presence of unobserved confounding. Recent works attempt to remove this bias by supplementing observational data with experimental data, which, when available, is typically on…

Cited by 0SourcecodeScholar
2024

The Fragility of Fairness: Causal Sensitivity Analysis for Fair Machine Learning

NeurIPS 2024poster

Fairness metrics are a core tool in the fair machine learning literature (FairML), used to determine that ML models are, in some sense, “fair.” Real-world data, however, are typically plagued by various measurement biases and other violated assumptions, which can render fairness assessments meaningl…

2023

Returning The Favour: When Regression Benefits From Probabilistic Causal Knowledge

ICML 2023oral

A directed acyclic graph (DAG) provides valuable prior knowledge that is often discarded in regression tasks in machine learning. We show that the independences arising from the presence of collider structures in DAGs provide meaningful inductive biases, which constrain the regression hypothesis spa…