Search papers, labs, and topics across Lattice.
This paper introduces a three-stage, source-aware multi-expert framework for COVID-19 CT classification designed to handle challenges like source shift and imbalance in multi-institutional settings. The framework combines a lung-aware 3D expert, two MedSigLIP-based experts for slice-wise representation and inter-slice context modeling, and a source classifier to predict the origin of each scan. By fusing expert predictions based on the predicted source, the model achieves strong performance across multiple datasets, demonstrating robustness to heterogeneous data.
Achieve state-of-the-art COVID-19 CT classification by predicting the data source and using that to intelligently fuse multiple specialized models.
Robust detection of COVID-19 from chest CT remains challenging in multi-institutional settings due to substantial source shift, source imbalance, and hidden test-source identities. In this work, we propose a three-stage source-aware multi-expert framework for multi-source COVID-19 CT classification. First, we build a lung-aware 3D expert by combining original CT volumes and lung-extracted CT volumes for volumetric classification. Second, we develop two MedSigLIP-based experts: a slice-wise representation and probability learning module, and a Transformer-based inter-slice context modeling module for capturing cross-slice dependency. Third, we train a source classifier to predict the latent source identity of each test scan. By leveraging the predicted source information, we perform model fusion and voting based on different experts. On the validation set covering all four sources, the Stage 1 model achieves the best macro-F1 of 0.9711, ACC of 0.9712, and AUC of 0.9791. Stage~2a and Stage~2b achieve the best AUC scores of 0.9864 and 0.9854, respectively. Stage~3 source classifier reaches 0.9107 ACC and 0.9114 F1. These results demonstrate that source-aware expert modeling and hierarchical voting provide an effective solution for robust COVID-19 CT classification under heterogeneous multi-source conditions.
