Self-Supervised Learning for Sparse Matrix Reordering
About
Rearranging the rows or columns of a sparse matrix using an appropriate ordering can significantly reduce fill-ins, i.e., new nonzeros introduced during matrix factorization, decreasing memory usage and runtime. However, finding an ordering that minimizes fill-ins is NP-complete. Existing approaches, including graph-theoretic and deep learning methods, rely on surrogate objectives without theoretical guarantees. The Fill-Path Theorem reveals a direct and intrinsic relationship between fill-in generation and the sparse structure of the matrix as path triplet inequalities. Here we first employ a multigrid graph network to capture structural information for each vertex. We then derive a triplet sampling strategy based on inequalities. Finally, we introduce an end-max chain loss function to reduce the number of triplets whose predicted scores satisfy these inequalities. Experimental evaluations on the publicly available SuiteSparse matrix collection demonstrate the superiority of the proposed method in terms of both fill-in reduction and speedup in LU factorization time.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Matrix Reordering for LU Factorization | Benchmark Test Matrices 2D3D | LU Factorization Time (s)91.9 | 8 | |
| Matrix Reordering for LU Factorization | Benchmark Test Matrices CFD | LU Factorization Time (s)99.91 | 8 | |
| Matrix Reordering for LU Factorization | Benchmark Test Matrices MRP | LU Factorization Time (s)7.83 | 8 | |
| Matrix Reordering for LU Factorization | Benchmark Matrices SP (test) | LU Factorization Time (s)76.43 | 8 | |
| Matrix Reordering for LU Factorization | Benchmark Test Matrices TP | LU Factorization Time (s)525 | 8 | |
| Matrix Reordering for LU Factorization | Benchmark Test Matrices All | LU Factorization Time (s)95.23 | 8 | |
| Sparse Matrix Reordering | SuiteSparse matrix collection (test) | CFD57.87 | 8 |