← Search

Peter Bailis

9 accepted papers

2024

Are More LLM Calls All You Need? Towards the Scaling Properties of Compound AI Systems

NeurIPS 2024poster

Many recent state-of-the-art results in language tasks were achieved using compound systems that perform multiple Language Model (LM) calls and aggregate their responses. However, there is little understanding of how the number of LM calls -- e.g., when asking the LM to answer each question multiple…

Cited by 13SourcePDFScholar
2024

Break the Sequential Dependency of LLM Inference Using Lookahead Decoding

ICML 2024poster

Autoregressive decoding of large language models (LLMs) is memory bandwidth bounded, resulting in high latency and significant wastes of the parallel processing power of modern accelerators. Existing methods for accelerating LLM decoding often require a draft model (e.g., speculative decoding), whic…

2024

Online Speculative Decoding

ICML 2024poster

Speculative decoding is a pivotal technique to accelerate the inference of large language models (LLMs) by employing a smaller draft model to predict the target model's outputs. However, its efficacy can be limited due to the low predictive accuracy of the draft model, particularly when faced with d…

2022

Similarity Search for Efficient Active Learning and Search of Rare Concepts

AAAI 2022technical

Many active learning and search approaches are intractable for large-scale industrial settings with billions of unlabeled examples. Existing approaches search globally for the optimal examples to label, scaling linearly or even quadratically with the unlabeled data. In this paper, we improve the com…

Cited by 41SourcePDFScholar
2020

Selection via Proxy: Efficient Data Selection for Deep Learning

ICLR 2020poster

Data selection methods, such as active learning and core-set selection, are useful tools for machine learning on large datasets. However, they can be prohibitively expensive to apply in deep learning because they depend on feature representations that need to be learned. In this work, we show that w…

Cited by 391SourcecodeScholar
2019

Compressed Factorization: Fast and Accurate Low-Rank Factorization of Compressively-Sensed Data

ICML 2019oral

What learning algorithms can be run directly on compressively-sensed data? In this work, we consider the question of accurately and efficiently computing low-rank matrix or tensor factorizations given data compressed via random projections. We examine the approach of first performing factorization i…

Cited by 19SourcePDFScholar
2019

LIT: Learned Intermediate Representation Training for Model Compression

ICML 2019oral

Researchers have proposed a range of model compression techniques to reduce the computational and memory footprint of deep neural networks (DNNs). In this work, we introduce Learned Intermediate representation Training (LIT), a novel model compression technique that outperforms a range of recent mod…

Cited by 80SourcePDFScholar
2019

Rehashing Kernel Evaluation in High Dimensions

ICML 2019oral

Kernel methods are effective but do not scale well to large scale data, especially in high dimensions where the geometric data structures used to accelerate kernel evaluation suffer from the curse of dimensionality. Recent theoretical advances have proposed fast kernel evaluation algorithms leveragi…

Cited by 48SourcePDFScholar