A Non–Convex Optimization Approach to Correlation Clustering

Authors

  • Erik Thiel Chalmer University of Technology
  • Morteza Haghir Chehreghani Chalmers University of Technology
  • Devdatt Dubhashi Chalmers University of Technology

DOI:

https://doi.org/10.1609/aaai.v33i01.33015159

Abstract

We develop a non-convex optimization approach to correlation clustering using the Frank-Wolfe (FW) framework. We show that the basic approach leads to a simple and natural local search algorithm with guaranteed convergence. This algorithm already beats alternative algorithms by substantial margins in both running time and quality of the clustering. Using ideas from FW algorithms, we develop subsampling and variance reduction paradigms for this approach. This yields both a practical improvement of the algorithm and some interesting further directions to investigate. We demonstrate the performance on both synthetic and real world data sets.

Downloads

Published

2019-07-17

How to Cite

Thiel, E., Chehreghani, M. H., & Dubhashi, D. (2019). A Non–Convex Optimization Approach to Correlation Clustering. Proceedings of the AAAI Conference on Artificial Intelligence, 33(01), 5159-5166. https://doi.org/10.1609/aaai.v33i01.33015159

Issue

Section

AAAI Technical Track: Machine Learning