Search papers, labs, and topics across Lattice.
This paper introduces a Bayesian meta-learning method that uses precomputed latent causal task embeddings to condition task-specific priors, enabling transfer based on mechanistic similarity. The approach addresses the problem of negative transfer in out-of-distribution meta-learning scenarios, particularly when adapting to new tasks with limited data and relying on noisy expert feedback about task similarity. The method demonstrates improved out-of-distribution adaptation and reduced negative transfer in simulations and a real-world clinical prediction task involving cross-disease transfer.
Causal embeddings can rescue meta-learning from negative transfer in out-of-distribution scenarios, even with noisy expert feedback.
Meta-learning methods perform well on new within-distribution tasks but often fail when adapting to out-of-distribution target tasks, where transfer from source tasks can induce negative transfer. We propose a causally-aware Bayesian meta-learning method, by conditioning task-specific priors on precomputed latent causal task embeddings, enabling transfer based on mechanistic similarity rather than spurious correlations. Our approach explicitly considers realistic deployment settings where access to target-task data is limited, and adaptation relies on noisy (expert-provided) pairwise judgments of causal similarity between source and target tasks. We provide a theoretical analysis showing that conditioning on causal embeddings controls prior mismatch and mitigates negative transfer under task shift. Empirically, we demonstrate reductions in negative transfer and improved out-of-distribution adaptation in both controlled simulations and a large-scale real-world clinical prediction setting for cross-disease transfer, where causal embeddings align with underlying clinical mechanisms.
