Search papers, labs, and topics across Lattice.
This paper addresses source-free domain adaptation (SFDA) for fundus image analysis by leveraging vision-language (ViL) models while mitigating the forgetting of target model predictions and incorporating fine-grained lesion knowledge. They propose a forgetting-resistant and lesion-aware (FRLA) method that includes a forgetting-resistant adaptation module to preserve confident target model predictions and a lesion-aware adaptation module to utilize patch-wise ViL model predictions for lesion area awareness. Experiments demonstrate that FRLA outperforms existing ViL-based SFDA methods and achieves state-of-the-art results.
Overcome the "forgetting" problem in source-free domain adaptation by explicitly preserving confident predictions, leading to state-of-the-art fundus image diagnosis.
Source-free domain adaptation (SFDA) aims to adapt a model trained in the source domain to perform well in the target domain, with only unlabeled target domain data and the source model. Taking into account that conventional SFDA methods are inevitably error-prone under domain shift, recently greater attention has been directed to SFDA assisted with off-the-shelf foundation models, e.g., vision-language (ViL) models. However, existing works of leveraging ViL models for SFDA confront two issues: (i) Although mutual information is exploited to consider the joint distribution between the predictions of ViL model and the target model, we argue that the forgetting of some superior predictions of the target model still occurs, as indicated by the decline of the accuracies of certain classes during adaptation; (ii) Prior research disregards the rich, fine-grained knowledge embedded in the ViL model, which offers detailed grounding for fundus image diagnosis. In this paper, we introduce a novel forgetting-resistant and lesion-aware (FRLA) method for SFDA of fundus image diagnosis with ViL model. Specifically, a forgetting-resistant adaptation module explicitly preserves the confident predictions of the target model, and a lesion-aware adaptation module yields patch-wise predictions from ViL model and employs them to help the target model be aware of the lesion areas and leverage the ViL model's fine-grained knowledge. Extensive experiments show that our method not only significantly outperforms the vision-language model, but also achieves consistent improvements over the state-of-the-art methods. Our code will be released.
