CLUSTERING DATA USING NEURAL NETWORKS BASED ON NORMALIZED CUTS

摘要

Methods, systems, and apparatus, including computer programs encoded on computer storage media, for training a clustering neural network. One of the methods includes obtaining unlabeled training data; and training the clustering neural network on the unlabeled training data to determine trained values of the clustering parameters by minimizing a normalized cuts loss function that includes a first term that measures an expected normalized cuts of clustering nodes in a graph representing the data set into the plurality of clusters according to clustering outputs generated by the clustering neural network.