Liu Ruhan, Wang Tianqin, Li Huating, Zhang Ping, Li Jing, Yang Xiaokang, Shen Dinggang, Sheng Bin
IEEE Trans Med Imaging. 2023 Apr;42(4):1083-1094. doi: 10.1109/TMI.2022.3223683. Epub 2023 Apr 3.
Rare diseases, which are severely underrepresented in basic and clinical research, can particularly benefit from machine learning techniques. However, current learning-based approaches usually focus on either mono-modal image data or matched multi-modal data, whereas the diagnosis of rare diseases necessitates the aggregation of unstructured and unmatched multi-modal image data due to their rare and diverse nature. In this study, we therefore propose diagnosis-guided multi-to-mono modal generation networks (TMM-Nets) along with training and testing procedures. TMM-Nets can transfer data from multiple sources to a single modality for diagnostic data structurization. To demonstrate their potential in the context of rare diseases, TMM-Nets were deployed to diagnose the lupus retinopathy (LR-SLE), leveraging unmatched regular and ultra-wide-field fundus images for transfer learning. The TMM-Nets encoded the transfer learning from diabetic retinopathy to LR-SLE based on the similarity of the fundus lesions. In addition, a lesion-aware multi-scale attention mechanism was developed for clinical alerts, enabling TMM-Nets not only to inform patient care, but also to provide insights consistent with those of clinicians. An adversarial strategy was also developed to refine multi- to mono-modal image generation based on diagnostic results and the data distribution to enhance the data augmentation performance. Compared to the baseline model, the TMM-Nets showed 35.19% and 33.56% F1 score improvements on the test and external validation sets, respectively. In addition, the TMM-Nets can be used to develop diagnostic models for other rare diseases.
罕见疾病在基础研究和临床研究中的代表性严重不足,尤其能从机器学习技术中受益。然而,当前基于学习的方法通常聚焦于单模态图像数据或匹配的多模态数据,而罕见疾病的诊断由于其罕见性和多样性,需要聚合非结构化且不匹配的多模态图像数据。因此,在本研究中,我们提出了诊断引导的多对单模态生成网络(TMM-Nets)以及训练和测试程序。TMM-Nets可以将来自多个源的数据转换为单一模态,以实现诊断数据的结构化。为了证明其在罕见疾病背景下的潜力,TMM-Nets被用于诊断狼疮性视网膜病变(LR-SLE),利用不匹配的常规和超广角眼底图像进行迁移学习。TMM-Nets基于眼底病变的相似性,对从糖尿病性视网膜病变到LR-SLE的迁移学习进行编码。此外,还开发了一种病变感知多尺度注意力机制用于临床警报,使TMM-Nets不仅能够为患者护理提供信息,还能提供与临床医生一致的见解。还开发了一种对抗策略,根据诊断结果和数据分布来优化多对单模态图像生成,以提高数据增强性能。与基线模型相比,TMM-Nets在测试集和外部验证集上的F1分数分别提高了35.19%和33.56%。此外,TMM-Nets可用于开发其他罕见疾病的诊断模型。