Universality of AMP via Tree Pairings

Abstract

We prove universality for Approximate Message Passing (AMP) with polynomial nonlinearities applied to symmetric sub-Gaussian matrices $A\in\mathbb R^{N\times N}$. Our approach is combinatorial: we represent AMP iterates as sums over trees and define a Wick pairing algebra that counts the number of valid row-wise pairings of edges. The number of such pairings coincides with the trees contribution to the state evolution formulas. This algebra works for non-Gaussian entries. For polynomial nonlinearities of degree at most $D$, we show that the moments of AMP iterates match their state evolution predictions for $t \lesssim \frac{\log N}{D\log D}$ iterations. The proof controls all "excess" trees via explicit enumeration bounds, showing non "Wick-paired" contributions vanish in the large-$N$ limit. The same framework should apply, with some modifications, to spiked AMP and tensor AMP models.