← Search

Ryan Kang

4 accepted papers

2024

Adaptive Sampling for Efficient Softmax Approximation

NeurIPS 2024poster

The softmax function is ubiquitous in machine learning and optimization applications. Computing the full softmax evaluation of a matrix-vector product can be computationally expensive in high-dimensional settings. In many applications, however, it is sufficient to calculate only the top few outputs…

Cited by 0SourcePDFScholar
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{…

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-…