通过多尺度特征融合提升眼底图像疾病分类准确率
MDF-MLLM: Deep Fusion Through Cross-Modal Feature Alignment for Contextually Aware Fundoscopic Image Classification
- 将U-Net多层特征融入LLaMA3的交叉注意力模块,实现视觉与文本深度对齐
- 在1305对眼底图像-文本数据上达到94%准确率,较基线提升56%
- 特别适合遗传性视网膜病等依赖临床文本的复杂疾病诊断
本研究提出一种新型多模态深度学习架构MDF-MLLM,旨在通过融合细粒度图像特征与全局文本语境,提升眼底图像的疾病分类准确性。现有多模态大模型常忽视诊断关键的低层空间细节。研究基于1,305对眼底图像-文本数据(来自FIVES、HRF和StoneRounds三个公开数据集),涵盖获得性和遗传性视网膜疾病,采用分类准确率与F1-score进行评估。该模型将四层U-Net编码器的跳跃连接特征注入到LLaMA 3.2 11B MLLM的交叉注意力块中,通过缩放交叉注意力与FiLM调制机制实现局部视觉特征与文本的逐片融合。基线模型在双类疾病分类任务中准确率为60%,而完全微调的MDF-MLLM达到94%的准确率,提升56%;召回率和F1-score分别提高67%和35%。消融实验表明,多层级特征融合显著增强空间推理能力,尤其在富含临床文本的遗传性疾病上表现突出。该框架具备可泛化、可解释、模块化特点,优于传统MLLM基线,有望应用于临床决策支持系统。未来工作将探索同步训练、扩展疾病种类并拓展至分割任务。
原文摘要 · Abstract (English)
This study aimed to enhance disease classification accuracy from retinal fundus images by integrating fine-grained image features and global textual context using a novel multimodal deep learning architecture. Existing multimodal large language models (MLLMs) often struggle to capture low-level spatial details critical for diagnosing retinal diseases such as glaucoma, diabetic retinopathy, and retinitis pigmentosa. This model development and validation study was conducted on 1,305 fundus image-text pairs compiled from three public datasets (FIVES, HRF, and StoneRounds), covering acquired and inherited retinal diseases, and evaluated using classification accuracy and F1-score. The MDF-MLLM integrates skip features from four U-Net encoder layers into cross-attention blocks within a LLaMA 3.2 11B MLLM. Vision features are patch-wise projected and fused using scaled cross-attention and FiLM-based U-Net modulation. Baseline MLLM achieved 60% accuracy on the dual-type disease classification task. MDF-MLLM, with both U-Net and MLLM components fully fine-tuned during training, achieved a significantly higher accuracy of 94%, representing a 56% improvement. Recall and F1-scores improved by as much as 67% and 35% over baseline, respectively. Ablation studies confirmed that the multi-depth fusion approach contributed to substantial gains in spatial reasoning and classification, particularly for inherited diseases with rich clinical text. MDF-MLLM presents a generalizable, interpretable, and modular framework for fundus image classification, outperforming traditional MLLM baselines through multi-scale feature fusion. The architecture holds promise for real-world deployment in clinical decision support systems. Future work will explore synchronized training techniques, a larger pool of diseases for more generalizability, and extending the model for segmentation tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。