Benjamin Coleman

Researcher, Google DeepMind

Joined

May 2021

Names

Benjamin Coleman (Preferred)

Emails

****@rice.edu (Confirmed)

****@google.com (Confirmed)

****@gmail.com (Confirmed)

Personal Links

Career & Education History

Researcher

Google DeepMind (google.com)

2022 – Present

PhD student

Rice University (rice.edu)

2018 – 2022

Undergrad student

Florida Atlantic University (fau.edu)

2012 – 2017

Advisors, Relations & Conflicts

PhD Advisor

Anshumali Shrivastava

2018 – 2022

Expertise

Recommendation Systems

Present

Sketching and Hashing Algorithms

Present

Publications

PACEvolve: Enabling Long-Horizon Progress-Aware Consistent Evolution
Minghao Yan, Bo Peng, Benjamin Coleman, Ziqi Chen, Zhouhang Xie, Shuo Chen, Zhankui He, Noveen Sachdeva, Isabella Ye, Weili Wang, Chi Wang, Ed H. Chi, Fernando Pereira, Wang-Cheng Kang, Derek Zhiyuan Cheng, Beidou Wang
- LLA 2026 Poster
- Readers: Everyone
How to train data-efficient LLMs
Noveen Sachdeva, Benjamin Coleman, Wang-Cheng Kang, Jianmo Ni, Lichan Hong, Ed H. Chi, James Caverlee, Julian McAuley, Derek Zhiyuan Cheng
- ICLR 2026 Poster
- Readers: Everyone
Down with the Hierarchy: The ‘H’ in HNSW Stands for “Hubs”
Blaise Munyampirwa, Vihan Lakshman, Benjamin Coleman
- VecDB 2025
- Readers: Everyone
DESSERT: An Efficient Algorithm for Vector Set Search with Vector Set Queries
Joshua Engels, Benjamin Coleman, Vihan Lakshman, Anshumali Shrivastava
- NeurIPS 2023 poster
- Readers: Everyone
Unified Embedding: Battle-Tested Feature Representations for Web-Scale ML Systems
Benjamin Coleman, Wang-Cheng Kang, Matthew Fahrbach, Ruoxi Wang, Lichan Hong, Ed H. Chi, Derek Zhiyuan Cheng
- NeurIPS 2023 spotlight
- Readers: Everyone
One-Pass Distribution Sketch for Measuring Data Heterogeneity in Federated Learning
Zichang Liu, Zhaozhuo Xu, Benjamin Coleman, Anshumali Shrivastava
- NeurIPS 2023 poster
- Readers: Everyone
BOLT: An Automated Deep Learning Framework for Training and Deploying Large-Scale Neural Networks on Commodity CPU Hardware
Nicholas Meisburger, Vihan Lakshman, Benito Geordie, Joshua Engels, David Torres Ramos, Pratik Pranav, Benjamin Coleman, Benjamin Meisburger, Shubh Gupta, Yashwanth Adunukota, Tharun Medini, Anshumali Shrivastava
- 2023 (modified: 25 Apr 2023)
- CoRR 2023
- Readers: Everyone
Density Sketches for Sampling and Estimation
Aditya Desai, Benjamin Coleman, Anshumali Shrivastava
- Published: 01 Feb 2023, Last Modified: 13 Feb 2023
- Submitted to ICLR 2023
- Readers: Everyone
Retaining Knowledge for Learning with Dynamic Definition
Zichang Liu, Benjamin Coleman, Tianyi Zhang, Anshumali Shrivastava
- Published: 31 Oct 2022, Last Modified: 16 Jan 2023
- NeurIPS 2022 Accept
- Readers: Everyone
Graph Reordering for Cache-Efficient Near Neighbor Search
Benjamin Coleman, Santiago Segarra, Alex Smola, Anshumali Shrivastava
- Published: 31 Oct 2022, Last Modified: 13 Oct 2022
- NeurIPS 2022 Accept
- Readers: Everyone

View all 32 publications

Co-Authors

View all 54 co-authors