activity
20202026
most citedOn Feature Collapse and Deep Kernel Learning for Single Forward Pass Uncertainty

56 citations · 85 across the 10 of their papers we have counts for

collaborators

10 papers

cs.LG2026

Advancing Open and Reproducible Relational Learning: RelArena-, TabPFN-Rel and RPI

Adrian Hayler, Klemens Flöge, Alan Arazi +44

This first release of Prior Labs in relational learning shows our continued commitment to open science. We open-source three pieces of software that we expect to accelerate researc…

cs.LG2025★ 3 cited

TabPFN-2.5: Advancing the State of the Art in Tabular Foundation Models

Léo Grinsztajn, Klemens Flöge, Oscar Key +23

The first tabular foundation model, TabPFN, and its successor TabPFNv2 have impacted tabular AI substantially, with dozens of methods building on it and hundreds of applications ac…

cs.LG2023★ 8 cited

No Train No Gain: Revisiting Efficient Training Algorithms For Transformer-based Language Models

Jean Kaddour, Oscar Key, Piotr Nawrot +2

The computation necessary for training Transformer-based language models has skyrocketed in recent years. This trend has motivated research on efficient training algorithms designe…

stat.ME2023★ 1 cited

Optimally-Weighted Estimators of the Maximum Mean Discrepancy for Likelihood-Free Inference

Ayush Bharti, Masha Naslidnyk, Oscar Key +2

Likelihood-free inference methods typically make use of a distance between simulated and real data. A common example is the maximum mean discrepancy (MMD), which has previously bee…

stat.ML2022

Towards Healing the Blindness of Score Matching

Mingtian Zhang, Oscar Key, Peter Hayes +3

Score-based divergences have been widely used in machine learning and statistics applications. Despite their empirical success, a blindness problem has been observed when using the…

stat.ML2021★ 2 cited

Composite Goodness-of-fit Tests with Kernels

Oscar Key, Arthur Gretton, François-Xavier Briol +1

Model misspecification can create significant challenges for the implementation of probabilistic models, and this has led to development of a range of robust methods which directly…

cs.LG2021★ 8 cited

Generating Interpretable Counterfactual Explanations By Implicit Minimisation of Epistemic and Aleatoric Uncertainties

Lisa Schut, Oscar Key, Rory McGrath +4

Counterfactual explanations (CEs) are a practical tool for demonstrating why machine learning classifiers make particular decisions. For CEs to be useful, it is important that they…

cs.LG2021★ 56 cited

On Feature Collapse and Deep Kernel Learning for Single Forward Pass Uncertainty

Joost van Amersfoort, Lewis Smith, Andrew Jesson +2

Inducing point Gaussian process approximations are often considered a gold standard in uncertainty estimation since they retain many of the properties of the exact GP and scale to…

stat.ML2020★ 2 cited

On Signal-to-Noise Ratio Issues in Variational Inference for Deep Gaussian Processes

Tim G. J. Rudner, Oscar Key, Yarin Gal +1

We show that the gradient estimates used in training Deep Gaussian Processes (DGPs) with importance-weighted variational inference are susceptible to signal-to-noise ratio (SNR) is…

cs.LG2020★ 5 cited

Interlocking Backpropagation: Improving depthwise model-parallelism

Aidan N. Gomez, Oscar Key, Kuba Perlin +4

The number of parameters in state of the art neural networks has drastically increased in recent years. This surge of interest in large scale neural networks has motivated the deve…