TY - JOUR AU - Lin, Tiancheng AU - Xu, Hongteng AU - Yang, Canqian AU - Xu, Yi PY - 2022/06/28 Y2 - 2024/03/28 TI - Interventional Multi-Instance Learning with Deconfounded Instance-Level Prediction JF - Proceedings of the AAAI Conference on Artificial Intelligence JA - AAAI VL - 36 IS - 2 SE - AAAI Technical Track on Computer Vision II DO - 10.1609/aaai.v36i2.20051 UR - https://ojs.aaai.org/index.php/AAAI/article/view/20051 SP - 1601-1609 AB - When applying multi-instance learning (MIL) to make predictions for bags of instances, the prediction accuracy of an instance often depends on not only the instance itself but also its context in the corresponding bag. From the viewpoint of causal inference, such bag contextual prior works as a confounder and may result in model robustness and interpretability issues. Focusing on this problem, we propose a novel interventional multi-instance learning (IMIL) framework to achieve deconfounded instance-level prediction. Unlike traditional likelihood-based strategies, we design an Expectation-Maximization (EM) algorithm based on causal intervention, providing a robust instance selection in the training phase and suppressing the bias caused by the bag contextual prior. Experiments on pathological image analysis demonstrate that our IMIL method substantially reduces false positives and outperforms state-of-the-art MIL methods. ER -