Publications

2022

• Robust Training in High Dimensions via Block Coordinate Geometric Median Descent (arXiv, slides, poster, code)
  A. Acharya, A. Hashemi, P. Jain, S. Sanghavi, I. Dhillon, U. Topcu.
  In International Conference on Artificial Intelligence and Statistics (AISTATS), March 2022.

2019

• Provable Non-linear Inductive Matrix Completion
  K. Zhong, Z. Song, P. Jain, I. Dhillon.
  To appear in Neural Information Processing Systems (NIPS), 2019.

2017

• Recovery Guarantees for One-hidden-layer Neural Networks (pdf, arXiv)
  K. Zhong, Z. Song, P. Jain, P. Bartlett, I. Dhillon.
  In International Conference on Machine Learning (ICML), pp. 4140-4149, August 2017. (Oral)
• Partial Hard Thresholding (pdf)
  P. Jain, A. Tewari, I. Dhillon.
  IEEE Transactions on Information Theory 63(5): 3029-3038, 2017.

2016

• Mixed Linear Regression with Multiple Components (pdf)
  K. Zhong, P. Jain, I. Dhillon.
  In Neural Information Processing Systems (NIPS), December 2016.
• Structured Sparse Regression via Greedy Hard Thresholding (pdf, arXiv)
  P. Jain, N. Rao, I. Dhillon.
  In Neural Information Processing Systems (NIPS), pp. 1516-1524, December 2016.

2015

• Efficient Matrix Sensing Using Rank-1 Gaussian Measurements (pdf)
  K. Zhong, P. Jain, I. Dhillon.
  In International Conference on Algorithmic Learning Theory (ALT), pp. 3-18, October 2015.

2014

• Large-scale Multi-label Learning with Missing Labels (pdf, software)
  H. Yu, P. Jain, P. Kar, I. Dhillon.
  In International Conference on Machine Learning (ICML), pp. 593–601, June 2014.

2012

• Metric and Kernel Learning using a Linear Transformation (pdf)
  P. Jain, B. Kulis, J. Davis, I. Dhillon.
  Journal of Machine Learning Research (JMLR) 13, pp. 519-547, March 2012.

2011

• Orthogonal Matching Pursuit with Replacement (pdf)
  P. Jain, A. Tewari, I. Dhillon.
  In Neural Information Processing Systems (NIPS), December 2011.

2010

• Guaranteed Rank Minimization via Singular Value Projection (pdf)
  R. Meka, P. Jain, I. Dhillon.
  In Neural Information Processing Systems (NIPS), pp. 937-945, December 2010.
• Inductive Regularized Learning of Kernel Functions (pdf)
  P. Jain, B. Kulis, I. Dhillon.
  In Neural Information Processing Systems (NIPS), pp. 946-954, December 2010.

2009

• Matrix Completion from Power-Law Distributed Samples (pdf)
  R. Meka, P. Jain, I. Dhillon.
  In Neural Information Processing Systems (NIPS), pp. 1258-1266, December 2009.
• Geometry-aware Metric Learning (pdf)
  Z. Lu, P. Jain, I. Dhillon.
  In International Conference on Machine Learning (ICML), June 2009.

2008

• Online Metric Learning and Fast Similarity Search (pdf)
  P. Jain, B. Kulis, I. Dhillon, K. Grauman.
  In Neural Information Processing Systems (NIPS), December 2008.
• Simultaneous Unsupervised Learning of Disparate Clusterings (pdf)
  P. Jain, R. Meka, I. Dhillon.
  Statistical Analysis and Data Mining 1(3), pp. 195-210, November 2008.
• Rank Minimization via Online Learning (pdf)
  R. Meka, P. Jain, C. Caramanis, I. Dhillon.
  In International Conference on Machine Learning (ICML), July 2008.
• Simultaneous Unsupervised Learning of Disparate Clusterings (pdf)
  P. Jain, R. Meka, I. Dhillon.
  In IEEE International Conference on Data Mining (ICDM), pp. 858-869, April 2008.

2007

• Information-Theoretic Metric Learning (pdf)
  J. Davis, B. Kulis, P. Jain, S. Sra, I. Dhillon.
  In International Conference on Machine Learning (ICML), June 2007.
• Online Linear Regression using Burg Entropy (pdf)
  P. Jain, B. Kulis, I. Dhillon.
  University of Texas Computer Science Technical Report (UTCS Technical Report) TR-07-08, February 2007.