摘要
arXiv:2504.15865v1 交叉类型: cross
摘要:深度学习(DL)在医学影像领域取得了显著进展。然而,将DL模型适应医学任务仍然是一项重大挑战,主要是由于两个关键因素:(1)架构选择,因为不同的任务需要专门设计的模型架构;(2)权重初始化,这直接影响模型的收敛速度和最终性能。尽管从ImageNet进行迁移学习是一种广泛采用的策略,但其效果受到自然图像和医学图像之间显著差异的限制。为了解决这些挑战,我们引入了Medical Neural Network Search(MedNNS),这是首个专门针对医学影像应用的神经网络搜索框架。MedNNS通过构建一个元空间来联合优化架构选择和权重初始化,该元空间基于数据集和模型的性能交织方式对它们进行编码。我们使用Supernetwork为基础的方法构建了这个空间,使模型动物园的规模扩大了51倍,超过了之前最先进的(SOTA)方法。此外,我们在空间构建过程中引入了排名损失和Fréchet Inception Distance(FID)损失,以捕捉模型间的相互关系和数据集间的相互关系,从而在元空间中实现更准确的对齐。在多个数据集上的实验结果表明,MedNNS显著优于ImageNet预训练的DL模型和SOTA神经架构搜索(NAS)方法,在多个数据集上平均提高了1.7%的准确率,同时收敛速度更快。代码和处理后的元空间可在https://github.com/BioMedIA-MBZUAI/MedNNS获得。