use minibatch k-means, still not fast enough :-/