mlsanityData to 7 Oct 2026

LucidAtlas: Learning Uncertainty-Aware, Covariate-Disentangled, Individualized Atlas Representations

arXiv 2502.08445 ↗First posted 12 Feb 2025cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

The goal of this work is to develop principled techniques to extract information from high dimensional data sets with complex dependencies in areas such as medicine that can provide insight into individual as well as population level variation. We develop textttLucidAtlas, an approach that can represent spatially varying information, and can capture the influence of covariates as well as population uncertainty. As a versatile atlas representation, textttLucidAtlas offers robust capabilities for covariate interpretation, individualized prediction, population trend analysis, and uncertainty estimation, with the flexibility to incorporate prior knowledge. Additionally, we discuss the trustworthiness and potential risks of neural additive models for analyzing dependent covariates and then introduce a marginalization approach to explain the dependence of an individual predictor on the models' response (the atlas). To validate our method, we demonstrate its generalizability on two medical datasets. Our findings underscore the critical role of by-construction interpretable models in advancing scientific discovery. Our code will be publicly available upon acceptance.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 38 of its 71 references that are in the map. First posted on arXiv 12 Feb 2025.

Builds on

38 of 71 references.

References · 38

  1. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
  2. 2017A Unified Approach to Interpreting Model Predictionscited by 6,488
  3. 2016"Why Should I Trust You?": Explaining the Predictions of Any Classifiercited by 5,040
  4. 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,416
  5. 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,888
  6. 2018CBAM: Convolutional Block Attention Modulecited by 2,463
  7. 2015Learning Deep Features for Discriminative Localizationcited by 2,430
  8. 20163D U-Net: Learning Dense Volumetric Segmentation from Sparse Annotationcited by 1,471
  9. 2019Aleatoric and Epistemic Uncertainty in Machine Learning: An Introduction to Concepts and Methodscited by 903
  10. 2019TabNet: Attentive Interpretable Tabular Learningcited by 640
  11. 2021Laplace Redux -- Effortless Bayesian Deep Learningcited by 359
  12. 2020Neural Additive Models: Interpretable Machine Learning with Neural Netscited by 318
26 more
  1. 2019InterpretML: A Unified Framework for Machine Learning Interpretabilitycited by 243
  2. 2018Disentangled Representation Learning for Non-Parallel Text Style Transfercited by 127
  3. 2021GAN-Control: Explicitly Controllable GANscited by 91
  4. 2020Causal Interpretability for Machine Learning -- Problems, Methods and Evaluationcited by 82
  5. 2018Unsupervised Discovery of Object Landmarks as Structural Representationscited by 73
  6. 2020Learning Geometry-Disentangled Representation for Complementary Understanding of 3D Object Point Cloudcited by 71
  7. 2019Disentangled Representation Learning in Cardiac Image Analysiscited by 69
  8. 2017Deep Lattice Networks and Partial Monotonic Functionscited by 62
  9. 2019Block Neural Autoregressive Flowcited by 62
  10. 2018Learning to Reconstruct Shapes from Unseen Classescited by 60
  11. 2020Guided Variational Autoencoder for Disentanglement Learningcited by 57
  12. 2020Certified Monotonic Neural Networkscited by 51
  13. 2020Learning identifiable and interpretable latent models of high-dimensional neural activity using pi-VAEcited by 48
  14. 2021Towards Better Explanations of Class Activation Mappingcited by 47
  15. 2022Faithful Heteroscedastic Regression with Neural Networkscited by 36
  16. 2022Constrained Monotonic Neural Networkscited by 31
  17. 2021Explaining Latent Representations with a Corpus of Examplescited by 30
  18. 2022Additive Gaussian Processes Revisitedcited by 19
  19. 2023Neural Additive Models for Location Scale and Shape: A Framework for Interpretable Neural Regression Beyond the Meancited by 18
  20. 2025From Aleatoric to Epistemic: Exploring Uncertainty Quantification Techniques in Artificial Intelligencecited by 17
  21. 2023Improving Neural Additive Models with Bayesian Principlescited by 15
  22. 2023Curve Your Enthusiasm: Concurvity Regularization in Differentiable Generalized Additive Modelscited by 12
  23. 2020Calibrating Healthcare AI: Towards Reliable and Interpretable Deep Predictive Modelscited by 9
  24. 2023NAISR: A 3D Neural Additive Model for Interpretable Shape Representationcited by 7
  25. 2023Expressive Monotonic Neural Networkscited by 7
  26. 2022Covariate-Balancing-Aware Interpretable Deep Learning models for Treatment Effect Estimationcited by 5

Built on by

Not cited yet.