Spatial Disease Mapping and Disparity Detection Using Generative AI: An Amortized Bayesian Learning Framework
使用生成式AI的空间疾病映射与差异检测:一种摊销贝叶斯学习框架
Luca Aiello, Sudipto Banerjee
AI总结 提出一种摊销贝叶斯框架,通过神经网络近似后验分布,实现跨不同区域图的空间边界检测,并在呼吸疾病和肺癌数据中验证其有效性。
详情
我们引入了一个用于空间边界检测的摊销贝叶斯框架,该框架能够推广到具有不同区域数量和多样邻接结构的区域图上的后验推断。底层模型将泊松计数似然与协变量驱动的规则相结合,以中断跨不相似相邻区域的平滑,并利用有向无环图自回归(DAGAR)先验来捕捉残差空间依赖性。为了逼近目标后验分布,我们在模拟地图上训练了一个神经引擎:一个置换不变摘要网络编码观测计数、偏移量、协变量和邻接矩阵的图感知表示,而一个条件归一化流生成近似的后验样本。模拟研究证明了准确的参数恢复、接近名义水平的区间覆盖、良好校准的后验预测行为以及信息丰富的后验边界概率。与马尔可夫链蒙特卡洛(MCMC)的基准测试证实了在主要边界证据上的紧密一致性,而消融研究验证了包含模型引导的图摘要的有效性。最后,应用于格拉斯哥呼吸系统疾病和加利福尼亚肺癌数据表明,一个训练好的神经引擎可以无缝部署到具有不同图结构的真实世界地图上,产生的边界结论与已建立的局部平滑分析一致。
We introduce an amortized Bayesian framework for spatial boundary detection that generalizes posterior inference across areal graphs with varying numbers of regions and diverse adjacency structures. The underlying model couples a Poisson count likelihood with a covariate-driven rule to interrupt smoothing across dissimilar neighboring areas, utilizing a directed acyclic graph autoregressive (DAGAR) prior to capture residual spatial dependence. To approximate the target posterior distribution, a neural engine is trained on simulated maps: a permutation-invariant summary network encodes graph-aware representations of the observed counts, offsets, covariates, and adjacency matrices, while a conditional normalizing flow generates the approximate posterior draws. Simulation studies demonstrate accurate parameter recovery, near-nominal interval coverage, well-calibrated posterior predictive behavior, and informative posterior boundary probabilities. Benchmarking against Markov chain Monte Carlo (MCMC) confirms close agreement regarding primary boundary evidence, and an ablation study validates the inclusion of model-guided graph summaries. Finally, applications to Glasgow respiratory disease and California lung cancer data demonstrate that a single trained neural engine can be seamlessly deployed across real-world maps with distinct graph structures, yielding boundary conclusions consistent with established localized smoothing analyses.