Graph Regularized PCA
Antonio Briola, Marwin Schmidt, Fabio Caccioli, Carlos Ros Perez, James Singleton, Christian Michler, Tomaso Aste
TL;DR
Graph-regularized PCA (GR-PCA) extends classical PCA to structured data by learning a sparse precision graph over features and adding a graph-Laplacian penalty to bias loadings toward low-frequency, graph-consistent patterns. The approach solves a joint objective that balances reconstruction accuracy, sparsity, and smoothness across the learned graph, enabling structure-aware dimensionality reduction that improves interpretability and robustness to anisotropic noise. Through a synthetic data generator and systematic experiments across ER/BA/WS topologies, GR-PCA demonstrates higher selectivity and alignment to ground-truth low-frequency components while maintaining competitive out-of-sample reconstruction, especially when high-frequency signals align with graph structure. The method is simple, scalable, and adaptable to learned or oracle precision graphs, offering a practical route to structure-aware PCA in domains like neuroscience, climate science, and genomics.
Abstract
High-dimensional data often exhibit dependencies among variables that violate the isotropic-noise assumption under which principal component analysis (PCA) is optimal. For cases where the noise is not independent and identically distributed across features (i.e., the covariance is not spherical) we introduce Graph Regularized PCA (GR-PCA). It is a graph-based regularization of PCA that incorporates the dependency structure of the data features by learning a sparse precision graph and biasing loadings toward the low-frequency Fourier modes of the corresponding graph Laplacian. Consequently, high-frequency signals are suppressed, while graph-coherent low-frequency ones are preserved, yielding interpretable principal components aligned with conditional relationships. We evaluate GR-PCA on synthetic data spanning diverse graph topologies, signal-to-noise ratios, and sparsity levels. Compared to mainstream alternatives, it concentrates variance on the intended support, produces loadings with lower graph-Laplacian energy, and remains competitive in out-of-sample reconstruction. When high-frequency signals are present, the graph Laplacian penalty prevents overfitting, reducing the reconstruction accuracy but improving structural fidelity. The advantage over PCA is most pronounced when high-frequency signals are graph-correlated, whereas PCA remains competitive when such signals are nearly rotationally invariant. The procedure is simple to implement, modular with respect to the precision estimator, and scalable, providing a practical route to structure-aware dimensionality reduction that improves structural fidelity without sacrificing predictive performance.
