← Search

Sebastian Thrun

9 accepted papers

2025

Restructuring Vector Quantization with the Rotation Trick

ICLR 2025oral

Vector Quantized Variational AutoEncoders (VQ-VAEs) are designed to compress a continuous input to a discrete latent space and reconstruct it with minimal distortion. They operate by maintaining a set of vectors---often referred to as the codebook---and quantizing each encoder output to the nearest…

Cited by 3SourcePDFScholar
2024

Context-Aware Meta-Learning

ICLR 2024poster

Large Language Models like ChatGPT demonstrate a remarkable capacity to learn new concepts during inference without any fine-tuning. However, visual models trained to detect new objects during inference have been unable to replicate this ability, and instead either perform poorly or require meta-tra…

2024

Faster Maximum Inner Product Search in High Dimensions

ICML 2024poster

Maximum Inner Product Search (MIPS) is a ubiquitous task in machine learning applications. Given a query vector and $n$ other vectors in $d$ dimensions, the MIPS problem is to find the atom that has the highest inner product with the query vector. Existing MIPS algorithms scale at least as $O(\sqrt{…

2024

MAPTree: Beating “Optimal” Decision Trees with Bayesian Decision Trees

AAAI 2024technical

Decision trees remain one of the most popular machine learning models today, largely due to their out-of-the-box performance and interpretability. In this work, we present a Bayesian approach to decision tree induction via maximum a posteriori inference of a posterior distribution over trees. We fir…

2023

BanditPAM++: Faster $k$-medoids Clustering

NeurIPS 2023poster

Clustering is a fundamental task in data science with wide-ranging applications. In $k$-medoids clustering, cluster centers must be actual datapoints and arbitrary distance metrics may be used; these features allow for greater interpretability of the cluster centers and the clustering of exotic obje…

2022

MABSplit: Faster Forest Training Using Multi-Armed Bandits

NeurIPS 2022accept

Random forests are some of the most widely used machine learning models today, especially in domains that necessitate interpretability. We present an algorithm that accelerates the training of random forests and other popular tree-based learning methods. At the core of our algorithm is a novel node-…

2020

BanditPAM: Almost Linear Time k-Medoids Clustering via Multi-Armed Bandits

NeurIPS 2020poster

Clustering is a ubiquitous task in data science. Compared to the commonly used k-means clustering, k-medoids clustering requires the cluster centers to be actual data points and supports arbitrary distance metrics, which permits greater interpretability and the clustering of structured objects. Curr…

2016

A Probabilistic Framework for Real-time 3D Segmentation using Spatial, Temporal, and Semantic Cues

RSS 2016poster

In order to track dynamic objects in a robot’s environment, one must first segment the scene into a collection of separate objects. Most real-time robotic vision systems today rely on simple spatial relations to segment the scene into separate objects. However, such methods fail under a variety of…

Cited by 54SourcePDFScholar