Multi-Instance Causal Representation Learning for Instance Label Prediction and Out-of-Distribution Generalization

Abstract

Learning causal representation from multi-instance bags significantly improves performances of both MIL and OOD tasks.