On Graph Reconstruction via Empirical Risk Minimization: Fast Learning Rates and Scalability

摘要

The problem of predicting connections between a set of data points finds many applications, in systems biology and social network analysis among others. This paper focuses on the graph reconstruction problem, where the prediction rule is obtained by minimizing the average error over all n(n - 1)/2 possible pairs of the n nodes of a training graph. Our first contribution is to derive learning rates of order O_P (log n) for this problem, significantly improving upon the slow rates of order O_P(1 /√n) established in the seminal work of Biau and Bleakley (2006). Strikingly, these fast rates are universal, in contrast to similar results known for other statistical learning problems (e.g., classification, density level set estimation, ranking, clustering) which require strong assumptions on the distribution of the data. Motivated by applications to large graphs, our second contribution deals with the computational complexity of graph reconstruction. Specifically, we investigate to which extent the learning rates can be preserved when replacing the empirical reconstruction risk by a computationally cheaper Monte-Carlo version, obtained by sampling with replacement B n~2 pairs of nodes. Finally, we illustrate our theoretical results by numerical experiments on synthetic and real graphs.