提出课程学习框架CLIMD,提升不平衡多模态医疗诊断准确率
CLIMD: A Curriculum Learning Framework for Imbalanced Multimodal Diagnosis
- 基于模态内置信度与跨模态互补性设计课程度量机制
- 在多个数据集上超越现有方法,尤其改善少数类诊断性能
- 可插拔集成,适合需要精准处理罕见病的临床诊断系统
临床医生通常结合多源信息进行诊断,推动了多模态深度学习在医疗中的应用。然而,真实临床中因发病率差异,多模态医学数据普遍存在类别不平衡问题,导致模型难以充分学习少数类特征。现有方法多采用重采样或损失重加权,但易过拟合或欠拟合,且无法有效捕捉跨模态交互。为此,我们提出面向不平衡多模态诊断的课程学习框架CLIMD。首先设计多模态课程度量器,融合模态内置信度与跨模态互补性两个指标,引导模型聚焦关键样本,并逐步适应复杂类别分布;其次引入类别分布引导的训练调度器,使模型在训练过程中渐进式适应不平衡分布。在多个多模态医疗数据集上的大量实验表明,所提方法在各项指标上均优于当前最优方法,尤其在处理不平衡多模态医学数据方面表现突出。此外,作为即插即用的课程学习框架,CLIMD可轻松嵌入其他模型,为提升多模态疾病诊断精度提供可行路径。代码已公开于https://github.com/KHan-UJS/CLIMD。
原文摘要 · Abstract (English)
Clinicians usually combine information from multiple sources to achieve the most accurate diagnosis, and this has sparked increasing interest in leveraging multimodal deep learning for diagnosis. However, in real clinical scenarios, due to differences in incidence rates, multimodal medical data commonly face the issue of class imbalance, which makes it difficult to adequately learn the features of minority classes. Most existing methods tackle this issue with resampling or loss reweighting, but they are prone to overfitting or underfitting and fail to capture cross-modal interactions. Therefore, we propose a Curriculum Learning framework for Imbalanced Multimodal Diagnosis (CLIMD). Specifically, we first design multimodal curriculum measurer that combines two indicators, intra-modal confidence and inter-modal complementarity, to enable the model to focus on key samples and gradually adapt to complex category distributions. Additionally, a class distribution-guided training scheduler is introduced, which enables the model to progressively adapt to the imbalanced class distribution during training. Extensive experiments on multiple multimodal medical datasets demonstrate that the proposed method outperforms state-of-the-art approaches across various metrics and excels in handling imbalanced multimodal medical data. Furthermore, as a plug-and-play CL framework, CLIMD can be easily integrated into other models, offering a promising path for improving multimodal disease diagnosis accuracy. Code is publicly available at https://github.com/KHan-UJS/CLIMD.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。