Generative Unsupervised Ensemble-based Self-supervision for Test-Time Adaptation in Medical Imaging

Friday 31 January 2025


Deep learning has revolutionized many areas of computer vision, but one challenge that remains is adapting models to new domains without extensive retraining. This problem is particularly pressing in medical imaging, where datasets are often limited and domain shifts can be significant.


A team of researchers has proposed a novel approach to tackle this issue by developing a test-time adaptation method that learns to generate perturbations on the fly. This technique, called GUES (Generative Unsupervised Ensemble-based Self-supervision), uses an ensemble of networks to predict individual perturbations for each input image.


The key insight behind GUES is that traditional saliency maps are not suitable for natural images, as they highlight all variations in the scene rather than focusing on relevant features. In contrast, fundus images have a simpler structure and background, making it easier to identify important regions. By leveraging this difference, GUES generates perturbations that are tailored to the unique characteristics of each input image.


The researchers evaluated their method on four standard diabetic retinopathy (DR) benchmarks, including APTOS, DDR, DeepDR, and Messidor-2. They found that GUES outperformed existing test-time adaptation methods in terms of accuracy, quadratic weighted kappa, and average performance.


One of the most interesting aspects of GUES is its ability to adapt to different batch sizes without significant performance degradation. This is particularly important in medical imaging, where data may be limited or come from diverse sources. By predicting individual perturbations for each input image, GUES can effectively compensate for variations in batch size and ensure consistent performance.


The team’s approach also provides valuable insights into the characteristics of different datasets. For example, they found that the RGB statistics of proliferative DR samples across the four datasets exhibited distinct fluctuations, highlighting the unique visual styles and characteristics of each dataset.


While GUES is a promising solution for test-time adaptation in medical imaging, there are still challenges to overcome. One area of future research is refining the self-supervised signal for natural images, which currently struggles to identify relevant features. However, the potential benefits of GUES make it an exciting development that has the potential to revolutionize the field of medical image analysis.


In this study, the researchers demonstrated the effectiveness of their proposed method by evaluating it on four standard diabetic retinopathy (DR) benchmarks and comparing its performance with existing test-time adaptation methods.


Cite this article: “Generative Unsupervised Ensemble-based Self-supervision for Test-Time Adaptation in Medical Imaging”, The Science Archive, 2025.


Deep Learning, Medical Imaging, Computer Vision, Diabetic Retinopathy, Test-Time Adaptation, Gues, Generative Unsupervised Ensemble-Based Self-Supervision, Saliency Maps, Perturbations, Ensemble Networks.


Reference: Wenxin Su, Song Tang, Xiaofeng Liu, Xiaojing Yi, Mao Ye, Chunxiao Zu, Jiahao Li, Xiatian Zhu, “Domain Adaptive Diabetic Retinopathy Grading with Model Absence and Flowing Data” (2024).


Leave a Reply