摘要
深度学习在癌症研究中的应用,特别是在早期诊断、病例理解和治疗策略设计方面,强调了高质量数据的必要性。生成式 AI,尤其是生成对抗网络 (GAN),已成为解决诸如类别不平衡、鲁棒学习和模型训练等挑战的领先解决方案,同时解决了患者隐私和真实数据稀缺带来的问题。尽管 GAN 很有前景,但它们面临着一些挑战,既有固有的挑战,也有针对组织病理学数据的特定挑战。固有问题包括训练不平衡、模式崩溃、从不足的鉴别器反馈中进行线性学习以及由于严格的反馈导致的硬边界收敛。组织病理学数据以其复杂的表示、高空间分辨率和多尺度特征呈现出独特的挑战。为了解决这些挑战,我们提出了一个由两个部分组成的框架。首先,我们引入了一种基于对比学习的多阶段渐进微调孪生神经网络 (MFT-SNN) 用于评估组织病理学斑块之间的相似性。其次,我们在 GAN 训练循环中实现了基于强化学习的外部优化器 (RL-EO),作为奖励信号生成器。修改后的鉴别器损失函数包含加权奖励,引导 GAN 最大化该奖励,同时最小化损失。这种方法为鉴别器提供了一个外部优化指南,防止生成器过度拟合并确保平滑收敛。我们提出的解决方案已针对最先进的 (SOTA) GAN 和去噪扩散概率模型进行了基准测试,在各种指标上优于之前的 SOTA,包括 FID 分数、KID 分数、感知路径长度和下游分类任务。