Amortized Factor Inference Networks for Posterior Inference
Amortized Factor Inference Networks (AFINs) introduce a single inference network that can generalize across different priors, likelihoods, and dimensions. It uses an encode-merge-decode architecture to map model descriptions and observations to a Gaussian posterior approximation. This allows for competitive accuracy compared to complex methods while requiring significantly less test-time computation.