← Search

Ankur Mallick

6 accepted papers

2025

BEST-Route: Adaptive LLM Routing with Test-Time Optimal Compute

ICML 2025poster

Large language models (LLMs) are powerful tools but are often expensive to deploy at scale. LLM query routing mitigates this by dynamically assigning queries to models of varying cost and quality to obtain a desired tradeoff. Prior query routing approaches generate only one response from the select…

Cited by 0SourcePDFScholar
2024

Hybrid LLM: Cost-Efficient and Quality-Aware Query Routing

ICLR 2024poster

Large language models (LLMs) excel in most NLP tasks but also require expensive cloud servers for deployment due to their size, while smaller models that can be deployed on lower cost (e.g., edge) devices, tend to lag behind in terms of response quality. Therefore in this work we propose a hybrid in…

2021

Deep kernels with probabilistic embeddings for small-data learning

UAI 2021poster

Gaussian Processes (GPs) are known to provide accurate predictions and uncertainty estimates even with small amounts of labeled data by capturing similarity between data points through their kernel function. However traditional GP kernels are not very effective at capturing similarity between high d…

2021

Leveraging Spatial and Temporal Correlations in Sparsified Mean Estimation

NeurIPS 2021poster

We study the problem of estimating at a central server the mean of a set of vectors distributed across several nodes (one vector per node). When the vectors are high-dimensional, the communication cost of sending entire vectors may be prohibitive, and it may be imperative for them to use sparsificat…

Cited by 18SourcePDFScholar
2019

Fast and Efficient Distributed Matrix-vector Multiplication Using Rateless Fountain Codes

ICASSP 2019accepted

We propose a rateless fountain coding strategy to alleviate the problem of straggling nodes - computing nodes that unpredictably slowdown or fail - in distributed matrix-vector multiplication. Our algorithm generates linear combinations of the m rows of the matrix, and assigns them to different work…

Cited by 0SourceScholar
2016

Bandlimited field reconstruction from samples obtained on a discrete grid with unknown random locations

ICASSP 2016accepted

Sampling spatial fields using sensors which are location unaware is an exciting topic. Due to symmetry and shift-invariance of bandlimited fields, it is known that uniformly distributed location-unaware sensors cannot infer the field. This work studies asymmetric (nonuniform) distributions on locati…

Cited by 0SourceScholar