:A 3D Probabilistic Neural Representation for Interpretable Shape Modeling
Yining Jiao, Shankar Bhamidi, Carlton ZDANSKI, Julia Kimbell, Andrew Prince, Cameron Worden, Samuel Kirse, Christopher Rutter, Benjamin Shields, William Dunn, Jisan Mahmud, Marc Niethammer
Abstract
Understanding how anatomical shapes evolve in response to developmental covariates-and quantifying their spatially varying uncertainties-is critical in healthcare research. Existing approaches typically rely on global time-warping formulations that ignore spatially heterogeneous dynamics. We introduce PRISM, a novel framework that bridges implicit neural representations with uncertainty-aware statistical shape analysis. PRISM models the conditional distribution of shapes given covariates, providing spatially continuous estimates of both the population mean and covariate-dependent uncertainty at arbitrary locations. A key theoretical contribution is a closed-form Fisher Information metric that enables efficient, analytically tractable local temporal uncertainty quantification via automatic differentiation. Experiments on three synthetic datasets and one clinical dataset demonstrate PRISM's strong performance across diverse tasks-from modeling shape evolution to personalized shape prediction and anomaly detection-within a unified framework, while providing interpretable and clinically meaningful uncertainty estimates. The code is publicly available at https://github.com/uncbiag/PRISM.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on10
- Implicit Neural Representations with Periodic Activation FunctionsVincent Sitzmann, Julien N. P. Martel, Alexander W. Bergman, David B. Lindell et al.NeurIPS 2020 · 4,008 citations
- A-SDF: Learning Disentangled Signed Distance Functions for Articulated Shape RepresentationJiteng Mu, Weichao Qiu, Adam Kortylewski, Alan L. Yuille et al.ICCV 2021 · 138 citations
- On the Pitfalls of Heteroscedastic Uncertainty Estimation with Probabilistic Neural NetworksMaximilian Seitzer, Arash Tavakoli, Dimitrije Antic, Georg MartiusICLR 2022 · 122 citations
- ICON: Learning Regular Maps Through Inverse ConsistencyThomas Hastings Greer, Roland Kwitt, François-Xavier Vialard, Marc NiethammerICCV 2021 · 38 citations
- ImplicitAtlas: Learning Deformable Shape Templates in Medical ImagingJiancheng Yang, Udaranga Wickramasinghe, Bingbing Ni, Pascal FuaCVPR 2022 · 34 citations
Related papers
- FisherPoser: Human Motion Estimation from Sparse Observations with Hierarchical Region-Wise Fisher-Matrix Uncertainty ModelingSongpengcheng Xia, Qingyu Zhang, Zhuo Su, Jiarui Yang et al.CVPR 2026
- Adaptive Shrinkage Estimation for Personalized Deep Kernel Regression in Modeling Brain TrajectoriesVasiliki Tassopoulou, Haochang Shou, Christos DavatzikosICLR 2025
- NAISR: A 3D Neural Additive Model for Interpretable Shape RepresentationYining Jiao, Carlton J. Zdanski, Julia S. Kimbell, Andrew Prince et al.ICLR 2024 · 7 citations
- CaSPR: Learning Canonical Spatiotemporal Point Cloud RepresentationsDavis Rempe, Tolga Birdal, Yongheng Zhao, Zan Gojcic et al.NeurIPS 2020 · 78 citations
- Hierarchical Uncertainty Estimation for Learning-based Registration in NeuroimagingXiaoling Hu, Karthik Gopinath, Peirong Liu, Malte Hoffmann et al.ICLR 2025
