Search papers, labs, and topics across Lattice.
This paper introduces a novel framework that integrates unsupervised domain adaptation with multitask balancing to address extreme class imbalance in image analysis. By evaluating this approach in the context of the Cherenkov Telescope Array Observatory (CTAO), the authors demonstrate how their method effectively mitigates the challenges posed by extreme label shifts. The results show significant improvements in model performance, highlighting the framework's potential for real-world applications in astrophysics and beyond.
Extreme label shifts can be effectively tackled by combining domain adaptation with multitask learning, leading to improved performance in complex image analysis tasks.
Unsupervised domain adaptation is a widespread set of methods that leverages the knowledge of a labeled source domain to train a model to perform well on a related unlabeled target domain. They generally introduce an auxiliary adaptation-related task that can be integrated into the multitask paradigm, which aims to merge multiple single-task models into a unified architecture. In this paper, we propose to associate domain adaptation and multitask balancing in the realistic context of an extreme class imbalance. Therefore, we propose a combined framework to cover and validate these approaches, and evaluate its performance in the physics-based context of the Cherenkov Telescope Array Observatory (CTAO). Along with a comparative study of some relevant adaptation techniques, we highlight the impact of extreme label shift and extend the investigations on importance weighting to rectify it. The complete code and results are published and available as open-source resources on Zenodo.
