用预训练+图推理提升舌象诊断准确率
MIRNet: Integrating Constrained Graph-Based Reasoning with Pre-training for Diagnostic Medical Imaging
- 结合自监督预训练与专家构建的图结构推理
- 在4000张舌象图上达到领先性能
- 适合医疗图像分析、小样本诊断场景
自动解读医学影像需建模复杂视觉-语义关系,同时应对标注稀缺、标签不平衡及临床合理性约束。本文提出MIRNet(医学图像推理网络),融合自监督预训练与受约束的图推理机制。针对舌象诊断这一需精细视觉与语义理解的任务,方法利用自监督掩码自编码器(MAE)从无标签数据学习可迁移视觉表征;通过图注意力网络(GAT)建模专家定义的标签相关性;采用基于KL散度与正则化损失的约束感知优化以融入临床先验;并用非对称损失(ASL)和提升集成缓解类别不平衡。为解决标注稀缺问题,我们构建了TongueAtlas-4K——首个涵盖4000张图像、22个诊断标签的专家标注基准数据集。实验验证该方法性能达当前最优水平。尽管专精于舌象诊断,该框架亦可推广至更广泛的医学影像诊断任务。
原文摘要 · Abstract (English)
Automated interpretation of medical images demands robust modeling of complex visual-semantic relationships while addressing annotation scarcity, label imbalance, and clinical plausibility constraints. We introduce MIRNet (Medical Image Reasoner Network), a novel framework that integrates self-supervised pre-training with constrained graph-based reasoning. Tongue image diagnosis is a particularly challenging domain that requires fine-grained visual and semantic understanding. Our approach leverages self-supervised masked autoencoder (MAE) to learn transferable visual representations from unlabeled data; employs graph attention networks (GAT) to model label correlations through expert-defined structured graphs; enforces clinical priors via constraint-aware optimization using KL divergence and regularization losses; and mitigates imbalance using asymmetric loss (ASL) and boosting ensembles. To address annotation scarcity, we also introduce TongueAtlas-4K, a comprehensive expert-curated benchmark comprising 4,000 images annotated with 22 diagnostic labels--representing the largest public dataset in tongue analysis. Validation shows our method achieves state-of-the-art performance. While optimized for tongue diagnosis, the framework readily generalizes to broader diagnostic medical imaging tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。