Learning to Trim: End-to-End Causal Graph Pruning with Dynamic Anatomical Feature Banks for Medical VQA
Researchers have developed a new framework called Learnable Causal Trimming (LCT) to improve the generalization of medical Visual Question Answering (MedVQA) models. This approach integrates causal pruning directly into the end-to-end optimization process. LCT utilizes a Dynamic Anatomical Feature Bank (DAFB) to capture common patterns and a differentiable trimming module that suppresses features correlated with these global prototypes, thereby encouraging the model to focus on instance-specific evidence. Experiments on several medical VQA datasets show that LCT enhances robustness and generalization compared to existing debiasing methods. AI
IMPACT Enhances robustness and generalization of medical VQA models by prioritizing causal signals over spurious correlations.