Aniket Rege

WAIV Lab

PhD Student, University

of Wisconsin-Madison

I’m a Ph.D student in Computer Sciences at the University of Wisconsin-Madison working on preference learning, computer vision, and generative models. I am fortunate to be jointly advised by Ramya Vinayak and Yong Jae Lee.

Previously, I was an masters student at the University of Washington, Seattle where I worked with the RAIVN Lab advised by Prof. Ali Farhadi and mentored by Aditya Kusupati. My MS research focused on large-scale efficient and deployable machine learning, visual representation learning, and web-scale search.

In Summer ‘22 I interned in the deep learning for autonomous vehicles team at NVIDIA Redmond on Multiview LidarNet, supervised by Dr. Nikolai Smolyanskiy and Dr. Stan Birchfield. Previously, I worked for several years as a research engineer at Samsung Research India, working on perception problems for the AI Camera, including as a founding member of Single Take Camera, the USP of the Galaxy S20 flagship smartphone.

I am also generally interested in contrastive learning, learning from minimal supervision, and model robustness to real-world distribution shifts.

Reach me via aniketr[at]cs[dot]wisc[dot]edu

news

Jun 18, 2024	PAL is accepted at ICML workshops: TF2M and MHFAIA (oral)!
Feb 7, 2024	I published a beginner-friendly blog about MRL following OpenAI’s training their new Matryoshka embedding models!
Aug 21, 2023	AdANNS is accepted at NeurIPS 2023! A previous version was also accepted at ICLR ‘23 PML4DC Workshop.
May 31, 2023	I successfully defended my masters thesis!
Feb 7, 2023	Our contribution to Google’s large-scale efficient deep learning published as a Google Blog.

selected publications

AdANNS: A Framework for Adaptive Semantic Search

Aniket Rege*, Aditya Kusupati*, Sharan Ranjit S, Alan Fan, Qingqing Cao, Sham Kakade, Prateek Jain, and Ali Farhadi

In Advances in Neural Information Processing Systems (NeurIPS), 2023

arXiv Bib Code

@inproceedings{rege2023adanns,
  title = {AdANNS: A Framework for Adaptive Semantic Search},
  author = {Rege*, Aniket and Kusupati*, Aditya and Ranjit S, Sharan and Fan, Alan and Cao, Qingqing and Kakade, Sham and Jain, Prateek and Farhadi, Ali},
  booktitle = {Advances in Neural Information Processing Systems (NeurIPS)},
  year = {2023},
}

Matryoshka Representation Learning

Aniket Rege*, Aditya Kusupati*, Gantavya Bhatt*, Matthew Wallingford, Aditya Sinha, Vivek Ramanujan, William Howard-Snyder, Kaifeng Chen, Sham Kakade, Prateek Jain, and Ali Farhadi

In Advances in Neural Information Processing Systems (NeurIPS), 2022

arXiv Bib Blog Code

@inproceedings{kusupati2022matryoshka,
  title = {Matryoshka Representation Learning},
  author = {Rege*, Aniket and Kusupati*, Aditya and Bhatt*, Gantavya and Wallingford, Matthew and Sinha, Aditya and Ramanujan, Vivek and Howard-Snyder, William and Chen, Kaifeng and Kakade, Sham and Jain, Prateek and Farhadi, Ali},
  booktitle = {Advances in Neural Information Processing Systems (NeurIPS)},
  year = {2022},
}