How to learn a graph from smooth signals
About
We propose a framework that learns the graph structure underlying a set of smooth signals. Given $X\in\mathbb{R}^{m\times n}$ whose rows reside on the vertices of an unknown graph, we learn the edge weights $w\in\mathbb{R}_+^{m(m-1)/2}$ under the smoothness assumption that $\text{tr}{X^\top LX}$ is small. We show that the problem is a weighted $\ell$-1 minimization that leads to naturally sparse solutions. We point out how known graph learning or construction techniques fall within our framework and propose a new model that performs better than the state of the art in many settings. We present efficient, scalable primal-dual based algorithms for both our model and the previous state of the art, and evaluate their performance on artificial and real data.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Binary graph structure recovery | Community (SBM) binary (test) | Community Score48.1 | 8 | |
| Binary graph structure recovery | Scale-free BA binary (test) | KS Test Score65.63 | 8 | |
| Binary graph structure recovery | Small-world (WS) binary (test) | Avg Shortest Path Length2.656 | 8 | |
| Graph Reconstruction | Scale-free BA (test) | GMSE0.4033 | 6 | |
| Graph Reconstruction | Community SBM (test) | GMSE0.3111 | 6 | |
| Graph Reconstruction | Small-world (WS) (test) | GMSE0.218 | 6 | |
| Graph Reconstruction | Random sparse (ER) (test) | GMSE0.3799 | 6 |