Search papers, labs, and topics across Lattice.
This paper introduces DsmNet, a GNN architecture that replaces the standard Laplacian with a Doubly Stochastic Matrix (DSM) derived from the inverse of the modified Laplacian to encode multi-hop proximity and local centrality. To address the computational cost of DSM calculation, they approximate it using a truncated Neumann series and further introduce a Residual Mass Compensation mechanism (DsmNet-compensate) to maintain row-stochasticity. Theoretical analysis demonstrates that DsmNet mitigates over-smoothing by bounding Dirichlet energy decay, and empirical results validate its performance on homophilic benchmarks and its applicability to Graph Transformers.
Escaping the limitations of standard graph Laplacians, DsmNet unlocks efficient multi-hop message passing and mitigates over-smoothing
Graph Neural Networks (GNNs) conventionally rely on standard Laplacian or adjacency matrices for structural message passing. In this work, we substitute the traditional Laplacian with a Doubly Stochastic graph Matrix (DSM), derived from the inverse of the modified Laplacian, to naturally encode continuous multi-hop proximity and strict local centrality. To overcome the intractable $O(n^3)$ complexity of exact matrix inversion, we first utilize a truncated Neumann series to scalably approximate the DSM, which serves as the foundation for our proposed DsmNet. Furthermore, because algebraic truncation inherently causes probability mass leakage, we introduce DsmNet-compensate. This variant features a mathematically rigorous Residual Mass Compensation mechanism that analytically re-injects the truncated tail mass into self-loops, strictly restoring row-stochasticity and structural dominance. Extensive theoretical and empirical analyses demonstrate that our decoupled architectures operate efficiently in $O(K|E|)$ time and effectively mitigate over-smoothing by bounding Dirichlet energy decay, providing robust empirical validation on homophilic benchmarks. Finally, we establish the theoretical boundaries of the DSM on heterophilic topologies and demonstrate its versatility as a continuous structural encoding for Graph Transformers.
