neural-estimator

InfoAtlas是一种基础模型，能够通过单次前向传播直接估算互信息，在保持准确性的同时，比传统神经估算器实现100倍加速。该模型在合成数据上预训练，并能泛化到现实场景。