Contextualized Machine Learning
Not on the map. Why?
Abstract (arXiv)
We examine Contextualized Machine Learning (ML), a paradigm for learning heterogeneous and context-dependent effects. Contextualized ML estimates heterogeneous functions by applying deep learning to the meta-relationship between contextual information and context-specific parametric models. This is a form of varying-coefficient modeling that unifies existing frameworks including cluster analysis and cohort modeling by introducing two reusable concepts: a context encoder which translates sample context into model parameters, and sample-specific model which operates on sample predictors. We review the process of developing contextualized models, nonparametric inference from contextualized models, and identifiability conditions of contextualized models. Finally, we present the open-source PyTorch package ContextualizedML.
Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Builds on 16 of its 35 references that are in the map. First posted on arXiv 17 Oct 2023.
Builds on
16 of 35 references.
References · 16
- 2016"Why Should I Trust You?": Explaining the Predictions of Any Classifiercited by 5,070
- 2020Neural Additive Models: Interpretable Machine Learning with Neural Netscited by 319
- 2016Not Just a Black Box: Learning Important Features Through Propagating Activation Differencescited by 297
- 2016Model-Agnostic Interpretability of Machine Learningcited by 219
- 2016Generating Visual Explanationscited by 158
- 2017Interpretable & Explorable Approximations of Black Box Modelscited by 54
- 2019Unremarkable AI: Fitting Intelligent Decision Support into Critical, Clinical Decision-Making Processescited by 53
- 2019Purifying Interaction Effects with the Functional ANOVA: An Efficient Algorithm for Recovering Identifiable Additive Modelscited by 42
- 2018Contextual Parameter Generation for Universal Neural Machine Translationcited by 40
- 2017Contextual Explanation Networkscited by 21
- 2018Identifiability of Nonparametric Mixture Models and Bayes Optimal Clusteringcited by 15
- 2019Learning Sample-Specific Models with Low-Rank Personalized Regressioncited by 5
4 more
Built on by
Citing papers per year: 2023: 1; 2024: 2; 2025: 0; 2026: 1.
4 papers.
- 2023Contextual Feature Selection with Conditional Stochastic Gatescited by 7
- 2024Toward AI-Driven Digital Organism: Multiscale Foundation Models for Predicting, Simulating and Programming Biology at All Levelscited by 5
- 2026Retrieval-Augmented Interpretable Learning: Towards Task-Specific Zero-Shot Models in Healthcarecited by 1
- 2024Patient-Specific Models of Treatment Effects Explain Heterogeneity in Tuberculosiscited by 0
- 2026Retrieval-Augmented Interpretable Learning: Towards Task-Specific Zero-Shot Models in Healthcare20 Jul 2026
- 2024Toward AI-Driven Digital Organism: Multiscale Foundation Models for Predicting, Simulating and Programming Biology at All Levels9 Dec 2024
- 2024Patient-Specific Models of Treatment Effects Explain Heterogeneity in Tuberculosis16 Nov 2024
- 2023Contextual Feature Selection with Conditional Stochastic Gates21 Dec 2023