Google Research, Mountain View, CA, USA.
DeepMind, London, UK.
Nat Biomed Eng. 2023 Jun;7(6):756-779. doi: 10.1038/s41551-023-01049-7. Epub 2023 Jun 8.
Machine-learning models for medical tasks can match or surpass the performance of clinical experts. However, in settings differing from those of the training dataset, the performance of a model can deteriorate substantially. Here we report a representation-learning strategy for machine-learning models applied to medical-imaging tasks that mitigates such 'out of distribution' performance problem and that improves model robustness and training efficiency. The strategy, which we named REMEDIS (for 'Robust and Efficient Medical Imaging with Self-supervision'), combines large-scale supervised transfer learning on natural images and intermediate contrastive self-supervised learning on medical images and requires minimal task-specific customization. We show the utility of REMEDIS in a range of diagnostic-imaging tasks covering six imaging domains and 15 test datasets, and by simulating three realistic out-of-distribution scenarios. REMEDIS improved in-distribution diagnostic accuracies up to 11.5% with respect to strong supervised baseline models, and in out-of-distribution settings required only 1-33% of the data for retraining to match the performance of supervised models retrained using all available data. REMEDIS may accelerate the development lifecycle of machine-learning models for medical imaging.
用于医学任务的机器学习模型可以与临床专家的表现相匹配或超越。然而,在与训练数据集不同的环境中,模型的性能可能会大幅下降。在这里,我们报告了一种用于医学成像任务的机器学习模型的表示学习策略,该策略可以缓解这种“分布外”性能问题,并提高模型的鲁棒性和训练效率。该策略名为 REMEDIS(代表“具有自我监督的稳健和高效医学成像”),它结合了自然图像的大规模监督迁移学习和医学图像的中间对比自我监督学习,并且只需要最小的特定于任务的定制。我们在涵盖六个成像领域和 15 个测试数据集的一系列诊断成像任务中展示了 REMEDIS 的实用性,并通过模拟三种现实的分布外场景来展示其实用性。与强大的监督基线模型相比,REMEID 提高了 11.5%的分布内诊断准确率,而在分布外设置中,仅需重新训练 1-33%的数据即可达到使用所有可用数据重新训练的监督模型的性能。REMEID 可能会加速医学成像领域的机器学习模型的开发周期。
Nat Biomed Eng. 2022-12
Comput Methods Programs Biomed. 2022-7
Med Image Anal. 2019-7-26
J Am Med Inform Assoc. 2023-11-17
Nat Biomed Eng. 2025-9-5
Insights Imaging. 2025-8-6
NPJ Digit Med. 2025-7-22
Imaging Sci Dent. 2025-6
Aerobiologia (Bologna). 2025
Patterns (N Y). 2021-12-7
NPJ Digit Med. 2021-4-19
Science. 2021-4-2