Aug, 2023

跨体系结构自监督下的高效医疗表示学习

TL;DR在医疗和生物医学应用中,极端的计算要求对采用表示学习构成了重大障碍。本研究提出了一种名为 CASS(Cross Architectural - Self Supervision)的创新孪生自监督学习方法,通过协同使用 Transformer 和卷积神经网络(CNN)进行高效学习,从而克服了现有自监督学习方法在临床实验环境中使用较小批量大小或较短预训练周期时性能降低的问题,并且在四个多样化的医疗数据集上优于现有方法。实验证明,CASS 训练的 CNN 和 Transformer 在仅有 1% 标记数据用于微调时平均性能提高了 3.8%,在 10% 标记数据下提高了 5.9%,而在 100% 标记数据下提高了惊人的 10.13%。值得注意的是,CASS 相比现有最先进方法减少了预训练时间 69%,更适合临床实施,并且 CASS 对批量大小和预训练周期的变化更为稳健,使其成为医疗应用中的机器学习的合适选择。