Publications

Full list of papers including preprints: google scholar

Desai, Aditya, Gaurav Gupta, Tianyi Zhang, and Anshumali Shrivastava. “IDentity with Locality: An ideal hash for gene sequence search.” arXiv preprint arXiv:2406.14901 (2024). [To Appear in KDD 2025]

Desai, Aditya * , Kimia Saedi, Apoorv Walia, Jihyeong Lee, Keren Zhou, and Anshumali Shrivastava. “SS1: Accelerating Inference with Fast and Expressive Sketch Structured Transform.” In The Thirty-eighth Annual Conference on Neural Information Processing Systems.

A. Desai and A. Shrivastava, “In defense of parameter sharing for model compression,” in The Twelfth International Conference on Learning Representations, 2024.

A. Desai, K. Zhou, and A. Shrivastava, “Hardware-Aware Compression with Random Operation Access Specific Tile (ROAST) Hashing,” in International Conference on Machine Learning, PMLR, 2023, pp. 7732–7749.

Z. Liu, A. Desai, F. Liao, et al., “Scissorhands: Exploiting the Persistence of Importance Hypothesis for LLM KV Cache Compression at Test Time,” in Advances in Neural Information Processing Systems, 2023.

A. Desai, L. Chou, and A. Shrivastava, “Random Offset Block Embedding (ROBE) for compressed embedding tables in deep learning recommendation systems,” in Proceedings of Machine Learning and Systems, D. Marculescu, Y. Chi, and C. Wu, Eds., Outstanding paper award , vol. 4, 2022, pp. 762–778.

A. Desai and A. Shrivastava, “The trade-offs of model size in large recommendation models: 100GB to 10MB Criteo-tb DLRM model,” in Advances in Neural Information Processing Systems, S. Koyejo, S. Mohamed, A. Agarwal, D. Belgrave, K. Cho, and A. Oh, Eds., vol. 35, Curran Associates, Inc., 2022, pp. 33 961–33 972.  url: https: //proceedings.neurips.cc/paper_files/paper/2022/file/dbae915128892556134f1c5375855590- Paper-Conference.pdf.

Z. Dai, A. Desai, R. Heckel, and A. Shrivastava, “Active sampling count sketch (ASCS) for online sparse estimation of a trillion scale covariance matrix,” in Proceedings of the 2021 International Conference on Management of Data, 2021, pp. 352–364.

A. Desai, Z. Xu, M. Gupta, A. Chandran, A. Vial-Aussavy, and A. Shrivastava, “Raw Nav-merge Seismic Data to Subsurface Properties with MLP based Multi-Modal Information Unscrambler,” in Advances in Neural Information Processing Systems, M. Ranzato, A. Beygelzimer, Y. Dauphin, P. Liang, and J. W. Vaughan, Eds., vol. 34, Curran Associates, Inc., 2021, pp. 8740–8752.  url: https: //proceedings.neurips.cc/paper_files/paper/2021/file/498f2c21688f6451d9f5fd09d53edda7- Paper.pdf.

A. Desai, S. Gulwani, V. Hingorani, et al., “Program synthesis using natural language,” in Proceedings of the 38th International Conference on Software Engineering, 2016, pp. 345–356.

A. Desai, E. Jain, and S. Roy, “Facilitating Verification in Program Loops by Identification of Static Iteration Patterns,” in 2013 20th Asia-Pacific Software Engineering Conference (APSEC), IEEE, vol. 1, 2013, pp. 83–90.