Mutual Information Estimation via Score-to-Fisher Bridge for Nonlinear Gaussian Noise Channels
Tadashi Wadayama
TL;DR
This work addresses estimating mutual information in nonlinear Gaussian noise channels without posterior access by establishing a Score-to-Fisher bridge. It uses denoising score matching to recover the score of the channel output, from which Fisher information is estimated and integrated via a Fisher integral representation to obtain $I(X;Y_T)$, exploiting the equivalence $I(X;Y_t)=I(W;Y_t)$ with $W=f(X)$ for deterministic front-ends. The framework includes two practical training schemes (per-$t$ and noise-conditional), a robust log-domain trapezoid integration with tail corrections, and is validated on Gaussian and nonlinear channels against analytic solutions and KDE baselines, showing high accuracy and scalability. The approach avoids posterior computations and partitions functions, offering a scalable, channel-agnostic tool with direct relevance to coding, preprocessing, and sensing system design where nonlinear front-ends are present.
Abstract
We present a numerical method to evaluate mutual information (MI) in nonlinear Gaussian noise channels by using denoising score matching (DSM) learning for estimating the score function of channel output. Via de Bruijn's identity, Fisher information estimated from the learned score function yields accurate estimates of MI through a Fisher integral representation for a variety of priors and channel nonlinearities. In this work, we propose a comprehensive theoretical foundation for the Score-to-Fisher bridge methodology, along with practical guidelines for its implementation. We also conduct extensive validation experiments, comparing our approach with closed-form solutions and a kernel density estimation baseline. The results of our numerical experiments demonstrate that the proposed method is both practical and efficient for MI estimation in nonlinear Gaussian noise channels.
