マルチインスタンス学習(MIL)を適用してインスタンスのバッグの予測を行う場合、インスタンスの予測精度は、インスタンス自体だけでなく、対応するバッグ内のコンテキストにも依存することがよくあります。因果推論の観点から、このようなバッグのコンテキストの事前処理は交絡因子として機能し、モデルの堅牢性と解釈可能性の問題を引き起こす可能性があります。この問題に焦点を当て、混乱を解消したインスタンスレベルの予測を実現するための新しい介入型マルチインスタンス学習(IMIL)フレームワークを提案します。従来の尤度ベースの戦略とは異なり、因果的介入に基づいて期待値最大化(EM)アルゴリズムを設計し、トレーニングフェーズで堅牢なインスタンス選択を提供し、バッグのコンテキストプライアによって引き起こされるバイアスを抑制します。病理学的画像分析の実験は、私たちのIMIL法が誤検出を大幅に減らし、最先端のMIL法よりも優れていることを示しています。
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.