Aniket Rege

Profile

I'm a PhD student in Computer Science at the University of Wisconsin-Madison working on machine learning, specifically in the areas of efficient retrieval, value alignment, and pluralistic generative models. During my masters at the University of Washington, I worked on making learned representations and search indexing structures more flexible and efficient through approaches like Matryoshka Representation Learning, which allows a single embedding to adapt to different computational constraints. During my PhD, I've been exploring pluralistic approaches to aligning generative models to human values, developing frameworks that can capture diverse human preferences rather than assuming universal values.

I'm particularly interested in making AI systems that are both computationally efficient and aligned with human values. My work on adaptive representations has shown promising results for tasks like large-scale retrieval and few-shot learning, while maintaining robustness. On the alignment side, I'm working on models that can learn reward functions that reflect the plurality of human preferences while remaining efficient to train and deploy.

Publications

PAL: Pluralistic Alignment Framework for Learning from Heterogeneous Preferences

PAL: Pluralistic Alignment Framework for Learning from Heterogeneous Preferences

Daiwei Chen, Yi Chen, Aniket Rege, Ramya Korlakai Vinayak

arXiv.org 2024

AdANNS: A Framework for Adaptive Semantic Search

AdANNS: A Framework for Adaptive Semantic Search

Aniket Rege, Aditya Kusupati, S. SharanRanjit, Alan Fan, Qingqing Cao, S. Kakade, Prateek Jain, Ali Farhadi

Neural Information Processing Systems 2023

Matryoshka Representation Learning

Matryoshka Representation Learning

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

Neural Information Processing Systems 2022

Spatio-Temporal Video Representation Learning for AI Based Video Playback Style Prediction

Spatio-Temporal Video Representation Learning for AI Based Video Playback Style Prediction

Rishubh Parihar, Gaurav Ramola, Ranajit Saha, Raviprasad Kini, Aniket Rege, S. Velusamy

arXiv.org 2021

FabSoften: Face Beautification via Dynamic Skin Smoothing, Guided Feathering, and Texture Restoration

S. Velusamy, Rishubh Parihar, Raviprasad Kini, Aniket Rege

2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops (CVPRW) 2020

QUIC Protocol Performance in Wireless Networks

QUIC Protocol Performance in Wireless Networks

P. Kharat, Aniket Rege, A. Goel, M. Kulkarni

International Conference on Cryptography, Security and Privacy 2018

Final NeurIPS 2022 Conference Paper 779 Reviewer DT 9 D Comment

Gantavya Bhatt, Aniket Rege, Aditya Sinha, Vivek Ramanujan, William Howard-Snyder, Kaifeng Chen, Prateek Jain, Ali Farhadi

Matryoshka Representations for Adaptive Deployment

Matryoshka Representations for Adaptive Deployment

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

arXiv.org 2022

FabSoften: Supplementary Material

FabSoften: Supplementary Material

S. Velusamy, Rishubh Parihar, Raviprasad Kini, Aniket Rege