ICML2026
Calibrated Knowledge Aggregation in Bayesian Mixture-of-Experts for Continual VQA
Mahsa Mozaffari, Hitesh Sapkota, Yu Kong, Xumin Liu, Qi Yu
摘要
Continual learning for visual question answering (VQA) is typically implemented by training one expert per task and routing each query using task-ID supervision. Yet continual VQA tasks overlap substantially: on the VQA-v2 task stream, a non-native expert outperforms the task’s own expert on of queries, so hard routing both wastes transferable knowledge and can be confidently wrong when mismatched. We propose a calibrated Bayesian mixture-of-experts that trains parameter-efficient per-task adapters, learns routing by directly maximizing expected VQA utility, and marginalizes expert identity at inference via Bayesian aggregation in a unified answer space; an entropy penalty prevents the utility objective from collapsing to one-hot routing, enabling evidence pooling across plausible experts. We reach accuracy with forgetting on VQA-v2 CL-LS ( accuracy, forgetting vs. the strongest prior method), with forgetting on TDIUC CL-LS (, ), and with forgetting on TDIUC CL-VS (, ). Calibration also improves on VQA-v2, reducing ECE from to .