Understanding pre-training data effects in retinal foundation models using two large fundus cohorts
Abstract
Abstract Medical foundation models, pre-trained on large-scale unlabelled data, show strong performance and data efficiency when adapted to various clinically relevant applications. However, how pre-training data shape the generalisability and fairness of these models remains unexplored. Here we address this using two cohorts from Moorfields Eye Hospital (UK) and the Shanghai Diabetes Prevention Program (China), each containing 904,170 fundus photographs for model pre-training. Using identical pipelines, we train parallel foundation models using individual cohort and evaluate them on downstream tasks with publicly available datasets and held-out data from each site. The parallel models show competitive performance to data that differ substantially from their pre-training data. Nevertheless, we observe fairness gaps over age subgroups, whereas sex and ethnicity show minimal impact. These results demonstrate the good generalisability of retinal foundation models and indicate that pre-training demographic attributes shape fairness differently, highlighting the importance of domain-specific, fine-grained data curation for efficient foundation model development.
Article Details
Authors (28)
Yukun Zhou
Department of Ophthalmology, Shanghai Changhai Hospital, Naval Medical University
Zheyuan Wang
Yilan Wu
Ariel Yuhan Ong
Siegfried K. Wagner
Eden Ruffell
Mark A. Chia
Zhouyu Guan
Lie Ju
Justin Engelmann
David A. Merle
Tingyao Li
Jia Shu
Paul Nderitu
Ke Zou
Jocelyn Hui Lin Goh
Qingshan Hou
Xiaoxuan Liu
School of Life Sciences, Division of Life Sciences and Medicine, University of Science and Technology of China
Yaxing Wang
State Key Laboratory of Radiation Medicine and Protection, School of Radiation Medicine and Protection, Collaborative Innovation Center of Radiological Medicine of Jiangsu Higher Education Institutions, Biomedical Basic Research Center (BBRC) of Jiangsu
Yih Chung Tham
Andre Altmann
Carol Y. Cheung
Daniel C. Alexander
Eric J. Topol
Alastair K. Denniston
Tien Yin Wong
Bin Sheng
Pearse A. Keane