My research is concerend with discrete optimization problems in machine learning. Most recently, I have been working on submodular functions and graph cuts.
I have also worked on theoretical aspects of clustering and density estimation, from the perspective of approximation as well as learning theory.
Below is a summary of projects, and links to code. I have also co-organized the NIPS workshop on Discrete Optimization in Machine Learning.
Structured problems with submodular cost
The generic setting here is as follows: take a combinatorial optimization problem, and replace the usual sum-of-weights cost by a nonlinear, submodular function. This makes the problem much harder but has nice applications. In particular, we worked on this setting with graph cuts, for which there are applications in inference and computer vision.
In addition, we derived online algorithms for structured decision problems (e.g., online spanning tree, or s-t cut) with submodular cost (in contrast, almost all previous results are for linear costs).
During our work, we came across several functions that are actually not submodular (but maybe some people hoped them to be), and thus I am collecting those functions in a bag of submodular non-examples.
Clustering & Graph Cuts
Nearest Neighbor Clustering: we developed a generic algorithm to minimize popular clustering objective functions, such as Normalized Cut or k-means. This method is consistent, thanks to pooling neighboring points. (with Ulrike von Luxburg, Sebastien Bubeck, Michael Kaufmann).
Download Demo Code
Generalized Clustering via Kernel Embeddings: The concept of clustering can be generalized to finding distributions that are maximally separated. The standard case is to measure separation by means of the distribution (k-means). MMD, the maximum mean discrepancy in a feature space, allows to also take higher-order moments into account. As a result, it is possible to e.g. separate two Gaussians with the same mean but different variance. Interestingly, maximizing MMD is closely related to kernel k-means and various other clustering criteria. (with Ulrike von Luxburg, Bernhard Schoelkopf, Arthur Gretton, Bharath Sriperumbudur)
Approximation Algorithms for Tensor Clustering: We prove an approximation factor for tensor clustering (i.e., "cutting a big cube into little cubes") for maybe the simplest possible algorithm. Various divergences are possible, such as Euclidean distance, Bregman divergences etc. (with Suvrit Sra, Arindam Banerjee)
Solution Stability in Linear Programming Relaxations: Graph Partitioning and Unsupervised Learning. Data is often noisy. If there are several good solutions to an optimization problem (here, graph partitioning), then the noise can be the determining part to make one solution optimal. However, how much do we want to trust this solution, if a small perturbation in the data suddenly makes another solution the best? this work proposes a method to test the stability of an optimal graph cut solution to perturbation of the edge weights (i.e., noise).We also show that several common clustering and graph partitioning objectives fall in our framework. (with Sebastian Nowozin)
Fast Kernel ICA: We developed a Newton-like method for kernel ICA that uses HSIC as the independence criterion (with Hao Shen, Arthur Gretton).
Download Code for Fast kernel ICA
S. Jegelka, F. Bach, S. Sra
(2013). Reflection methods for user-friendly submodular optimization In: Advances in Neural Information Processing Systems 26, (Ed) C.J.C. Burges and L. Bottou and M. Welling and Z. Ghahramani and K.Q. Weinberger, 1313–1321, 27th Annual Conference on Neural Information Processing Systems (NIPS 2013), (In Proceedings)
S. Jegelka, H. Lin, J. Bilmes
(2011). On Fast Approximate Submodular Minimization In: Advances in Neural Information Processing Systems 24, (Ed) J Shawe-Taylor and RS Zemel and P Bartlett and F Pereira and KQ Weinberger, 460-468, Twenty-Fifth Annual Conference on Neural Information Processing Systems (NIPS 2011), (In Proceedings)
S. Jegelka, S. Sra, A. Banerjee
(2009). Approximation Algorithms for Tensor Clustering In: ALT 2009, (Ed) Gavalda, R. , G. Lugosi, T. Zeugmann, S. Zilles, Algorithmic Learning Theory: 20th International Conference (ALT 2009), Springer, Berlin, Germany, 368-383, The 20th International Conference on Algorithmic Learning Theory, (In Proceedings)
S. Nowozin, S. Jegelka
(2009). Solution Stability in Linear Programming Relaxations: Graph Partitioning and Unsupervised Learning In: ICML 2009, (Ed) Danyluk, A. , L. Bottou, M. Littman, Proceedings of the 26th International Conference on Machine Learning (ICML 2009), ACM Press, New York, NY, USA, 769-776, 26th International Conference on Machine Learning, (In Proceedings)
S. Jegelka, A. Gretton, B. Schölkopf, BK. Sriperumbudur, U. Luxburg
(2009). Generalized Clustering via Kernel Embeddings In: KI 2009: AI and Automation, Lecture Notes in Computer Science, Vol. 5803, (Ed) B Mertsching and M Hund and Z Aziz, KI 2009: Advances in Artificial Intelligence, Springer, Berlin, Germany, 144-152, 32nd Annual Conference on Artificial Intelligence (KI), (In Proceedings)
U. Luxburg, S. Bubeck, S. Jegelka, M. Kaufmann
(2008). Consistent Minimization of Clustering Objective Functions In: Advances in neural information processing systems 20, (Ed) Platt, J. C., D. Koller, Y. Singer, S. Roweis, Advances in Neural Information Processing Systems 20: 21st Annual Conference on Neural Information Processing Systems 2007, Curran, Red Hook, NY, USA, 961-968, ISBN: 978-1-605-60352-0, Twenty-First Annual Conference on Neural Information Processing Systems (NIPS 2007), (In Proceedings)
H. Shen, S. Jegelka, A. Gretton
(2007). Fast Kernel ICA using an Approximate Newton Method In: JMLR Workshop and Conference Proceedings Volume 2: AISTATS 2007, (Ed) Meila, M. , X. Shen, Proceedings of the 11th International Conference on Artificial Intelligence and Statistics (AISTATS 2007), MIT Press, Cambridge, MA, USA, 476-483, 11th International Conference on Artificial Intelligence and Statistics, (In Proceedings)